53 lines
1.4 KiB
Go
53 lines
1.4 KiB
Go
|
// Package main 应用程序入口
|
|||
|
|
// 初始化配置、数据库、WebSocket Hub，启动HTTP服务
|
|||
|
|
// API服务运行在16702端口，WebSocket服务运行在16703端口
|
|||
|
|
package main
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"fmt"
|
|||
|
|
"log"
|
|||
|
|
"net/http"
|
|||
|
|
|
|||
|
|
"excel-api/config"
|
|||
|
|
"excel-api/models"
|
|||
|
|
"excel-api/router"
|
|||
|
|
ws "excel-api/websocket"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func main() {
|
|||
|
|
// 第一步：加载配置文件
|
|||
|
|
if err := config.Init(); err != nil {
|
|||
|
|
log.Fatalf("加载配置文件失败: %v", err)
|
|||
|
|
}
|
|||
|
|
log.Println("配置文件加载成功")
|
|||
|
|
|
|||
|
|
// 第二步：初始化数据库连接
|
|||
|
|
if err := models.InitDB(); err != nil {
|
|||
|
|
log.Fatalf("初始化数据库失败: %v", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 第三步：创建并启动WebSocket Hub
|
|||
|
|
hub := ws.NewHub()
|
|||
|
|
go hub.Run()
|
|||
|
|
log.Println("WebSocket Hub 已启动")
|
|||
|
|
|
|||
|
|
// 第四步：启动WebSocket独立服务（端口16703）
|
|||
|
|
wsAddr := fmt.Sprintf(":%d", config.App.Server.WsPort)
|
|||
|
|
wsRouter := http.NewServeMux()
|
|||
|
|
wsRouter.HandleFunc("/ws", ws.HandleWebSocketHTTP(hub))
|
|||
|
|
go func() {
|
|||
|
|
log.Printf("WebSocket服务启动，监听端口 %s", wsAddr)
|
|||
|
|
if err := http.ListenAndServe(wsAddr, wsRouter); err != nil {
|
|||
|
|
log.Printf("WebSocket服务启动失败: %v", err)
|
|||
|
|
}
|
|||
|
|
}()
|
|||
|
|
|
|||
|
|
// 第五步：配置路由并启动API HTTP服务（端口16702）
|
|||
|
|
r := router.Setup(hub)
|
|||
|
|
apiAddr := fmt.Sprintf(":%d", config.App.Server.Port)
|
|||
|
|
log.Printf("API服务启动，监听端口 %s", apiAddr)
|
|||
|
|
if err := r.Run(apiAddr); err != nil {
|
|||
|
|
log.Fatalf("API服务启动失败: %v", err)
|
|||
|
|
}
|
|||
|
|
}
|