From 2f3d47a439043d06c57aef2709edce91fb6b8490 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=90=B4=E6=96=87=E5=B3=B0?= <kevin@lmve.net>
Date: Thu, 28 May 2026 15:59:38 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20TCP=20=E5=92=8C=20Unix=20socket=20?=
 =?UTF-8?q?=E5=90=8C=E6=97=B6=E7=9B=91=E5=90=AC?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

之前是二选一（配了 unix 就不监听 TCP），调试不方便。
现在 TCP 始终监听（调试/健康检查），Unix socket 配了就额外监听（nginx 反代）。
启动时清理残留 socket 文件，避免 bind 失败。
---
 main.go | 53 ++++++++++++++++++++++++++++++++++-------------------
 1 file changed, 34 insertions(+), 19 deletions(-)

diff --git a/main.go b/main.go
index c6f59e2..5e8a4b7 100644
--- a/main.go
+++ b/main.go
@@ -7,6 +7,7 @@ import (
 	"os"
 	"path/filepath"
 	"strings"
+	"sync"
 
 	"simple_portal/config"
 	"simple_portal/database"
@@ -154,25 +155,39 @@ func main() {
 		}
 	}
 
-	// 启动服务器
+	// 启动服务器：同时监听 TCP 和 Unix socket（如果配置了的话）
+	var wg sync.WaitGroup
+
 	if config.Cfg.Server.Unix != "" {
-		// Unix socket 模式
-		listener, err := net.Listen("unix", config.Cfg.Server.Unix)
-		if err != nil {
-			log.Fatalf("监听 Unix socket 失败: %v", err)
-		}
-		// 设置 socket 文件权限，允许 nginx 等其他进程访问
-		os.Chmod(config.Cfg.Server.Unix, 0666)
-		log.Printf("启动 Portal 服务器，监听 Unix socket: %s", config.Cfg.Server.Unix)
-		if err := r.RunListener(listener); err != nil {
-			log.Fatalf("服务器启动失败: %v", err)
-		}
-	} else {
-		// TCP 模式
-		addr := config.Cfg.Server.Addr
-		log.Printf("启动 Portal 服务器，监听: %s", addr)
-		if err := r.Run(addr); err != nil {
-			log.Fatalf("服务器启动失败: %v", err)
-		}
+		wg.Add(1)
+		go func() {
+			defer wg.Done()
+			// 清理残留的 socket 文件
+			os.Remove(config.Cfg.Server.Unix)
+
+			listener, err := net.Listen("unix", config.Cfg.Server.Unix)
+			if err != nil {
+				log.Fatalf("监听 Unix socket 失败: %v", err)
+			}
+			// 设置 socket 文件权限，允许 nginx 等其他进程访问
+			os.Chmod(config.Cfg.Server.Unix, 0666)
+			log.Printf("监听 Unix socket: %s", config.Cfg.Server.Unix)
+			if err := r.RunListener(listener); err != nil {
+				log.Fatalf("Unix socket 服务启动失败: %v", err)
+			}
+		}()
 	}
+
+	// TCP 始终监听
+	wg.Add(1)
+	go func() {
+		defer wg.Done()
+		addr := config.Cfg.Server.Addr
+		log.Printf("监听 TCP: %s", addr)
+		if err := r.Run(addr); err != nil {
+			log.Fatalf("TCP 服务启动失败: %v", err)
+		}
+	}()
+
+	wg.Wait()
 }