feat: 웹소캣 데이터 전송 구조 변경

ImGdevel · ImGdevel · commit 5c5f3110a75e · 2025-07-29T18:13:52.000+09:00
type : data 방식으로 수정
웹소캣 데이터 형태및 이벤트에 따라 데이터가 달라질 수 있고 좀더 유연한 type 검사를 목적으로 함함
diff --git a/ProjectVG.Api/Controllers/HomeController.cs b/ProjectVG.Api/Controllers/HomeController.cs
@@ -17,5 +17,18 @@ public IActionResult GetConfig()
                 version = "1.0.0"
             });
         }
+        
+        [HttpGet("health")]
+        [HttpGet("/health")]
+        public IActionResult Health()
+        {
+            return Ok(new
+            {
+                status = "healthy",
+                timestamp = DateTime.UtcNow,
+                version = "1.0.0",
+                uptime = Environment.TickCount / 1000.0
+            });
+        }
     }
 } 
diff --git a/ProjectVG.Application/Middlewares/WebSocketMiddleware.cs b/ProjectVG.Application/Middlewares/WebSocketMiddleware.cs
@@ -4,6 +4,7 @@
 using Microsoft.AspNetCore.Http;
 using Microsoft.Extensions.DependencyInjection;
 using ProjectVG.Application.Services.Session;
+using ProjectVG.Application.Models.Chat;
 
 namespace ProjectVG.Application.Middlewares
 {
@@ -53,7 +54,7 @@ public async Task InvokeAsync(HttpContext context)
                 await sessionService.RegisterSessionAsync(sessionId, socket);
                 
                 // 클라이언트에게 세션 ID 전송
-                await SendSessionIdToClient(socket, sessionId);
+                await sessionService.SendSessionIdAsync(sessionId);
 
                 // WebSocket 연결 유지
                 await KeepWebSocketAlive(socket, sessionId, sessionService);
@@ -69,25 +70,6 @@ private string GenerateSessionId()
             return $"session_{DateTime.UtcNow.Ticks}_{Guid.NewGuid().ToString("N")[..8]}";
         }
         
-        private async Task SendSessionIdToClient(WebSocket socket, string sessionId)
-        {
-            try
-            {
-                var message = $"{{\"type\":\"session_id\",\"session_id\":\"{sessionId}\"}}";
-                var buffer = System.Text.Encoding.UTF8.GetBytes(message);
-                await socket.SendAsync(
-                    new ArraySegment<byte>(buffer),
-                    WebSocketMessageType.Text,
-                    true,
-                    CancellationToken.None
-                );
-            }
-            catch (Exception ex)
-            {
-                _logger.LogError(ex, "세션 ID 전송 실패: {SessionId}", sessionId);
-            }
-        }
-        
         private async Task KeepWebSocketAlive(WebSocket socket, string sessionId, ISessionService sessionService)
         {
             var buffer = new byte[1024];
diff --git a/ProjectVG.Application/Models/Chat/WebSocketMessage.cs b/ProjectVG.Application/Models/Chat/WebSocketMessage.cs
@@ -0,0 +1,24 @@
+using System.Text.Json.Serialization;
+
+namespace ProjectVG.Application.Models.Chat
+{
+    /// <summary>
+    /// WebSocket 메시지 기본 구조
+    /// </summary>
+    public class WebSocketMessage
+    {
+        [JsonPropertyName("type")]
+        public string Type { get; set; } = string.Empty;
+        
+        [JsonPropertyName("data")]
+        public object Data { get; set; } = new();
+        
+        public WebSocketMessage() { }
+        
+        public WebSocketMessage(string type, object data)
+        {
+            Type = type;
+            Data = data;
+        }
+    }
+} 
diff --git a/ProjectVG.Application/Services/Chat/Handlers/ResultSender.cs b/ProjectVG.Application/Services/Chat/Handlers/ResultSender.cs
@@ -29,14 +29,21 @@ public async Task SendResultAsync(ChatPreprocessContext context, ChatProcessResu
                         continue;
                     }
                     
-                    // 통합 메시지로 전송
-                    await _sessionService.SendIntegratedMessageAsync(
-                        context.SessionId,
-                        segment.Text,
-                        segment.AudioData,
-                        segment.AudioContentType ?? "wav",
-                        segment.AudioLength
-                    );
+                    // WebSocket 메시지로 래핑하여 전송
+                    var integratedMessage = new IntegratedChatMessage
+                    {
+                        SessionId = context.SessionId,
+                        Text = segment.Text,
+                        AudioFormat = segment.AudioContentType ?? "wav",
+                        AudioLength = segment.AudioLength,
+                        Timestamp = DateTime.UtcNow
+                    };
+                    
+                    // 오디오 데이터를 Base64로 변환
+                    integratedMessage.SetAudioData(segment.AudioData);
+                    
+                    var wsMessage = new WebSocketMessage("chat", integratedMessage);
+                    await _sessionService.SendWebSocketMessageAsync(context.SessionId, wsMessage);
                     
                     _logger.LogDebug("세그먼트 전송 완료: 세션 {SessionId}, 순서 {Order}, 텍스트: {HasText}, 오디오: {HasAudio}", 
                         context.SessionId, segment.Order, segment.HasText, segment.HasAudio);
diff --git a/ProjectVG.Application/Services/Session/ISessionService.cs b/ProjectVG.Application/Services/Session/ISessionService.cs
@@ -1,5 +1,6 @@
 using ProjectVG.Infrastructure.Services.Session;
 using System.Net.WebSockets;
+using ProjectVG.Application.Models.Chat;
 
 namespace ProjectVG.Application.Services.Session
 {
@@ -70,5 +71,18 @@ public interface ISessionService
         /// <param name="sessionId">세션 ID</param>
         /// <returns>세션 존재 여부</returns>
         Task<bool> SessionExistsAsync(string sessionId);
+
+        /// <summary>
+        /// WebSocket 메시지를 전송합니다
+        /// </summary>
+        /// <param name="sessionId">세션 ID</param>
+        /// <param name="message">전송할 WebSocket 메시지</param>
+        Task SendWebSocketMessageAsync(string sessionId, WebSocketMessage message);
+
+        /// <summary>
+        /// 세션 ID를 클라이언트에게 전송합니다
+        /// </summary>
+        /// <param name="sessionId">세션 ID</param>
+        Task SendSessionIdAsync(string sessionId);
     }
 } 
