webmatic/insight
webmatic/insight
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

web-proxy rewrite

Browse source
This commit is contained in:
Kevin Kai Berthold 2023-09-21 23:44:05 +02:00
parent 26c741ad03
commit 283fa1abc2
10 changed files with 63 additions and 134 deletions
Download patch file Download diff file Expand all files Collapse all files

20
src/Agent/Insight.Agent/Network/Handlers/ConsoleHandler.cs
View file

@ -10,24 +10,26 @@ public class ConsoleHandler : IMessageHandler<AgentSession>
{
public async ValueTask HandleAsync<TMessage>(AgentSession sender, TMessage message, CancellationToken cancellationToken) where TMessage : IMessage
{
if (message is ConsoleQueryRequest consoleQueryRequest)
if (message is Proxy<ConsoleQueryRequest> consoleQueryRequest)
{
await OnConsoleQueryRequestAsync(sender, consoleQueryRequest, cancellationToken);
}
}
private async ValueTask OnConsoleQueryRequestAsync(AgentSession sender, ConsoleQueryRequest consoleQueryRequest, CancellationToken cancellationToken)
private async ValueTask OnConsoleQueryRequestAsync(AgentSession sender, Proxy<ConsoleQueryRequest> consoleQueryRequest, CancellationToken cancellationToken)
{
var result = await QueryScriptAsync(consoleQueryRequest.Query);
var result = await QueryScriptAsync(consoleQueryRequest.Message.Query);
await sender.SendAsync(new ConsoleQuery
await sender.SendAsync(new Proxy<ConsoleQuery>
{
Id = consoleQueryRequest.Id,
RequestId = consoleQueryRequest.RequestId,
HostId = consoleQueryRequest.HostId,
Query = consoleQueryRequest.Query,
Data = result.Data,
Errors = result.Errors,
HadErrors = result.HadErrors
Message = new ConsoleQuery
{
Data = result.Data,
Errors = result.Errors,
HadErrors = result.HadErrors
}
}, cancellationToken);
}