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() }