144 lines
3.4 KiB
C#
144 lines
3.4 KiB
C#
using System.Collections.Concurrent;
|
|
using System.Diagnostics;
|
|
using ClientTester.Packet;
|
|
using LiteNetLib;
|
|
using LiteNetLib.Utils;
|
|
using Serilog;
|
|
|
|
namespace ClientTester.EchoDummyService;
|
|
|
|
public class DummyClients
|
|
{
|
|
private NetManager manager;
|
|
private EventBasedNetListener listener;
|
|
private NetDataWriter? writer;
|
|
public NetPeer? peer;
|
|
public int clientId;
|
|
|
|
// seq → 송신 타임스탬프 (Stopwatch tick)
|
|
private ConcurrentDictionary<int, long> pendingPings = new();
|
|
private int seqNumber;
|
|
|
|
// 유닛 테스트용 (0 = 제한 없음)
|
|
public int TestCount
|
|
{
|
|
get;
|
|
set;
|
|
} = 0;
|
|
|
|
// 통계
|
|
public int SentCount
|
|
{
|
|
set;
|
|
get;
|
|
}
|
|
|
|
public int ReceivedCount
|
|
{
|
|
set;
|
|
get;
|
|
}
|
|
|
|
public double LastRttMs
|
|
{
|
|
set;
|
|
get;
|
|
}
|
|
|
|
public double TotalRttMs
|
|
{
|
|
set;
|
|
get;
|
|
}
|
|
|
|
public int RttCount
|
|
{
|
|
set;
|
|
get;
|
|
}
|
|
|
|
public DummyClients(int clientId, string ip, int port, string key)
|
|
{
|
|
this.clientId = clientId;
|
|
listener = new EventBasedNetListener();
|
|
manager = new NetManager(listener);
|
|
writer = new NetDataWriter();
|
|
|
|
listener.PeerConnectedEvent += netPeer =>
|
|
{
|
|
peer = netPeer;
|
|
Log.Information("[Client {ClientId:00}] 연결됨", this.clientId);
|
|
};
|
|
|
|
listener.NetworkReceiveEvent += (peer, reader, channel, deliveryMethod) =>
|
|
{
|
|
short code = reader.GetShort();
|
|
short bodyLength = reader.GetShort();
|
|
string? msg = reader.GetString();
|
|
long sentTick;
|
|
|
|
if (msg != null && msg.StartsWith("Echo seq:") &&
|
|
int.TryParse(msg.Substring("Echo seq:".Length), out int seq) &&
|
|
pendingPings.TryRemove(seq, out sentTick))
|
|
{
|
|
double rttMs = (Stopwatch.GetTimestamp() - sentTick) * 1000.0 / Stopwatch.Frequency;
|
|
LastRttMs = rttMs;
|
|
TotalRttMs += rttMs;
|
|
RttCount++;
|
|
}
|
|
|
|
ReceivedCount++;
|
|
|
|
if (TestCount > 0 && ReceivedCount >= TestCount)
|
|
{
|
|
peer.Disconnect();
|
|
}
|
|
|
|
reader.Recycle();
|
|
};
|
|
|
|
listener.PeerDisconnectedEvent += (peer, info) =>
|
|
{
|
|
Log.Warning("[Client {ClientId:00}] 연결 끊김: {Reason}", this.clientId, info.Reason);
|
|
this.peer = null;
|
|
};
|
|
|
|
manager.Start();
|
|
manager.Connect(ip, port, key);
|
|
}
|
|
|
|
public void SendPing()
|
|
{
|
|
if (peer == null || writer == null)
|
|
{
|
|
return;
|
|
}
|
|
|
|
int seq = seqNumber++;
|
|
pendingPings[seq] = Stopwatch.GetTimestamp();
|
|
|
|
PacketHeader packetHeader = new PacketHeader();
|
|
packetHeader.Code = 0;
|
|
packetHeader.BodyLength = $"seq:{seq}".Length;
|
|
writer.Put((short)packetHeader.Code);
|
|
writer.Put((short)packetHeader.BodyLength);
|
|
writer.Put($"Echo seq:{seq}");
|
|
// 순서보장 안함 HOL Blocking 제거
|
|
peer.Send(writer, DeliveryMethod.ReliableUnordered);
|
|
SentCount++;
|
|
writer.Reset();
|
|
}
|
|
|
|
public double AvgRttMs => RttCount > 0 ? TotalRttMs / RttCount : 0.0;
|
|
|
|
public void PollEvents()
|
|
{
|
|
manager.PollEvents();
|
|
}
|
|
|
|
public void Stop()
|
|
{
|
|
manager.Stop();
|
|
}
|
|
}
|