chore: initial project import

backend/src/Monitor.Backend.Gateway/GatewayServiceCollectionExtensions.cs

@@ -0,0 +1,49 @@
+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)));
+    }
+}