product advisories, stella router improval, tests streghthening

src/Gateway/StellaOps.Gateway.WebService/Services/GatewayHostedService.cs

@@ -9,6 +9,7 @@ using StellaOps.Router.Common.Models;
 using StellaOps.Router.Gateway.OpenApi;
 using StellaOps.Router.Transport.Tcp;
 using StellaOps.Router.Transport.Tls;
+using StellaOps.Router.Transport.Messaging;
 
 namespace StellaOps.Gateway.WebService.Services;
 
@@ -16,6 +17,7 @@ public sealed class GatewayHostedService : IHostedService
 {
     private readonly TcpTransportServer _tcpServer;
     private readonly TlsTransportServer _tlsServer;
+    private readonly MessagingTransportServer? _messagingServer;
     private readonly IGlobalRoutingState _routingState;
     private readonly GatewayTransportClient _transportClient;
     private readonly IEffectiveClaimsStore _claimsStore;
@@ -26,6 +28,7 @@ public sealed class GatewayHostedService : IHostedService
     private readonly JsonSerializerOptions _jsonOptions;
     private bool _tcpEnabled;
     private bool _tlsEnabled;
+    private bool _messagingEnabled;
 
     public GatewayHostedService(
         TcpTransportServer tcpServer,
@@ -36,10 +39,12 @@ public sealed class GatewayHostedService : IHostedService
         IOptions<GatewayOptions> options,
         GatewayServiceStatus status,
         ILogger<GatewayHostedService> logger,
-        IRouterOpenApiDocumentCache? openApiCache = null)
+        IRouterOpenApiDocumentCache? openApiCache = null,
+        MessagingTransportServer? messagingServer = null)
     {
         _tcpServer = tcpServer;
         _tlsServer = tlsServer;
+        _messagingServer = messagingServer;
         _routingState = routingState;
         _transportClient = transportClient;
         _claimsStore = claimsStore;
@@ -59,8 +64,9 @@ public sealed class GatewayHostedService : IHostedService
         var options = _options.Value;
         _tcpEnabled = options.Transports.Tcp.Enabled;
         _tlsEnabled = options.Transports.Tls.Enabled;
+        _messagingEnabled = options.Transports.Messaging.Enabled && _messagingServer is not null;
 
-        if (!_tcpEnabled && !_tlsEnabled)
+        if (!_tcpEnabled && !_tlsEnabled && !_messagingEnabled)
         {
             _logger.LogWarning("No transports enabled; gateway will not accept microservice connections.");
             _status.MarkStarted();
@@ -84,6 +90,17 @@ public sealed class GatewayHostedService : IHostedService
             _logger.LogInformation("TLS transport started on port {Port}", options.Transports.Tls.Port);
         }
 
+        if (_messagingEnabled && _messagingServer is not null)
+        {
+            _messagingServer.OnHelloReceived += HandleMessagingHello;
+            _messagingServer.OnHeartbeatReceived += HandleMessagingHeartbeat;
+            _messagingServer.OnResponseReceived += HandleMessagingResponse;
+            _messagingServer.OnConnectionClosed += HandleMessagingDisconnection;
+            await _messagingServer.StartAsync(cancellationToken);
+            _logger.LogInformation("Messaging transport started (Valkey connection: {Connection})",
+                options.Transports.Messaging.ConnectionString);
+        }
+
         _status.MarkStarted();
         _status.MarkReady();
     }
@@ -110,6 +127,15 @@ public sealed class GatewayHostedService : IHostedService
             _tlsServer.OnFrame -= HandleTlsFrame;
             _tlsServer.OnDisconnection -= HandleTlsDisconnection;
         }
+
+        if (_messagingEnabled && _messagingServer is not null)
+        {
+            await _messagingServer.StopAsync(cancellationToken);
+            _messagingServer.OnHelloReceived -= HandleMessagingHello;
+            _messagingServer.OnHeartbeatReceived -= HandleMessagingHeartbeat;
+            _messagingServer.OnResponseReceived -= HandleMessagingResponse;
+            _messagingServer.OnConnectionClosed -= HandleMessagingDisconnection;
+        }
     }
 
     private void HandleTcpFrame(string connectionId, Frame frame)
