refactor(battle-node): unify IMatchingBridge.RegisterBattle signature

Single RegisterBattle(p1, p2?, type) with contract validation throws on
invalid combinations (Pvp requires both; Bot requires p2==null; Scripted
accepts either). PendingBattle carries Type + P1 + nullable P2. Handler
+ controller adapt; v1.2 behaviour preserved because Scripted is the
only type used today (Phase 2 adds Pvp, Phase 3 adds Bot).
This commit is contained in:
gamer147
2026-06-01 20:00:52 -04:00
parent acd0997cfb
commit d665f88067
8 changed files with 111 additions and 37 deletions

View File

@@ -1,12 +1,25 @@
using SVSim.BattleNode.Sessions;
namespace SVSim.BattleNode.Bridge; namespace SVSim.BattleNode.Bridge;
public interface IMatchingBridge public interface IMatchingBridge
{ {
/// <summary> /// <summary>
/// Mint a battle id, register a pending session for the given viewer with their per-battle /// Mint a battle id, register a pending session, return the URL the client should
/// MatchContext snapshot, and return the URL the client should open a socket to. /// open a socket to.
/// </summary> /// </summary>
PendingMatch RegisterPendingBattle(long viewerId, MatchContext context); /// <remarks>
/// Contract rules (enforced; throws <see cref="ArgumentException"/>):
/// <list type="bullet">
/// <item><c>Pvp</c>: <paramref name="p2"/> required. Both viewers expected to
/// connect WS within 60s.</item>
/// <item><c>Bot</c>: <paramref name="p2"/> must be null. One viewer expected;
/// opponent runs in client.</item>
/// <item><c>Scripted</c>: <paramref name="p2"/> currently null; future
/// server-driven bot config rides on <paramref name="p2"/>.</item>
/// </list>
/// </remarks>
PendingMatch RegisterBattle(BattlePlayer p1, BattlePlayer? p2, BattleType type);
} }
public sealed record PendingMatch(string BattleId, string NodeServerUrl); public sealed record PendingMatch(string BattleId, string NodeServerUrl);

View File

@@ -5,10 +5,8 @@ namespace SVSim.BattleNode.Bridge;
/// <summary> /// <summary>
/// In-process implementation of <see cref="IMatchingBridge"/>. The HTTP-side /// In-process implementation of <see cref="IMatchingBridge"/>. The HTTP-side
/// <c>do_matching</c> controller calls <see cref="RegisterPendingBattle"/>, which mints a /// matching queue calls <see cref="RegisterBattle"/> once it has decided "these two
/// 12-digit decimal battle id, stashes a <see cref="PendingBattle"/> entry in the /// play each other" or "this viewer is solo (bot/scripted)."
/// <see cref="IBattleSessionStore"/>, and returns the node URL the client should connect to.
/// The WebSocket handler resolves the same battle id back to the viewer on connect.
/// </summary> /// </summary>
public sealed class MatchingBridge : IMatchingBridge public sealed class MatchingBridge : IMatchingBridge
{ {
@@ -21,17 +19,39 @@ public sealed class MatchingBridge : IMatchingBridge
_options = options; _options = options;
} }
public PendingMatch RegisterPendingBattle(long viewerId, MatchContext context) public PendingMatch RegisterBattle(BattlePlayer p1, BattlePlayer? p2, BattleType type)
{ {
ValidateContract(p1, p2, type);
// 12-digit decimal battle id mirrors the captures (e.g. "975695075012"). // 12-digit decimal battle id mirrors the captures (e.g. "975695075012").
// Two unbiased 6-digit draws concatenated — RandomNumberGenerator.GetInt32 uses // Two unbiased 6-digit draws concatenated — RandomNumberGenerator.GetInt32 uses
// rejection sampling so the result is uniform on [0, 10^6). The previous // rejection sampling so the result is uniform on [0, 10^6).
// implementation (Guid.GetHashCode + mod) collapsed 128 bits into ~32 of entropy
// and tripped Math.Abs(int.MinValue) UB on the unlucky hash.
var hi = RandomNumberGenerator.GetInt32(0, 1_000_000); var hi = RandomNumberGenerator.GetInt32(0, 1_000_000);
var lo = RandomNumberGenerator.GetInt32(0, 1_000_000); var lo = RandomNumberGenerator.GetInt32(0, 1_000_000);
var battleId = $"{hi:D6}{lo:D6}"; var battleId = $"{hi:D6}{lo:D6}";
_store.RegisterPending(new PendingBattle(battleId, viewerId, context));
_store.RegisterPending(new PendingBattle(battleId, type, p1, p2));
return new PendingMatch(battleId, _options.NodeServerUrl); return new PendingMatch(battleId, _options.NodeServerUrl);
} }
private static void ValidateContract(BattlePlayer p1, BattlePlayer? p2, BattleType type)
{
if (p1 is null) throw new ArgumentNullException(nameof(p1));
switch (type)
{
case BattleType.Pvp:
if (p2 is null) throw new ArgumentException("Pvp requires both p1 and p2.", nameof(p2));
if (p1.ViewerId == p2.ViewerId)
throw new ArgumentException("Pvp requires distinct viewer ids.", nameof(p2));
break;
case BattleType.Bot:
if (p2 is not null) throw new ArgumentException("Bot must have p2==null.", nameof(p2));
break;
case BattleType.Scripted:
// p2 currently null; future server-driven bot will populate it.
break;
default:
throw new ArgumentOutOfRangeException(nameof(type), type, "Unknown BattleType.");
}
}
} }