diff --git a/ProjectVG.Application/Services/Session/SessionService.cs b/ProjectVG.Application/Services/Session/SessionService.cs
@@ -139,7 +139,9 @@ public async Task SendIntegratedMessageAsync(string sessionId, string? text = nu
                 // 오디오 데이터를 Base64로 변환
                 integratedMessage.SetAudioData(audioData);
 
-                var jsonMessage = JsonSerializer.Serialize(integratedMessage);
+                // WebSocket 메시지로 래핑
+                var wsMessage = new WebSocketMessage("chat", integratedMessage);
+                var jsonMessage = JsonSerializer.Serialize(wsMessage);
                 var buffer = Encoding.UTF8.GetBytes(jsonMessage);
                 
                 await connection.WebSocket.SendAsync(
@@ -276,5 +278,53 @@ public async Task<bool> SessionExistsAsync(string sessionId)
                 return false;
             }
         }
+
+        public async Task SendWebSocketMessageAsync(string sessionId, WebSocketMessage message)
+        {
+            try
+            {
+                var connection = await _sessionRepository.GetAsync(sessionId);
+                if (connection == null)
+                {
+                    _logger.LogWarning("세션을 찾을 수 없음: {SessionId}", sessionId);
+                    return;
+                }
+
+                var jsonMessage = JsonSerializer.Serialize(message);
+                var buffer = Encoding.UTF8.GetBytes(jsonMessage);
+                
+                await connection.WebSocket.SendAsync(
+                    new ArraySegment<byte>(buffer),
+                    WebSocketMessageType.Text,
+                    true,
+                    CancellationToken.None
+                );
+
+                _logger.LogInformation("WebSocket 메시지 전송 완료: {SessionId}, 타입: {MessageType}", 
+                    sessionId, message.Type);
+            }
+            catch (Exception ex)
+            {
+                _logger.LogError(ex, "WebSocket 메시지 전송 실패: {SessionId}", sessionId);
+                throw;
+            }
+        }
+
+        public async Task SendSessionIdAsync(string sessionId)
+        {
+            try
+            {
+                var sessionData = new { session_id = sessionId };
+                var wsMessage = new WebSocketMessage("session_id", sessionData);
+                await SendWebSocketMessageAsync(sessionId, wsMessage);
+                
+                _logger.LogInformation("세션 ID 전송 완료: {SessionId}", sessionId);
+            }
+            catch (Exception ex)
+            {
+                _logger.LogError(ex, "세션 ID 전송 실패: {SessionId}", sessionId);
+                throw;
+            }
+        }
     }
 } 
diff --git a/run-prod.ps1 b/run-prod.ps1
@@ -1,14 +1,37 @@
 # 프로덕션 환경 실행 스크립트
 Write-Host "🚀 ProjectVG 프로덕션 환경 시작 중..." -ForegroundColor Green
 
-# 기존 컨테이너 정리
-Write-Host "📦 기존 컨테이너 정리 중..." -ForegroundColor Yellow
+# 1. 새 이미지 빌드 (기존 컨테이너는 그대로 유지)
+Write-Host "🔨 새 이미지 빌드 중..." -ForegroundColor Yellow
+docker-compose build --no-cache
+
+# 2. 빌드 성공 확인
+if ($LASTEXITCODE -ne 0) {
+    Write-Host "❌ 이미지 빌드 실패!" -ForegroundColor Red
+    exit 1
+}
+
+# 3. 기존 컨테이너 중지 및 제거
+Write-Host "📦 기존 컨테이너 교체 중..." -ForegroundColor Yellow
 docker-compose down
 
