fix: 第二轮修复 — 数据库启动检查、会话持久化、URL路由、设备排序等
1. DevTools 启动前检查数据库状态,失败时自动尝试启动 2. ai-core 添加数据库断线重连机制 (30秒间隔) 3. Dashboard 添加数据库状态卡片 (启动/停止/重启) 4. Gateway 会话空闲超时管理 (30分钟标记空闲) 5. 会话/消息 PostgreSQL 持久化 (SessionStore + REST API) 6. 前端服务端会话持久化 + URL hash 路由 + 侧边栏管理 7. 管理员回到主对话按钮 8. IoT 设备卡片固定排序 9. 更新相关文档
This commit is contained in:
@@ -14,6 +14,7 @@ import (
|
||||
"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"
|
||||
)
|
||||
|
||||
@@ -26,6 +27,17 @@ func main() {
|
||||
// 加载配置
|
||||
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)
|
||||
@@ -39,13 +51,18 @@ func main() {
|
||||
|
||||
// 初始化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)
|
||||
router.Setup(r, hub, cfg, sessionStore)
|
||||
|
||||
// 启动服务
|
||||
srv := &http.Server{
|
||||
@@ -68,6 +85,13 @@ func main() {
|
||||
|
||||
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)
|
||||
|
||||
Reference in New Issue
Block a user