@@ -438,8 +464,55 @@ public sealed class GatewayHostedService : IHostedService
         {
             _tlsServer.GetConnection(connectionId)?.Close();
         }
+
+        // Messaging transport connections are managed by the queue system
+        // and do not support explicit close operations
     }
 
+    #region Messaging Transport Event Handlers
+
+    private Task HandleMessagingHello(ConnectionState state, HelloPayload payload)
+    {
+        // The MessagingTransportServer already built the ConnectionState with TransportType.Messaging
+        // We need to add it to the routing state and update the claims store
+        _routingState.AddConnection(state);
+        _claimsStore.UpdateFromMicroservice(payload.Instance.ServiceName, payload.Endpoints);
+        _openApiCache?.Invalidate();
+
+        _logger.LogInformation(
+            "Messaging connection registered: {ConnectionId} service={ServiceName} version={Version}",
+            state.ConnectionId,
+            state.Instance.ServiceName,
+            state.Instance.Version);
+
+        return Task.CompletedTask;
+    }
+
+    private Task HandleMessagingHeartbeat(ConnectionState state, HeartbeatPayload payload)
+    {
+        _routingState.UpdateConnection(state.ConnectionId, conn =>
+        {
+            conn.LastHeartbeatUtc = DateTime.UtcNow;
+            conn.Status = payload.Status;
+        });
+
+        return Task.CompletedTask;
+    }
+
+    private Task HandleMessagingResponse(ConnectionState state, Frame frame)
+    {
+        _transportClient.HandleResponseFrame(frame);
+        return Task.CompletedTask;
+    }
+
+    private Task HandleMessagingDisconnection(string connectionId)
+    {
+        HandleDisconnect(connectionId);
+        return Task.CompletedTask;
+    }
+
+    #endregion
+
     private sealed class EndpointKeyComparer : IEqualityComparer<(string Method, string Path)>
     {
         public bool Equals((string Method, string Path) x, (string Method, string Path) y)

src/Gateway/StellaOps.Gateway.WebService/Services/GatewayTransportClient.cs

@@ -6,6 +6,7 @@ using StellaOps.Router.Common.Enums;
 using StellaOps.Router.Common.Models;
 using StellaOps.Router.Transport.Tcp;
 using StellaOps.Router.Transport.Tls;
+using StellaOps.Router.Transport.Messaging;
 
 namespace StellaOps.Gateway.WebService.Services;
 
@@ -13,6 +14,7 @@ public sealed class GatewayTransportClient : ITransportClient
 {
     private readonly TcpTransportServer _tcpServer;
     private readonly TlsTransportServer _tlsServer;
+    private readonly MessagingTransportServer? _messagingServer;
     private readonly ILogger<GatewayTransportClient> _logger;
     private readonly ConcurrentDictionary<string, TaskCompletionSource<Frame>> _pendingRequests = new();
     private readonly ConcurrentDictionary<string, Channel<Frame>> _streamingResponses = new();
@@ -20,10 +22,12 @@ public sealed class GatewayTransportClient : ITransportClient
     public GatewayTransportClient(
         TcpTransportServer tcpServer,
         TlsTransportServer tlsServer,
-        ILogger<GatewayTransportClient> logger)
+        ILogger<GatewayTransportClient> logger,
+        MessagingTransportServer? messagingServer = null)
     {
         _tcpServer = tcpServer;
         _tlsServer = tlsServer;
+        _messagingServer = messagingServer;
         _logger = logger;
     }
 
@@ -147,6 +151,13 @@ public sealed class GatewayTransportClient : ITransportClient
             case TransportType.Certificate:
                 await _tlsServer.SendFrameAsync(connection.ConnectionId, frame, cancellationToken);
                 break;
+            case TransportType.Messaging:
+                if (_messagingServer is null)
+                {
+                    throw new InvalidOperationException("Messaging transport is not enabled");
+                }
+                await _messagingServer.SendToMicroserviceAsync(connection.ConnectionId, frame, cancellationToken);
+                break;
             default:
                 throw new NotSupportedException($"Transport type {connection.TransportType} is not supported by the gateway.");
         }