View File

@@ -93,18 +93,20 @@ public sealed class BattleNodeWebSocketHandler
ctx.Response.StatusCode = StatusCodes.Status404NotFound; ctx.Response.StatusCode = StatusCodes.Status404NotFound;
return; return;
} }
if (pending.ViewerId != viewerId) if (pending.P1.ViewerId != viewerId)
{ {
_log.LogWarning( _log.LogWarning(
"WS upgrade viewer-id mismatch on BattleId={Bid}: bridge expected={Expected}, decrypted={Got}.", "WS upgrade viewer-id mismatch on BattleId={Bid}: bridge expected={Expected}, decrypted={Got}.",
battleId, pending.ViewerId, viewerId); battleId, pending.P1.ViewerId, viewerId);
ctx.Response.StatusCode = StatusCodes.Status401Unauthorized; ctx.Response.StatusCode = StatusCodes.Status401Unauthorized;
return; return;
} }
var ws = await ctx.WebSockets.AcceptWebSocketAsync(); var ws = await ctx.WebSockets.AcceptWebSocketAsync();
_store.RemovePending(battleId); _store.RemovePending(battleId);
var session = new BattleSession(ws, battleId, viewerId, pending.Context, _loggerFactory.CreateLogger<BattleSession>()); // Phase 1: handler still constructs the old single-WS BattleSession.
// Task 9 switches to BattleSessionV2 + RealParticipant + ScriptedBotParticipant.
var session = new BattleSession(ws, battleId, viewerId, pending.P1.Context, _loggerFactory.CreateLogger<BattleSession>());
await session.RunAsync(ctx.RequestAborted); await session.RunAsync(ctx.RequestAborted);
} }

View File

@@ -3,8 +3,8 @@ using SVSim.BattleNode.Bridge;
namespace SVSim.BattleNode.Sessions; namespace SVSim.BattleNode.Sessions;
/// <summary> /// <summary>
/// Sparse pre-connect record: viewer id + the per-battle MatchContext snapshot. Enough to /// Sparse pre-connect record. Carries the battle type + one or two players. The
/// validate the incoming WS connect, resolve the viewer, and seed the BattleSession with the /// WebSocket handler reads this to validate the incoming WS connect and to
/// player-half lifecycle data. Full BattleSession is created on connect. /// construct the right participants.
/// </summary> /// </summary>
public sealed record PendingBattle(string BattleId, long ViewerId, MatchContext Context); public sealed record PendingBattle(string BattleId, BattleType Type, BattlePlayer P1, BattlePlayer? P2);

View File

