micialware/mROA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Abstraction of representation module rewrite

Browse Source
This commit is contained in:
micialware
2025-05-02 13:34:40 +03:00
parent 4e9b595620
commit e6466b133b
6 changed files with 104 additions and 105 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mROA/Abstract/ISerialisationModule.cs → mROA/Abstract/IRepresentationModule.cs
+5 -7
View File
@@ -1,4 +1,5 @@
using System;
using System.Collections.Generic;
using System.Threading;
using System.Threading.Tasks;
using mROA.Implementation;
@@ -9,16 +10,13 @@ namespace mROA.Abstract
{
int Id { get; }
Task<T> GetMessageAsync<T>(Guid? requestId = null, EMessageType? messageType = null,
CancellationToken token = default);
Task<(object parced, EMessageType originalType)> GetSingle(Predicate<NetworkMessageHeader> rule, CancellationToken token,
params Func<NetworkMessageHeader, Type?>[] converter);
T GetMessage<T>(Guid? requestId = null, EMessageType? messageType = null);
Task<byte[]> GetRawMessage(Guid? requestId = null, EMessageType? messageType = null,
CancellationToken token = default);
IAsyncEnumerable<(object parced, EMessageType originalType)> GetStream(Predicate<NetworkMessageHeader> rule, CancellationToken token,
params Func<NetworkMessageHeader, Type?>[] converter);
Task PostCallMessageAsync<T>(Guid id, EMessageType eMessageType, T payload) where T : notnull;
Task PostCallMessageAsync(Guid id, EMessageType eMessageType, object payload, Type payloadType);
void PostCallMessage<T>(Guid id, EMessageType eMessageType, T payload) where T : notnull;
void PostCallMessage(Guid id, EMessageType eMessageType, object payload, Type payloadType);
}
mROA/Implementation/Frontend/NetworkFrontendBridge.cs
+2
View File
@@ -1,4 +1,6 @@
using System;
using System.Collections.Generic;
using System.Linq;
using System.Net;
using System.Net.Sockets;
using System.Threading;
mROA/Implementation/Frontend/RequestExtractor.cs
+45 -56
View File
@@ -4,7 +4,6 @@ using System.Threading;
using System.Threading.Tasks;
using mROA.Abstract;
using mROA.Implementation.Backend;
using mROA.Implementation.CommandExecution;
// ReSharper disable MethodHasAsyncOverload
@@ -45,22 +44,32 @@ namespace mROA.Implementation.Frontend
}
}
public Task StartExtraction()
public async Task StartExtraction()
{
return Task.Run(() =>
{
ThrowIfNotInjected();
var multiClientOwnershipRepository =
TransmissionConfig.OwnershipRepository as MultiClientOwnershipRepository;
multiClientOwnershipRepository?.RegisterOwnership(_representationModule.Id);
ThrowIfNotInjected();
var multiClientOwnershipRepository =
TransmissionConfig.OwnershipRepository as MultiClientOwnershipRepository;
multiClientOwnershipRepository?.RegisterOwnership(_representationModule!.Id);
try
{
try
{
#if TRACE
var sw = new Stopwatch();
#endif
while (true)
{
var streamTokenSource = new CancellationTokenSource();
var query = _representationModule!.GetStream(m =>
m.MessageType is EMessageType.CallRequest or EMessageType.CancelRequest
or EMessageType.EventRequest or EMessageType.ClientDisconnect, streamTokenSource.Token,
m => m.MessageType == EMessageType.CallRequest ? typeof(DefaultCallRequest) : null,
m => m.MessageType == EMessageType.CancelRequest ? typeof(CancelRequest) : null,
m => m.MessageType == EMessageType.EventRequest ? typeof(DefaultCallRequest) : null,
m => m.MessageType == EMessageType.ClientDisconnect ? typeof(ClientDisconnect) : null);
await foreach (var command in query)
{
#if TRACE
Console.WriteLine("Waiting for request...");
if (sw.IsRunning)
@@ -69,50 +78,34 @@ namespace mROA.Implementation.Frontend
Console.WriteLine($"Request handling took {Math.Round(sw.Elapsed.TotalMilliseconds * 1000.0)} microseconds.");
}
#endif
var tokenSource = new CancellationTokenSource();
var token = tokenSource.Token;
var defaultRequest =
_representationModule!.GetMessageAsync<DefaultCallRequest>(
messageType: EMessageType.CallRequest, token: token);
var cancelRequest =
_representationModule!.GetMessageAsync<CancelRequest>(
messageType: EMessageType.CancelRequest, token: token);
var eventRequest =
_representationModule!.GetMessageAsync<DefaultCallRequest>(
messageType: EMessageType.EventRequest, token: token);
var disconnectRequest =
_representationModule!.GetMessageAsync<ClientDisconnect>(
messageType: EMessageType.ClientDisconnect, token:token);
Task.WaitAny(defaultRequest, cancelRequest, eventRequest, disconnectRequest);
#if TRACE
Console.WriteLine("Request received");
sw.Restart();
#endif
if (cancelRequest.IsCompleted)
{
#if TRACE
Console.WriteLine("Cancelling request");
#endif
HandleCancelRequest(tokenSource, cancelRequest.Result);
}
else if (defaultRequest.IsCompleted)
{
HandleCallRequest(tokenSource, defaultRequest.Result);
}
else if(eventRequest.IsCompleted)
{
HandleEventRequest(tokenSource, eventRequest.Result);
}else if (disconnectRequest.IsCompleted)
{
switch (command.originalType)
{
case EMessageType.CallRequest:
HandleCallRequest((command.parced as DefaultCallRequest)!);
break;
}
case EMessageType.ClientDisconnect:
return;
case EMessageType.EventRequest:
HandleEventRequest((command.parced as DefaultCallRequest)!);
break;
case EMessageType.CancelRequest:
HandleCancelRequest((command.parced as CancelRequest)!);
break;
default:
continue;
}
}
catch
{
multiClientOwnershipRepository?.FreeOwnership();
}
});
}
catch
{
multiClientOwnershipRepository?.FreeOwnership();
}
}
private void ThrowIfNotInjected()
@@ -129,16 +122,13 @@ namespace mROA.Implementation.Frontend
throw new NullReferenceException("Method repository is null.");
}
private void HandleCancelRequest(CancellationTokenSource tokenSource, CancelRequest req)
private void HandleCancelRequest(CancelRequest req)
{
tokenSource.Cancel();
_executeModule!.Execute(req, _realContextRepository!, _representationModule!);
}
private void HandleCallRequest(CancellationTokenSource tokenSource, DefaultCallRequest request)
private void HandleCallRequest(DefaultCallRequest request)
{
tokenSource.Cancel();
var result = _executeModule!.Execute(request, _realContextRepository!, _representationModule!);
var resultType = result.MessageType;
@@ -151,9 +141,8 @@ namespace mROA.Implementation.Frontend
_representationModule!.PostCallMessage(request.Id, resultType, result, result.GetType());
}
private void HandleEventRequest(CancellationTokenSource tokenSource, DefaultCallRequest request)
private void HandleEventRequest(DefaultCallRequest request)
{
tokenSource.Cancel();
_executeModule!.Execute(request, _remoteContextRepository!, _representationModule!);
}
}
mROA/Implementation/NetworkMessageHeader.cs
+5
View File
@@ -1,6 +1,7 @@
using System;
using System.Text.Json.Serialization;
using mROA.Abstract;
using mROA.Implementation.Attributes;
// ReSharper disable UnusedMember.Global
@@ -45,5 +46,9 @@ namespace mROA.Implementation
public EMessageType MessageType { get; set; }
public byte[] Data { get; set; }
[JsonIgnore]
[SerializationIgnore]
public object Parced { get; set; }
}
}
mROA/Implementation/RemoteObjectBase.cs
+34 -40
View File
@@ -10,7 +10,7 @@ namespace mROA.Implementation
{
public abstract class RemoteObjectBase : IDisposable
{
protected bool Equals(RemoteObjectBase other)
public bool Equals(RemoteObjectBase other)
{
return _identifier.Equals(other._identifier);
}
@@ -60,40 +60,36 @@ namespace mROA.Implementation
var localTokenSource = new CancellationTokenSource();
var successResponse =
_representationModule.GetMessageAsync<FinalCommandExecution<T>>(request.Id,
EMessageType.FinishedCommandExecution,
localTokenSource.Token);
var errorResponse =
_representationModule.GetMessageAsync<ExceptionCommandExecution>(requestId: request.Id,
EMessageType.ExceptionCommandExecution, localTokenSource.Token);
var responseRequestTask = _representationModule.GetSingle(
m => m.MessageType is EMessageType.FinishedCommandExecution or EMessageType.ExceptionCommandExecution,
localTokenSource.Token,
m => m.MessageType is EMessageType.FinishedCommandExecution ? typeof(FinalCommandExecution<T>) : null,
m => m.MessageType is EMessageType.ExceptionCommandExecution
? typeof(ExceptionCommandExecution)
: null);
cancellationToken.Register(async () =>
cancellationToken.Register(() =>
{
#if TRACE
Console.WriteLine("Cancelling task");
#endif
await _representationModule.PostCallMessageAsync(request.Id, EMessageType.CancelRequest,
_representationModule.PostCallMessageAsync(request.Id, EMessageType.CancelRequest,
new CancelRequest
{
Id = request.Id
});
localTokenSource.Cancel();
}).ContinueWith(_ => localTokenSource.Cancel());
});
Task.WaitAny(new Task[]
{
successResponse, errorResponse
}, cancellationToken);
var response = await responseRequestTask;
if (successResponse.IsCompletedSuccessfully)
if (response.parced is FinalCommandExecution<T> successResponse)
{
localTokenSource.Cancel();
return successResponse.Result.Result!;
return successResponse.Result!;
}
localTokenSource.Cancel();
throw errorResponse.Result.GetException();
throw (response.parced as ExceptionCommandExecution)!.GetException();
}
protected async Task CallAsync(int methodId, object?[]? parameters = null,
@@ -107,40 +103,38 @@ namespace mROA.Implementation
var localTokenSource = new CancellationTokenSource();
var successResponse =
_representationModule.GetMessageAsync<FinalCommandExecution>(request.Id,
EMessageType.FinishedCommandExecution,
localTokenSource.Token);
var errorResponse =
_representationModule.GetMessageAsync<ExceptionCommandExecution>(requestId: request.Id,
EMessageType.ExceptionCommandExecution, localTokenSource.Token);
var responceRequestTask = _representationModule.GetSingle(
m => m.MessageType is EMessageType.FinishedCommandExecution or EMessageType.ExceptionCommandExecution,
localTokenSource.Token,
m => m.MessageType is EMessageType.FinishedCommandExecution ? typeof(FinalCommandExecution) : null,
m => m.MessageType is EMessageType.ExceptionCommandExecution
? typeof(ExceptionCommandExecution)
: null);
cancellationToken.Register(async () =>
cancellationToken.Register(() =>
{
#if TRACE
Console.WriteLine("Cancelling task");
#endif
await _representationModule.PostCallMessageAsync(request.Id, EMessageType.CancelRequest,
_representationModule.PostCallMessageAsync(request.Id, EMessageType.CancelRequest,
new CancelRequest
{
Id = request.Id
});
localTokenSource.Cancel();
}).ContinueWith(_ => localTokenSource.Cancel());
});
Task.WaitAny(new Task[]
{
errorResponse, successResponse
}, cancellationToken);
var responseRequest = await responceRequestTask;
#if TRACE
Console.WriteLine($"Handling message");
#endif
if (successResponse.IsCompletedSuccessfully)
return;
if (errorResponse.IsCompletedSuccessfully)
throw errorResponse.Result.GetException();
switch (responseRequest.originalType)
{
case EMessageType.FinishedCommandExecution:
return;
case EMessageType.ExceptionCommandExecution:
throw (responseRequest.parced as ExceptionCommandExecution)!.GetException();
}
}
public override string ToString()
mROA/Implementation/RepresentationModule.cs
+11
View File
@@ -1,4 +1,5 @@
using System;
using System.Collections.Generic;
using System.Threading;
using System.Threading.Tasks;
using mROA.Abstract;
@@ -28,6 +29,16 @@ namespace mROA.Implementation
public int Id => (_interaction ?? throw new NullReferenceException("Interaction is not initialized"))
.ConnectionId;
public Task<(object parced, EMessageType originalType)> GetSingle(Predicate<NetworkMessageHeader> rule, CancellationToken token, params Func<NetworkMessageHeader, Type?>[] converter)
{
throw new NotImplementedException();
}
public IAsyncEnumerable<(object parced, EMessageType originalType)> GetStream(Predicate<NetworkMessageHeader> rule, CancellationToken token, params Func<NetworkMessageHeader, Type?>[] converter)
{
throw new NotImplementedException();
}
public async Task<T> GetMessageAsync<T>(Guid? requestId, EMessageType? messageType,
CancellationToken token = default)
{