-# 프로덕션 환경 빌드 및 실행
-Write-Host "🔨 프로덕션 환경 빌드 및 실행 중..." -ForegroundColor Yellow
-docker-compose up --build -d
+# 4. 새 컨테이너 시작
+Write-Host "🚀 새 컨테이너 시작 중..." -ForegroundColor Yellow
+docker-compose up -d
+
+# 5. 시작 확인
+Write-Host "⏳ 컨테이너 시작 대기 중..." -ForegroundColor Yellow
+Start-Sleep -Seconds 5
 
-Write-Host "✅ 프로덕션 환경이 시작되었습니다!" -ForegroundColor Green
-Write-Host "🌐 API: http://localhost:7900" -ForegroundColor Cyan
-Write-Host "📚 Swagger: http://localhost:7900/swagger" -ForegroundColor Cyan 
+# 6. 상태 확인
+$containers = docker-compose ps -q
+if ($containers) {
+    Write-Host "✅ 프로덕션 환경이 성공적으로 시작되었습니다!" -ForegroundColor Green
+    Write-Host "🌐 API: http://localhost:7900" -ForegroundColor Cyan
+    Write-Host "📚 Swagger: http://localhost:7900/swagger" -ForegroundColor Cyan
+    Write-Host "📊 컨테이너 상태:" -ForegroundColor Cyan
+    docker-compose ps
+} else {
+    Write-Host "❌ 컨테이너 시작 실패!" -ForegroundColor Red
+    exit 1
+} 
diff --git a/test-clients/test-client.html b/test-clients/test-client.html
@@ -251,22 +251,77 @@
         if (typeof event.data === "string") {
           try {
             const data = JSON.parse(event.data);
+            console.log("수신된 메시지:", data);
             
-            // 세션 ID 처리
+            // 새로운 WebSocket 메시지 구조 처리 (우선순위)
+            if (data.type && data.data !== undefined) {
+              console.log(`메시지 타입: ${data.type}`);
+              
+              // 세션 ID 처리
+              if (data.type === "session_id") {
+                sessionId = data.data.session_id;
+                appendLog(`<b>[세션 ID: ${sessionId}]</b>`);
+                return;
+              }
+              
+              // 채팅 메시지 처리
+              if (data.type === "chat") {
+                const chatData = data.data;
+                let messageText = "";
+                
+                // 텍스트 메시지가 있는 경우
+                if (chatData.text) {
+                  messageText += `<b>AI:</b> ${chatData.text}`;
+                }
+                
+                // 오디오 데이터가 있는 경우
+                if (chatData.audioData) {
+                  try {
+                    // Base64 디코딩
+                    const audioBytes = atob(chatData.audioData);
+                    const audioArray = new Uint8Array(audioBytes.length);
+                    for (let i = 0; i < audioBytes.length; i++) {
+                      audioArray[i] = audioBytes.charCodeAt(i);
+                    }
+                    
+                    const blob = new Blob([audioArray], { type: `audio/${chatData.audioFormat || 'wav'}` });
+                    audioQueue.push(blob);
+                    if (!isPlayingAudio) playNextAudio();
+                    
+                    messageText += chatData.text ? " [오디오 포함]" : "<b>AI:</b> [음성 메시지]";
+                  } catch (e) {
+                    console.error("오디오 데이터 파싱 오류:", e);
+                    messageText += " [오디오 파싱 오류]";
+                  }
+                }
+                
+                if (messageText) {
+                  appendLog(messageText);
+                }
+                
+                // 메타데이터 표시 (개발용)
+                if (chatData.metadata) {
+                  appendLog(`<small style='color:#666'>[메타데이터: ${JSON.stringify(chatData.metadata)}]</small>`);
+                }
+                
+                return;
+              }
+              
+              // 기타 메시지 타입 처리
+              appendLog(`<small style='color:#666'>[메시지 타입: ${data.type}]</small>`);
+              return;
+            }
+            
+            // 하위 호환성을 위한 기존 구조 처리
             if (data.type === "session_id") {
               sessionId = data.session_id;
               appendLog(`<b>[세션 ID: ${sessionId}]</b>`);
               return;
             }
             
-            // 통합 메시지 처리
             if (data.type === "chat") {
               let messageText = "";
               
-              // 메시지 타입 확인
-              const messageType = data.messageType || "json";
-              console.log(`메시지 타입: ${messageType}`);
-              
               // 텍스트 메시지가 있는 경우
               if (data.text) {
                 messageText += `<b>AI:</b> ${data.text}`;

Original file line number	Diff line number	Diff line change
`@@ -17,5 +17,18 @@ public IActionResult GetConfig()`
`17`	`17`	`version = "1.0.0"`
`18`	`18`	`});`
`19`	`19`	`}`
	`20`	`+`
	`21`	`+ [HttpGet("health")]`
	`22`	`+ [HttpGet("/health")]`
	`23`	`+ public IActionResult Health()`
	`24`	`+ {`
	`25`	`+ return Ok(new`
	`26`	`+ {`
	`27`	`+ status = "healthy",`
	`28`	`+ timestamp = DateTime.UtcNow,`
	`29`	`+ version = "1.0.0",`
	`30`	`+ uptime = Environment.TickCount / 1000.0`
	`31`	`+ });`
	`32`	`+ }`
`20`	`33`	`}`
`21`	`34`	`}`