feat: add ClientFlags bitfield with thread-safe holder

This commit is contained in:
Joseph Doherty
2026-02-22 23:33:21 -05:00
parent 8bbfa54058
commit 1a916a3f36
2 changed files with 94 additions and 0 deletions

View File

@@ -0,0 +1,53 @@
namespace NATS.Server.Tests;
public class ClientFlagsTests
{
[Fact]
public void SetFlag_and_HasFlag_work()
{
var holder = new ClientFlagHolder();
holder.HasFlag(ClientFlags.ConnectReceived).ShouldBeFalse();
holder.SetFlag(ClientFlags.ConnectReceived);
holder.HasFlag(ClientFlags.ConnectReceived).ShouldBeTrue();
}
[Fact]
public void ClearFlag_removes_flag()
{
var holder = new ClientFlagHolder();
holder.SetFlag(ClientFlags.ConnectReceived);
holder.SetFlag(ClientFlags.IsSlowConsumer);
holder.ClearFlag(ClientFlags.ConnectReceived);
holder.HasFlag(ClientFlags.ConnectReceived).ShouldBeFalse();
holder.HasFlag(ClientFlags.IsSlowConsumer).ShouldBeTrue();
}
[Fact]
public void Multiple_flags_can_be_set_independently()
{
var holder = new ClientFlagHolder();
holder.SetFlag(ClientFlags.ConnectReceived);
holder.SetFlag(ClientFlags.WriteLoopStarted);
holder.SetFlag(ClientFlags.FirstPongSent);
holder.HasFlag(ClientFlags.ConnectReceived).ShouldBeTrue();
holder.HasFlag(ClientFlags.WriteLoopStarted).ShouldBeTrue();
holder.HasFlag(ClientFlags.FirstPongSent).ShouldBeTrue();
holder.HasFlag(ClientFlags.IsSlowConsumer).ShouldBeFalse();
}
[Fact]
public void SetFlag_is_thread_safe()
{
var holder = new ClientFlagHolder();
var flags = Enum.GetValues<ClientFlags>();
Parallel.ForEach(flags, flag => holder.SetFlag(flag));
foreach (var flag in flags)
holder.HasFlag(flag).ShouldBeTrue();
}
}