feat: websocket 리팩토링

Author: ImGdevel

ProjectVG.Api/Middleware/WebSocketMiddleware.cs

@@ -0,0 +1,98 @@
+using Microsoft.AspNetCore.Http;
+using System.Net.WebSockets;
+using ProjectVG.Application.Services.WebSocket;
+using ProjectVG.Application.Services.Session;
+using ProjectVG.Common.Models.Session;
+
+namespace ProjectVG.Api.Middleware
+{
+    public class WebSocketMiddleware
+    {
+        private readonly RequestDelegate _next;
+        private readonly ILogger<WebSocketMiddleware> _logger;
+        private readonly IWebSocketManager _webSocketService;
+        private readonly IConnectionRegistry _connectionRegistry;
+        private readonly IClientConnectionFactory _connectionFactory;
+
+        public WebSocketMiddleware(
+            RequestDelegate next,
+            ILogger<WebSocketMiddleware> logger,
+            IWebSocketManager webSocketService,
+            IConnectionRegistry connectionRegistry,
+            IClientConnectionFactory connectionFactory)
+        {
+            _next = next;
+            _logger = logger;
+            _webSocketService = webSocketService;
+            _connectionRegistry = connectionRegistry;
+            _connectionFactory = connectionFactory;
+        }
+
+        public async Task InvokeAsync(HttpContext context)
+        {
+            if (context.Request.Path == "/ws") {
+                if (!context.WebSockets.IsWebSocketRequest) {
+                    _logger.LogWarning("WebSocket 요청이 아님");
+                    context.Response.StatusCode = 400;
+                    return;
+                }
+
+                var sessionId = context.Request.Query["sessionId"].ToString();
+                var socket = await context.WebSockets.AcceptWebSocketAsync();
+
+                // 1. 세션 ID 생성 (연결 등록 없이)
+                var actualSessionId = string.IsNullOrWhiteSpace(sessionId) ?
+                    $"session_{DateTime.UtcNow.Ticks}_{Guid.NewGuid().ToString("N")[..8]}" : sessionId;
+
+                // 2. 기존 연결이 있다면 정리
+                if (_connectionRegistry.TryGet(actualSessionId, out var existingConnection) && existingConnection != null) {
+                    _logger.LogInformation("기존 연결을 정리합니다: {SessionId}", actualSessionId);
+                    await _webSocketService.DisconnectAsync(actualSessionId);
+                }
+
+                // 3. 연결 생성 및 등록
+                var connection = _connectionFactory.Create(actualSessionId, socket, userId: null);
+                _connectionRegistry.Register(actualSessionId, connection);
+
+                // 4. 세션 생성 (이제 연결이 등록된 상태)
+                await _webSocketService.ConnectAsync(actualSessionId);
+
+                await HandleWebSocketConnection(socket, actualSessionId);
+                return;
+            }
+
+            await _next(context);
+        }
+
+        private async Task HandleWebSocketConnection(WebSocket socket, string sessionId)
+        {
+            var buffer = new byte[1024];
+            try {
+                while (socket.State == WebSocketState.Open) {
+                    var result = await socket.ReceiveAsync(new ArraySegment<byte>(buffer), CancellationToken.None);
+
+                    if (result.MessageType == WebSocketMessageType.Close) {
+                        _logger.LogInformation("WebSocket 연결 종료 요청: {SessionId}", sessionId);
+                        break;
+                    }
+                    else if (result.MessageType == WebSocketMessageType.Text) {
+                        var message = System.Text.Encoding.UTF8.GetString(buffer, 0, result.Count);
+                        await _webSocketService.HandleMessageAsync(sessionId, message);
+                    }
+                    else if (result.MessageType == WebSocketMessageType.Binary) {
+                        var data = new byte[result.Count];
+                        Array.Copy(buffer, data, result.Count);
+                        await _webSocketService.HandleBinaryMessageAsync(sessionId, data);
+                    }
+                }
+            }
+            catch (Exception ex) {
+                _logger.LogError(ex, "WebSocket 연결 유지 중 오류: {SessionId}", sessionId);
+            }
+            finally {
+                _logger.LogInformation("WebSocket 연결 해제: {SessionId}", sessionId);
+                await _webSocketService.DisconnectAsync(sessionId);
+            }
+        }
+    }
+}

ProjectVG.Api/Program.cs

