@@ -15,7 +15,7 @@ public class LobbyCommandServer
1515 {
1616 Socket serverSocket ;
1717 readonly int port ;
18- readonly OrderManager orderManager ;
18+ OrderManager orderManager ;
1919 bool isRunning ;
2020 public const string CurrentApiVersion = "1.0" ;
2121
@@ -82,6 +82,11 @@ public LobbyCommandServer(int port, OrderManager orderManager)
8282 this . orderManager = orderManager ;
8383 }
8484
85+ public void UpdateOrderManager ( OrderManager orderManager )
86+ {
87+ this . orderManager = orderManager ;
88+ }
89+
8590 ~ LobbyCommandServer ( )
8691 {
8792 End ( ) ;
@@ -291,11 +296,22 @@ async Task HandleClient(Socket clientSocket)
291296 if ( request . Params == null )
292297 request . Params = new JObject ( ) ;
293298
299+ var currentOrderManager = orderManager ;
300+ if ( currentOrderManager == null )
301+ {
302+ SendErrorResponse ( clientSocket , new MCPError
303+ {
304+ Code = MCPErrorCodes . InternalError ,
305+ Message = GetErrorMessage ( "INTERNAL_ERROR" , language )
306+ } , request . RequestId , DebugMode ) ;
307+ return ;
308+ }
309+
294310 if ( CommandHandlers . TryGetValue ( request . Command , out var commandHandler ) )
295311 {
296312 try
297313 {
298- var result = commandHandler ? . Invoke ( request . Params , orderManager ) ;
314+ var result = commandHandler ? . Invoke ( request . Params , currentOrderManager ) ;
299315 SendSuccessResponse ( clientSocket , result , request . RequestId , null , DebugMode ) ;
300316 }
301317 catch ( Exception ex )
@@ -319,7 +335,7 @@ async Task HandleClient(Socket clientSocket)
319335 {
320336 try
321337 {
322- var resultJson = queryHandler ? . Invoke ( request . Params , orderManager ) ;
338+ var resultJson = queryHandler ? . Invoke ( request . Params , currentOrderManager ) ;
323339 SendSuccessResponse ( clientSocket , null , request . RequestId , resultJson , DebugMode ) ;
324340 }
325341 catch ( Exception ex )
0 commit comments