Merge pull request #6 from Team-TicketMate/20251205_#5_컨테이너_로그_실시간_로깅_기능_추가

컨테이너_로그_실시간_로깅_기능_추가 : feat : 실시간 로깅 구현 https://github.com/Team-Ticke…

Author: Chuseok22

internal/dockercli/docker_cli.go

@@ -2,7 +2,9 @@ package dockercli
 
 import (
 	"bytes"
+	"context"
 	"fmt"
+	"io"
 	"os/exec"
 	"strings"
 )
@@ -101,3 +103,40 @@ func FetchContainerLogs(containerID string, tailLines *int) (string, error) {
 
 	return output, nil
 }
+
+func StreamContainerLogs(ctx context.Context, containerID string, tailLines *int, output io.Writer) error {
+	trimmedID := strings.TrimSpace(containerID)
+	if trimmedID == "" {
+		return fmt.Errorf("컨테이너 ID가 비어있습니다")
+	}
+
+	args := []string{"logs"}
+
+	if tailLines != nil {
+		if *tailLines <= 0 {
+			return fmt.Errorf("tailLines 는 1 이상의 숫자여야합니다")
+		}
+		args = append(args, "--tail", fmt.Sprintf("%d", *tailLines))
+	}
+
+	// docker logs --tail N --follow <id>
+	args = append(args, "--follow", trimmedID)
+
+	// CommandContext를 사용해서 request 컨텍스트가 취소되면 docker 프로세스도 종료
+	cmd := exec.CommandContext(ctx, "docker", args...)
+
+	// stdout, stderr 모두 HTTP 응답으로 보냄
+	cmd.Stdout = output
+	cmd.Stderr = output
+
+	if err := cmd.Start(); err != nil {
+		return fmt.Errorf("'docker logs --follow' 시작에 실패했습니다: %w", err)
+	}
+
+	// Wait은 컨텍스트 취소 또는 프로세스 종료까지 블로킹
+	if err := cmd.Wait(); err != nil {
+		return fmt.Errorf("'docker logs --follow' 실행 중 오류가 발생했습니다: %w", err)
+	}
+
+	return nil
+}

internal/httpapi/containers_handler.go

@@ -10,6 +10,36 @@ import (
 	"ticketmate-logviewer/internal/dockercli"
 )
 
+type flushWriter struct {
+	writer  http.ResponseWriter
+	flusher http.Flusher
+}
+
+// ResponseWriter 가 http.Flusher를 지원하는지 확인하고 감싸주는 함수
+func newFlushWriter(writer http.ResponseWriter) (*flushWriter, error) {
+	flusher, ok := writer.(http.Flusher)
+	if !ok {
+		return nil, fmt.Errorf("http.ResponseWriter가 스트리밍을 지원하지 않습니다")
+	}
+
+	return &flushWriter{
+		writer:  writer,
+		flusher: flusher,
+	}, nil
+}
+
+// io.Writer 인터페이스 구현
+func (flushWriterInstance *flushWriter) Write(p []byte) (int, error) {
+	written, err := flushWriterInstance.writer.Write(p)
+	if err != nil {
+		return written, err
+	}
+
+	// 매번 쓰고 나서 즉시 클라이언트로 flush
+	flushWriterInstance.flusher.Flush()
+	return written, nil
+}
+
 func containersHandler(writer http.ResponseWriter, request *http.Request) {
 	if request.Method != http.MethodGet {
 		http.Error(writer, "허용되지 않은 HTTP 메서드 압니다", http.StatusMethodNotAllowed)
@@ -62,6 +92,34 @@ func containerLogsHandler(writer http.ResponseWriter, request *http.Request) {
 		return
 	}
 
+	// follow 파라미터 처리: follow = true 또는 follow = 1이면 실시간 스트리밍
+	followParam := request.URL.Query().Get("follow")
+	follow := strings.EqualFold(followParam, "true") || followParam == "1"
+
+	// 로그는 text/plain 스트리밍 또는 전체 응답
+	writer.Header().Set("Content-Type", "text/plain; charset=utf-8")
+
+	if follow {
+		// 실시간 스트리밍 모드
+		streamWriter, newWriterErr := newFlushWriter(writer)
+		if newWriterErr != nil {
+			log.Printf("HTTP 스트리밍을 지원하지 않는 환경입니다: %v", newWriterErr)
+			http.Error(writer, "스트리밍을 지원하지 않는 환경입니다", http.StatusInternalServerError)
+			return
+		}
+
+		log.Printf("컨테이너 로그 스트리밍 시작 (follow 모드). 컨테이너 ID: %s", containerID)
+
+		// request.Context() 를 넘겨서 클라이언트가 연결을 끊으면 doker 프로세스도 종료
+		streamErr := dockercli.StreamContainerLogs(request.Context(), containerID, tailLines, streamWriter)
+		if streamErr != nil {
+			// 스트리밍 중 에러는 서버 로그로만 출력. 응답 바디에는 보내지 않음
+			log.Printf("컨테이너 로그 스트리밍 실패. 컨테이너 ID: %s: %v", containerID, streamErr)
+		}
+		log.Printf("컨테이너 로그 스트리밍 종료. 컨테이너 ID: %s", containerID)
+		return
+	}
+
 	logText, err := dockercli.FetchContainerLogs(containerID, tailLines)
 	if err != nil {
 		log.Printf("컨테이너 로그 조회에 실패했습니다. 컨테이너 ID: %s: %v", containerID, err)