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