Fix

Assets/AvvyLand/Scripts/ECS/Systems/InitSystemGroup/ResetAllSystem.cs

@@ -57,13 +57,6 @@ namespace AvvyLand.Scripts.ECS.Systems.InitSystemGroup
                         var client = EntityManager.GetSharedComponentData<GridWebSocketClientComponent>(gridEntity);
                         client.Client.Disconnect();
                     }
-                    
-                    if (HasSingleton<UCSWebSocketClientComponent>())
-                    {
-                        var gridEntity = GetSingletonEntity<UCSWebSocketClientComponent>();
-                        var ucsClient = EntityManager.GetSharedComponentData<UCSWebSocketClientComponent>(gridEntity);
-                        ucsClient.Client.Disconnect();
-                    }
 
 
                     if (HasSingleton<SnapShotSingletonComponent>())

Assets/AvvyLand/Scripts/ECS/Systems/RebootSystems/ClearArVoxelGridSystem.cs

 using AvvyLand.Scripts.ECS.Components.Grid;
 using AvvyLand.Scripts.ECS.Components.NetworkLoop;
 using AvvyLand.Scripts.ECS.Components.RebootComponents;
+using AvvyLand.Scripts.ECS.Systems.Grpc;
+using AvvyLand.Scripts.ECS.Systems.InitSystemGroup;
 using AvvyLand.Scripts.ECS.Systems.NetworkClient;
 using AvvyLand.Scripts.ECS.Systems.UCSGrpc;
+using Ucs;
 using Unity.Entities;
+using UnityEngine;
 
 namespace AvvyLand.Scripts.ECS.Systems.RebootSystems
 {
@@ -19,7 +23,6 @@ namespace AvvyLand.Scripts.ECS.Systems.RebootSystems
         protected override void OnCreate()
         {
             RequireForUpdate(GetEntityQuery(typeof(ArVoxelGridComponentSingleton)));
-            RequireForUpdate(GetEntityQuery(typeof(UCSWebSocketClientComponent)));
         }
 
         protected override void OnUpdate()
@@ -33,14 +36,16 @@ namespace AvvyLand.Scripts.ECS.Systems.RebootSystems
                     
                     var client = EntityManager.GetSharedComponentData<GridWebSocketClientComponent>(gridEntity);
                     client.Client.Disconnect();
-                    
-                    var ucsWebSocketEntity = GetSingletonEntity<UCSWebSocketClientComponent>();
-                    var ucsClient = EntityManager.GetSharedComponentData<UCSWebSocketClientComponent>(ucsWebSocketEntity);
-                    ucsClient.Client.DisconnectAsync(() =>
+
+
+                    if (HasSingleton<GrpcClientSharedComponent<UCS.UCSClient>>())
                     {
-                        EntityManager.AddComponent<NetworkNotConnectedTag>(ucsWebSocketEntity);
-                    });
+                        var ucsGrpcEntity = GetSingletonEntity<GrpcClientSharedComponent<UCS.UCSClient>>();
+                        var ucsGrpc = EntityManager.GetSharedComponentData<GrpcClientSharedComponent<UCS.UCSClient>>(ucsGrpcEntity);
+                        ucsGrpc.GrpcClient.Disconnect();
+                    }
                     
+                    Debug.Log("ClearArVoxelGridSystem");
                     EntityManager.AddComponent<RemoveArVoxelGridComponentSingleton>(gridEntity);
                     EntityManager.DestroyEntity(e);                    
                 }).Run();