AskaEth
e7b7eff0d8
1. DevTools 启动前检查数据库状态，失败时自动尝试启动 2. ai-core 添加数据库断线重连机制 (30秒间隔) 3. Dashboard 添加数据库状态卡片 (启动/停止/重启) 4. Gateway 会话空闲超时管理 (30分钟标记空闲) 5. 会话/消息 PostgreSQL 持久化 (SessionStore + REST API) 6. 前端服务端会话持久化 + URL hash 路由 + 侧边栏管理 7. 管理员回到主对话按钮 8. IoT 设备卡片固定排序 9. 更新相关文档
100 lines
2.8 KiB
Go
100 lines
2.8 KiB
Go
package main
|
||
|
||
import (
|
||
"context"
|
||
"log"
|
||
"net/http"
|
||
"os"
|
||
"os/signal"
|
||
"syscall"
|
||
"time"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
"github.com/joho/godotenv"
|
||
"github.com/yourname/cyrene-ai/gateway/internal/config"
|
||
"github.com/yourname/cyrene-ai/gateway/internal/middleware"
|
||
"github.com/yourname/cyrene-ai/gateway/internal/router"
|
||
"github.com/yourname/cyrene-ai/gateway/internal/store"
|
||
"github.com/yourname/cyrene-ai/gateway/internal/ws"
|
||
)
|
||
|
||
func main() {
|
||
// 自动加载 .env 文件（来自 backend/.env）
|
||
if err := godotenv.Load("../.env"); err != nil {
|
||
log.Println("ℹ 未找到 .env 文件，将使用环境变量或默认值")
|
||
}
|
||
|
||
// 加载配置
|
||
cfg := config.Load()
|
||
|
||
// 初始化数据库持久化存储 (降级：连接失败不崩溃)
|
||
var sessionStore *store.SessionStore
|
||
databaseURL := cfg.DatabaseURL()
|
||
if s, err := store.NewSessionStore(databaseURL); err != nil {
|
||
log.Printf("⚠ 会话持久化存储初始化失败 (数据库不可用): %v", err)
|
||
log.Println("⚠ Gateway 将以仅内存模式运行 — 会话数据在重启后丢失")
|
||
} else {
|
||
sessionStore = s
|
||
log.Println("✅ 会话持久化存储已启用 (PostgreSQL)")
|
||
}
|
||
|
||
// 初始化Gin
|
||
if cfg.Env == "production" {
|
||
gin.SetMode(gin.ReleaseMode)
|
||
}
|
||
r := gin.New()
|
||
|
||
// 中间件
|
||
r.Use(middleware.CORS())
|
||
r.Use(middleware.RequestLogging())
|
||
r.Use(gin.Recovery())
|
||
|
||
// 初始化WebSocket Hub
|
||
hub := ws.NewHub()
|
||
hub.SetStore(sessionStore)
|
||
hub.SetIdleTimeout(cfg.SessionIdleTimeoutMin)
|
||
go hub.Run()
|
||
|
||
// 启动闲置会话清理 (标记超时会话为 idle，不删除)
|
||
hub.StartIdleCleanup()
|
||
|
||
// 启动 IoT 设备状态广播（每10秒向所有WebSocket客户端推送设备状态）
|
||
hub.StartIoTBroadcast(cfg.IoTDebugServiceURL)
|
||
|
||
// 注册路由
|
||
router.Setup(r, hub, cfg, sessionStore)
|
||
|
||
// 启动服务
|
||
srv := &http.Server{
|
||
Addr: ":" + cfg.Port,
|
||
Handler: r,
|
||
}
|
||
|
||
go func() {
|
||
log.Printf("🚀 Gateway 启动在端口 %s", cfg.Port)
|
||
if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
||
log.Fatalf("服务启动失败: %v", err)
|
||
}
|
||
}()
|
||
|
||
// 优雅关闭
|
||
quit := make(chan os.Signal, 1)
|
||
signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
|
||
<-quit
|
||
log.Println("正在关闭服务...")
|
||
|
||
hub.StopIoTBroadcast()
|
||
|
||
// 关闭数据库连接
|
||
if sessionStore != nil {
|
||
if err := sessionStore.Close(); err != nil {
|
||
log.Printf("⚠ 关闭数据库连接失败: %v", err)
|
||
}
|
||
}
|
||
|
||
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
||
defer cancel()
|
||
srv.Shutdown(ctx)
|
||
log.Println("服务已关闭")
|
||
}
|