@@ -30,7 +30,10 @@ public class ArenaTwoPickBattleController : SVSimController
try try
{ {
var ctx = await _matchContextBuilder.BuildForTwoPickAsync(vid); var ctx = await _matchContextBuilder.BuildForTwoPickAsync(vid);
var match = _matching.RegisterPendingBattle(vid, ctx); var match = _matching.RegisterBattle(
new SVSim.BattleNode.Bridge.BattlePlayer(vid, ctx),
p2: null,
SVSim.BattleNode.Sessions.BattleType.Scripted);
return Ok(new DoMatchingResponseDto return Ok(new DoMatchingResponseDto
{ {
MatchingState = 3004, MatchingState = 3004,

View File

@@ -8,44 +8,74 @@ namespace SVSim.UnitTests.BattleNode.Bridge;
public class MatchingBridgeTests public class MatchingBridgeTests
{ {
[Test] [Test]
public void RegisterPendingBattle_RegistersInStoreAndReturnsNodeUrl() public void RegisterBattle_Scripted_stores_pending_and_returns_node_url()
{ {
var store = new InMemoryBattleSessionStore(); var store = new InMemoryBattleSessionStore();
var bridge = new MatchingBridge(store, new BattleNodeOptions { NodeServerUrl = "localhost:5148/socket.io/" }); var bridge = new MatchingBridge(store, new BattleNodeOptions { NodeServerUrl = "localhost:5148/socket.io/" });
var ctx = FixtureCtx(); var p1 = new BattlePlayer(906243102, FixtureCtx());
var match = bridge.RegisterPendingBattle(viewerId: 906243102, context: ctx); var match = bridge.RegisterBattle(p1, p2: null, BattleType.Scripted);
Assert.That(match.NodeServerUrl, Is.EqualTo("localhost:5148/socket.io/")); Assert.That(match.NodeServerUrl, Is.EqualTo("localhost:5148/socket.io/"));
Assert.That(match.BattleId, Is.Not.Empty); Assert.That(match.BattleId, Is.Not.Empty);
var pending = store.TryGetPending(match.BattleId); var pending = store.TryGetPending(match.BattleId);
Assert.That(pending, Is.Not.Null); Assert.That(pending, Is.Not.Null);
Assert.That(pending!.ViewerId, Is.EqualTo(906243102)); Assert.That(pending!.Type, Is.EqualTo(BattleType.Scripted));
Assert.That(pending.Context, Is.SameAs(ctx)); Assert.That(pending.P1.ViewerId, Is.EqualTo(906243102));
Assert.That(pending.P2, Is.Null);
} }
[Test] [Test]
public void RegisterPendingBattle_MintsUniqueBattleIds() public void RegisterBattle_mints_unique_BattleIds_per_call()
{ {
var bridge = new MatchingBridge(new InMemoryBattleSessionStore(), new BattleNodeOptions()); var bridge = new MatchingBridge(new InMemoryBattleSessionStore(), new BattleNodeOptions());
var a = bridge.RegisterPendingBattle(1, FixtureCtx()); var a = bridge.RegisterBattle(new BattlePlayer(1, FixtureCtx()), null, BattleType.Scripted);
var b = bridge.RegisterPendingBattle(2, FixtureCtx()); var b = bridge.RegisterBattle(new BattlePlayer(2, FixtureCtx()), null, BattleType.Scripted);
Assert.That(a.BattleId, Is.Not.EqualTo(b.BattleId)); Assert.That(a.BattleId, Is.Not.EqualTo(b.BattleId));
} }
[Test] [Test]
public void RegisterPendingBattle_ProducesTwelveDigitDecimalBattleId() public void RegisterBattle_produces_twelve_digit_decimal_BattleId()
{ {
var bridge = new MatchingBridge(new InMemoryBattleSessionStore(), new BattleNodeOptions()); var bridge = new MatchingBridge(new InMemoryBattleSessionStore(), new BattleNodeOptions());
var match = bridge.RegisterPendingBattle(viewerId: 1, context: FixtureCtx()); var match = bridge.RegisterBattle(new BattlePlayer(1, FixtureCtx()), null, BattleType.Scripted);
Assert.That(match.BattleId, Has.Length.EqualTo(12)); Assert.That(match.BattleId, Has.Length.EqualTo(12));
Assert.That(match.BattleId, Does.Match("^[0-9]{12}$")); Assert.That(match.BattleId, Does.Match("^[0-9]{12}$"));
} }
[Test]
public void RegisterBattle_Pvp_requires_both_players()
{
var bridge = new MatchingBridge(new InMemoryBattleSessionStore(), new BattleNodeOptions());
Assert.Throws<ArgumentException>(() =>
bridge.RegisterBattle(new BattlePlayer(1, FixtureCtx()), p2: null, BattleType.Pvp));
}
[Test]
public void RegisterBattle_Pvp_rejects_same_viewer_twice()
{
var bridge = new MatchingBridge(new InMemoryBattleSessionStore(), new BattleNodeOptions());
Assert.Throws<ArgumentException>(() =>
bridge.RegisterBattle(
new BattlePlayer(1, FixtureCtx()), new BattlePlayer(1, FixtureCtx()), BattleType.Pvp));
}
[Test]
public void RegisterBattle_Bot_rejects_non_null_p2()
{
var bridge = new MatchingBridge(new InMemoryBattleSessionStore(), new BattleNodeOptions());
Assert.Throws<ArgumentException>(() =>
bridge.RegisterBattle(
new BattlePlayer(1, FixtureCtx()), new BattlePlayer(2, FixtureCtx()), BattleType.Bot));
}
private static MatchContext FixtureCtx() => new( private static MatchContext FixtureCtx() => new(
SelfDeckCardIds: Enumerable.Range(1, 30).Select(i => 100_011_010L).ToList(), SelfDeckCardIds: Enumerable.Range(1, 30).Select(i => 100_011_010L).ToList(),
ClassId: "1", CharaId: "1", CardMasterName: "card_master_node_10015", ClassId: "1", CharaId: "1", CardMasterName: "card_master_node_10015",

View File

@@ -32,7 +32,10 @@ public class BattleNodeFlowTests
using var cts = new CancellationTokenSource(TimeSpan.FromSeconds(15)); using var cts = new CancellationTokenSource(TimeSpan.FromSeconds(15));
var ct = cts.Token; var ct = cts.Token;
var pending = bridge.RegisterPendingBattle(viewerId: 906243102, context: FixtureCtx()); var pending = bridge.RegisterBattle(
new SVSim.BattleNode.Bridge.BattlePlayer(906243102, FixtureCtx()),
p2: null,
SVSim.BattleNode.Sessions.BattleType.Scripted);
var key = MakeKey(); var key = MakeKey();
var encryptedVid = NodeCrypto.EncryptForNode("906243102", key); var encryptedVid = NodeCrypto.EncryptForNode("906243102", key);
@@ -132,7 +135,10 @@ public class BattleNodeFlowTests
using var cts = new CancellationTokenSource(TimeSpan.FromSeconds(15)); using var cts = new CancellationTokenSource(TimeSpan.FromSeconds(15));
var ct = cts.Token; var ct = cts.Token;
var pending = bridge.RegisterPendingBattle(viewerId: vid, context: ctx); var pending = bridge.RegisterBattle(
new SVSim.BattleNode.Bridge.BattlePlayer(vid, ctx),
p2: null,
SVSim.BattleNode.Sessions.BattleType.Scripted);
var key = MakeKey(); var key = MakeKey();
var encryptedVid = NodeCrypto.EncryptForNode(vid.ToString(), key); var encryptedVid = NodeCrypto.EncryptForNode(vid.ToString(), key);

View File

@@ -14,7 +14,7 @@ public class InMemoryBattleSessionStoreTests
[Test] [Test]
public void RegisterThenGet_ReturnsRegisteredBattle() public void RegisterThenGet_ReturnsRegisteredBattle()
{ {
var battle = new PendingBattle("bid-1", 906243102, FixtureCtx()); var battle = new PendingBattle("bid-1", BattleType.Scripted, new BattlePlayer(906243102, FixtureCtx()), null);
_store.RegisterPending(battle); _store.RegisterPending(battle);
Assert.That(_store.TryGetPending("bid-1"), Is.EqualTo(battle)); Assert.That(_store.TryGetPending("bid-1"), Is.EqualTo(battle));
@@ -29,7 +29,7 @@ public class InMemoryBattleSessionStoreTests
[Test] [Test]
public void Remove_ReturnsTrueWhenPresent_FalseWhenAbsent() public void Remove_ReturnsTrueWhenPresent_FalseWhenAbsent()
{ {
_store.RegisterPending(new PendingBattle("bid", 1, FixtureCtx())); _store.RegisterPending(new PendingBattle("bid", BattleType.Scripted, new BattlePlayer(1, FixtureCtx()), null));
Assert.That(_store.RemovePending("bid"), Is.True); Assert.That(_store.RemovePending("bid"), Is.True);
Assert.That(_store.RemovePending("bid"), Is.False); Assert.That(_store.RemovePending("bid"), Is.False);
} }
@@ -37,9 +37,9 @@ public class InMemoryBattleSessionStoreTests
[Test] [Test]
public void Register_DuplicateBattleId_OverwritesPrior() public void Register_DuplicateBattleId_OverwritesPrior()
{ {
_store.RegisterPending(new PendingBattle("bid", 1, FixtureCtx())); _store.RegisterPending(new PendingBattle("bid", BattleType.Scripted, new BattlePlayer(1, FixtureCtx()), null));
_store.RegisterPending(new PendingBattle("bid", 2, FixtureCtx())); _store.RegisterPending(new PendingBattle("bid", BattleType.Scripted, new BattlePlayer(2, FixtureCtx()), null));
Assert.That(_store.TryGetPending("bid")!.ViewerId, Is.EqualTo(2)); Assert.That(_store.TryGetPending("bid")!.P1.ViewerId, Is.EqualTo(2));
} }
private static MatchContext FixtureCtx() => new( private static MatchContext FixtureCtx() => new(