feat: TCP 和 Unix socket 同时监听

之前是二选一(配了 unix 就不监听 TCP),调试不方便。
现在 TCP 始终监听(调试/健康检查),Unix socket 配了就额外监听(nginx 反代)。
启动时清理残留 socket 文件,避免 bind 失败。
This commit is contained in:
2026-05-28 15:59:38 +08:00
parent 28f8f19538
commit 2f3d47a439
+34 -19
View File
@@ -7,6 +7,7 @@ import (
"os" "os"
"path/filepath" "path/filepath"
"strings" "strings"
"sync"
"simple_portal/config" "simple_portal/config"
"simple_portal/database" "simple_portal/database"
@@ -154,25 +155,39 @@ func main() {
} }
} }
// 启动服务器 // 启动服务器:同时监听 TCP 和 Unix socket(如果配置了的话)
var wg sync.WaitGroup
if config.Cfg.Server.Unix != "" { if config.Cfg.Server.Unix != "" {
// Unix socket 模式 wg.Add(1)
listener, err := net.Listen("unix", config.Cfg.Server.Unix) go func() {
if err != nil { defer wg.Done()
log.Fatalf("监听 Unix socket 失败: %v", err) // 清理残留的 socket 文件
} os.Remove(config.Cfg.Server.Unix)
// 设置 socket 文件权限,允许 nginx 等其他进程访问
os.Chmod(config.Cfg.Server.Unix, 0666) listener, err := net.Listen("unix", config.Cfg.Server.Unix)
log.Printf("启动 Portal 服务器,监听 Unix socket: %s", config.Cfg.Server.Unix) if err != nil {
if err := r.RunListener(listener); err != nil { log.Fatalf("监听 Unix socket 失败: %v", err)
log.Fatalf("服务器启动失败: %v", err) }
} // 设置 socket 文件权限,允许 nginx 等其他进程访问
} else { os.Chmod(config.Cfg.Server.Unix, 0666)
// TCP 模式 log.Printf("监听 Unix socket: %s", config.Cfg.Server.Unix)
addr := config.Cfg.Server.Addr if err := r.RunListener(listener); err != nil {
log.Printf("启动 Portal 服务器,监听: %s", addr) log.Fatalf("Unix socket 服务启动失败: %v", err)
if err := r.Run(addr); err != nil { }
log.Fatalf("服务器启动失败: %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()
} }