@@ -1,13 +1,10 @@
-using ProjectVG.Application.Middlewares;
 using Microsoft.AspNetCore.Authentication.Negotiate;
 using ProjectVG.Infrastructure.Integrations.LLMClient;
 using ProjectVG.Infrastructure.Integrations.MemoryClient;
 using ProjectVG.Application.Services.Chat;
 using ProjectVG.Application.Services;
-using ProjectVG.Application.Services.Character;
 using ProjectVG.Application.Services.Conversation;
 using ProjectVG.Application.Services.Session;
-using ProjectVG.Application.Services.User;
 using ProjectVG.Infrastructure.Persistence.Session;
 using ProjectVG.Infrastructure.Integrations.TextToSpeechClient;
 
@@ -16,7 +13,7 @@
 using ProjectVG.Api.Middleware;
 using ProjectVG.Api.Filters;
 using ProjectVG.Infrastructure.Realtime.WebSocketConnection;
-using ProjectVG.Application.Services.Messaging;
+
 using ProjectVG.Common.Models.Session;
 using ProjectVG.Infrastructure.Persistence.EfCore;
 using Microsoft.EntityFrameworkCore;
@@ -96,7 +93,6 @@
 
 builder.Services.AddSingleton<IConnectionRegistry, ConnectionRegistry>();
 builder.Services.AddSingleton<IClientConnectionFactory, WebSocketClientConnectionFactory>();
-builder.Services.AddSingleton<IMessageBroker, MessageBroker>();
 
 // Infrastructure Repositories 
 builder.Services.AddScoped<ICharacterRepository, SqlServerCharacterRepository>();
@@ -138,7 +134,7 @@
 
 app.UseWebSockets();
 
-// WebSocket 미들웨어를 특정 경로에만 적용
+// WebSocket 미들웨어 등록
 app.UseMiddleware<WebSocketMiddleware>();
 
 app.Use(async (ctx, next) => {

ProjectVG.Application/Middlewares/WebSocketMiddleware.cs

@@ -5,6 +5,7 @@
 using ProjectVG.Application.Services.Chat.Preprocessors;
 using ProjectVG.Application.Services.Chat.Processors;
 using ProjectVG.Application.Services.Chat.Validators;
+using ProjectVG.Application.Services.WebSocket;
 
 namespace ProjectVG.Application.Services
 {
@@ -28,6 +29,8 @@ public static IServiceCollection AddApplicationServices(this IServiceCollection
             services.AddScoped<UserInputActionProcessor>();
             services.AddScoped<MemoryContextPreprocessor>();
 
+            services.AddScoped<IWebSocketManager, WebSocketManager>();
+            
             return services;
         }
     }

ProjectVG.Application/Services/ApplicationServiceCollectionExtensions.cs

@@ -1,7 +1,7 @@
 using ProjectVG.Application.Models.Chat;
 using ProjectVG.Application.Models.WebSocket;
 using ProjectVG.Application.Services.Conversation;
-using ProjectVG.Application.Services.Messaging;
+using ProjectVG.Application.Services.WebSocket;
 using ProjectVG.Infrastructure.Integrations.MemoryClient;
 using ProjectVG.Domain.Enums;
 
@@ -12,18 +12,18 @@ public class ChatResultProcessor
         private readonly ILogger<ChatResultProcessor> _logger;
         private readonly IConversationService _conversationService;
         private readonly IMemoryClient _memoryClient;
-        private readonly IMessageBroker _messageBroker;
+        private readonly IWebSocketManager _webSocketService;
 
         public ChatResultProcessor(
             ILogger<ChatResultProcessor> logger,
             IConversationService conversationService,
             IMemoryClient memoryClient,
-            IMessageBroker messageBroker)
+            IWebSocketManager webSocketService)
         {
             _logger = logger;
             _conversationService = conversationService;
             _memoryClient = memoryClient;
-            _messageBroker = messageBroker;
+            _webSocketService = webSocketService;
         }
 
         public async Task PersistResultsAsync(ChatPreprocessContext context, ChatProcessResult result)
@@ -51,7 +51,7 @@ public async Task SendResultsAsync(ChatPreprocessContext context, ChatProcessRes
                 integratedMessage.SetAudioData(segment.AudioData);
 
                 var wsMessage = new WebSocketMessage("chat", integratedMessage);
-                await _messageBroker.SendWebSocketMessageAsync(context.SessionId, wsMessage);
+                await _webSocketService.SendAsync(context.SessionId, wsMessage);
             }
 
             _logger.LogDebug("채팅 결과 전송 완료: 세션 {SessionId}, 세그먼트 {SegmentCount}개",