50 lines
2.3 KiB
C#
50 lines
2.3 KiB
C#
using Fleet.Core;
|
|
using Fleet.Rpc;
|
|
using Fleet.Server;
|
|
using Microsoft.Extensions.DependencyInjection;
|
|
using Monitor.Backend.Application;
|
|
using Monitor.Protocol;
|
|
|
|
namespace Monitor.Backend.Gateway;
|
|
|
|
public static class GatewayServiceCollectionExtensions
|
|
{
|
|
/// <summary>
|
|
/// 在 Fleet.Server 之上接入 Monitor：期望状态（TaskSet）、RawChunk 收件箱、<c>monitor.*</c> 方法、注册挂钩和节点事件。
|
|
/// 需要事先注册 <see cref="IFleetStore"/> 和 <see cref="TimeProvider"/>，并在 <c>AddMonitorApplication</c> 之后调用。
|
|
/// </summary>
|
|
public static IServiceCollection AddMonitorGateway(this IServiceCollection services)
|
|
{
|
|
services.AddSingleton<IAgentGateway, FleetGateway>();
|
|
services.AddHostedService<ConnectionShutdown>();
|
|
services.AddFleetEnrollment();
|
|
services.AddFleetServer(options => options.NodeIdPrefix = AgentIds.Prefix)
|
|
.AddState<TaskSetStateProvider>()
|
|
.AddInbox<RawChunkInbox>()
|
|
.MapMethods(MapMonitorMethods);
|
|
return services;
|
|
}
|
|
|
|
private static void MapMonitorMethods(RpcRouter<NodeCallContext> router, IServiceProvider services)
|
|
{
|
|
router
|
|
.OnNotify(
|
|
MonitorMethods.Event,
|
|
MonitorJsonContext.Default.AgentEventParams,
|
|
(context, parameters, ct) => services.GetRequiredService<AgentEventService>().OnEventAsync(context.Node.Id, parameters, ct))
|
|
.OnNotify(
|
|
MonitorMethods.Host,
|
|
MonitorJsonContext.Default.HostSnapshot,
|
|
(context, snapshot, ct) => services.GetRequiredService<HostStatusService>().OnHostAsync(context.Node.Id, snapshot, ct))
|
|
.OnNotify(
|
|
MonitorMethods.TestStatus,
|
|
MonitorJsonContext.Default.TestStatusParams,
|
|
(context, parameters, ct) => services.GetRequiredService<TestService>().OnStatusAsync(context.Node.Id, parameters, ct))
|
|
.On(
|
|
MonitorMethods.ProbeAgents,
|
|
MonitorJsonContext.Default.ProbeAgentsParams,
|
|
MonitorJsonContext.Default.ProbeAgentsResult,
|
|
async (context, parameters, ct) => Result<ProbeAgentsResult>.Ok(await services.GetRequiredService<AgentService>().ProbeAgentsAsync(ct)));
|
|
}
|
|
}
|