feat: TCP 和 Unix socket 同时监听
之前是二选一(配了 unix 就不监听 TCP),调试不方便。 现在 TCP 始终监听(调试/健康检查),Unix socket 配了就额外监听(nginx 反代)。 启动时清理残留 socket 文件,避免 bind 失败。
This commit is contained in:
@@ -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()
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user