опять добавлены токены в обработку сообщений

mROA/Abstract/ISerialisationModule.cs

@@ -1,5 +1,7 @@
 using System;
+using System.Threading;
 using System.Threading.Tasks;
+using System.Windows.Input;
 using mROA.Implementation;
 using mROA.Implementation.CommandExecution;
 
@@ -22,9 +24,9 @@ namespace mROA.Abstract
     public interface IRepresentationModule : IInjectableModule
     {
         int Id { get; }
-        Task<T> GetMessageAsync<T>(Guid? requestId = null, MessageType? messageType = null);
+        Task<T> GetMessageAsync<T>(Guid? requestId = null, MessageType? messageType = null, CancellationToken token = default);
         T GetMessage<T>(Guid? requestId = null, MessageType? messageType = null);
-        Task<byte[]> GetRawMessage(Guid? requestId = null, MessageType? messageType = null);
+        Task<byte[]> GetRawMessage(Guid? requestId = null, MessageType? messageType = null, CancellationToken token = default);
     
         Task PostCallMessageAsync<T>(Guid id, MessageType messageType, T payload) where T : notnull;
         Task PostCallMessageAsync(Guid id, MessageType messageType, object payload, Type payloadType);

mROA/Implementation/Frontend/RequestExtractor.cs

@@ -85,7 +85,7 @@ namespace mROA.Implementation.Frontend
             }
             catch
             {
-                multiClientOwnershipRepository?.RegisterOwnership(_representationModule.Id);
+                multiClientOwnershipRepository?.FreeOwnership();
             }
         }
     }

mROA/Implementation/NextGenerationInteractionModule.cs

@@ -34,7 +34,7 @@ namespace mROA.Implementation
         public Task<NetworkMessage> GetNextMessageReceiving()
         {
             if (_currentReceiving != null) return _currentReceiving;
-            _currentReceiving = Task.Run(GetNextMessage);
+            _currentReceiving = Task.Run(async () => await GetNextMessage());
             return _currentReceiving;
 
         }
@@ -87,7 +87,7 @@ namespace mROA.Implementation
 
             var message = _serialization.Deserialize<NetworkMessage>(localSpan.Span);
             _messageBuffer.Add(message!);
-            _currentReceiving = GetNextMessage();
+            _currentReceiving = Task.Run(async () => await GetNextMessage());
         
             return message!;
         }

mROA/Implementation/RemoteObjectBase.cs

@@ -1,4 +1,5 @@
-using System.Threading.Tasks;
+using System.Threading;
+using System.Threading.Tasks;
 using mROA.Abstract;
 using mROA.Implementation.CommandExecution;
 
@@ -26,16 +27,22 @@ namespace mROA.Implementation
                 { CommandId = methodId, ObjectId = _id, Parameter = parameter, ParameterType = parameter?.GetType() };
             await _representationModule.PostCallMessageAsync(request.Id, MessageType.CallRequest, request);
 
+            var localTokenSource = new CancellationTokenSource();
+
             var successResponse =
-                _representationModule.GetMessageAsync<FinalCommandExecution<T>>(
-                    messageType: MessageType.FinishedCommandExecution, requestId: request.Id);
+                _representationModule.GetMessageAsync<FinalCommandExecution<T>>(request.Id,
+                    MessageType.FinishedCommandExecution,
+                    localTokenSource.Token);
             var errorResponse =
-                _representationModule.GetMessageAsync<ExceptionCommandExecution>(
-                    messageType: MessageType.ExceptionCommandExecution, requestId: request.Id);
+                _representationModule.GetMessageAsync<ExceptionCommandExecution>(requestId: request.Id,
+                    MessageType.ExceptionCommandExecution, localTokenSource.Token);
+
             Task.WaitAny(successResponse, errorResponse);
 
             if (successResponse.IsCompletedSuccessfully)
+            {
                 return successResponse.Result.Result!;
+            }
 
             throw errorResponse.Result.GetException();
         }

mROA/Implementation/RepresentationModule.cs

@@ -1,4 +1,5 @@
 using System;
+using System.Threading;
 using System.Threading.Tasks;
 using mROA.Abstract;
 
@@ -24,12 +25,12 @@ namespace mROA.Implementation
 
         public int Id => (_interaction ?? throw new NullReferenceException("Interaction is not initialized")).ConnectionId;
 
-        public async Task<T> GetMessageAsync<T>(Guid? requestId, MessageType? messageType)
+        public async Task<T> GetMessageAsync<T>(Guid? requestId, MessageType? messageType, CancellationToken token = default)
         {
             if (_serialization == null)
                 throw new NullReferenceException("Serialization toolkit is not initialized");
         
-            return _serialization.Deserialize<T>(await GetRawMessage(requestId, messageType))!;
+            return _serialization.Deserialize<T>(await GetRawMessage(requestId, messageType, token))!;
         }
 
         public T GetMessage<T>(Guid? requestId = null, MessageType? messageType = null)
@@ -40,7 +41,7 @@ namespace mROA.Implementation
             return _serialization.Deserialize<T>(GetRawMessage(requestId, messageType).GetAwaiter().GetResult())!;
         }
 
-        public async Task<byte[]> GetRawMessage(Guid? requestId = null, MessageType? messageType = null)
+        public async Task<byte[]> GetRawMessage(Guid? requestId = null, MessageType? messageType = null, CancellationToken token = default)
         {
             if (_interaction == null)
                 throw new NullReferenceException("Interaction toolkit is not initialized");
@@ -52,7 +53,7 @@ namespace mROA.Implementation
         
             if (fromBuffer == null)
             {
-                while (true)
+                while (token.IsCancellationRequested == false)
                 {
                     var message = await _interaction.GetNextMessageReceiving();
                     if ((requestId is not null && message.Id != requestId) ||