Add a benchmark for device networking (#12619)

2022-12-07 00:22:57 +01:00
parent 21906f45bf
commit cc8b90dc44
2 changed files with 138 additions and 0 deletions
--- a/Content.Benchmarks/DeviceNetworkingBenchmark.cs
+++ b/Content.Benchmarks/DeviceNetworkingBenchmark.cs
@@ -0,0 +1,133 @@
+using System;
+using System.Collections.Generic;
+using System.Threading.Tasks;
+using BenchmarkDotNet.Attributes;
+using Content.IntegrationTests;
+using Content.IntegrationTests.Tests.DeviceNetwork;
+using Content.Server.DeviceNetwork;
+using Content.Server.DeviceNetwork.Systems;
+using Robust.Shared;
+using Robust.Shared.Analyzers;
+using Robust.Shared.GameObjects;
+using Robust.Shared.IoC;
+using Robust.Shared.Map;
+
+namespace Content.Benchmarks;
+
+[Virtual]
+[MemoryDiagnoser]
+public class DeviceNetworkingBenchmark
+{
+    private PairTracker _pair = default!;
+    private IEntityManager _entityManager = default!;
+    private DeviceNetworkTestSystem _deviceNetTestSystem = default!;
+    private DeviceNetworkSystem _deviceNetworkSystem = default!;
+    private EntityUid _sourceEntity;
+    private EntityUid _sourceWirelessEntity;
+    private List<EntityUid> _targetEntities = new();
+    private List<EntityUid> _targetWirelessEntities = new();
+
+
+    private NetworkPayload _payload = default!;
+    private const string Prototypes = @"
+- type: entity
+  name: DummyNetworkDevice
+  id: DummyNetworkDevice
+  components:
+    - type: DeviceNetwork
+      transmitFrequency: 100
+      receiveFrequency: 100
+      deviceNetId: Private
+- type: entity
+  name: DummyWirelessNetworkDevice
+  id: DummyWirelessNetworkDevice
+  components:
+    - type: DeviceNetwork
+      transmitFrequency: 100
+      receiveFrequency: 100
+      deviceNetId: Wireless
+    - type: WirelessNetworkConnection
+      range: 100
+        ";
+
+    //public static IEnumerable<int> EntityCountSource { get; set; }
+
+    //[ParamsSource(nameof(EntityCountSource))]
+    public int EntityCount = 500;
+
+    [GlobalSetup]
+    public void Setup()
+    {
+        ProgramShared.PathOffset = "../../../../";
+        _pair = PoolManager.GetServerClient(new PoolSettings{NoClient = true, ExtraPrototypes = Prototypes}).GetAwaiter().GetResult();
+        var server = _pair.Pair.Server;
+
+        _entityManager = server.ResolveDependency<IEntityManager>();
+        _deviceNetworkSystem = _entityManager.EntitySysManager.GetEntitySystem<DeviceNetworkSystem>();
+        _deviceNetTestSystem = _entityManager.EntitySysManager.GetEntitySystem<DeviceNetworkTestSystem>();
+
+        IoCManager.InitThread(_pair.Pair.Server.InstanceDependencyCollection);
+
+        var testValue = "test";
+        _payload = new NetworkPayload
+        {
+            ["Test"] = testValue,
+            ["testnumber"] = 1,
+            ["testbool"] = true
+        };
+
+        _sourceEntity = _entityManager.SpawnEntity("DummyNetworkDevice", MapCoordinates.Nullspace);
+        _sourceWirelessEntity = _entityManager.SpawnEntity("DummyWirelessNetworkDevice", MapCoordinates.Nullspace);
+
+        for (var i = 0; i < EntityCount; i++)
+        {
+            _targetEntities.Add(_entityManager.SpawnEntity("DummyNetworkDevice", MapCoordinates.Nullspace));
+            _targetWirelessEntities.Add(_entityManager.SpawnEntity("DummyWirelessNetworkDevice", MapCoordinates.Nullspace));
+        }
+    }
+
+    [Benchmark(Baseline = true, Description = "Entity Events")]
+    public async Task EventSentBaseline()
+    {
+        var server = _pair.Pair.Server;
+
+        _pair.Pair.Server.Post(() =>
+        {
+            foreach (var entity in _targetEntities)
+            {
+                _deviceNetTestSystem.SendBaselineTestEvent(entity);
+            }
+        });
+
+        await server.WaitRunTicks(1);
+        await server.WaitIdleAsync();
+    }
+
+    [Benchmark(Description = "Device Net Broadcast No Connection Checks")]
+    public async Task DeviceNetworkBroadcastNoConnectionChecks()
+    {
+        var server = _pair.Pair.Server;
+
+        _pair.Pair.Server.Post(() =>
+        {
+            _deviceNetworkSystem.QueuePacket(_sourceEntity, null, _payload, 100);
+        });
+
+        await server.WaitRunTicks(1);
+        await server.WaitIdleAsync();
+    }
+
+    [Benchmark(Description = "Device Net Broadcast Wireless Connection Checks")]
+    public async Task DeviceNetworkBroadcastWirelessConnectionChecks()
+    {
+        var server = _pair.Pair.Server;
+
+        _pair.Pair.Server.Post(() =>
+        {
+            _deviceNetworkSystem.QueuePacket(_sourceWirelessEntity, null, _payload, 100);
+        });
+
+        await server.WaitRunTicks(1);
+        await server.WaitIdleAsync();
+    }
+}