init: 昔涟项目骨架

2026-05-15 20:10:35 +08:00
commit 6bde87f807
84 changed files with 3635 additions and 0 deletions
@@ -0,0 +1,64 @@
+package main
+
+import (
+    "context"
+    "log"
+    "net/http"
+    "os"
+    "os/signal"
+    "syscall"
+    "time"
+
+    "github.com/gin-gonic/gin"
+    "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/ws"
+)
+
+func main() {
+    // 加载配置
+    cfg := config.Load()
+
+    // 初始化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()
+    go hub.Run()
+
+    // 注册路由
+    router.Setup(r, hub, cfg)
+
+    // 启动服务
+    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("正在关闭服务...")
+
+    ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
+    defer cancel()
+    srv.Shutdown(ctx)
+    log.Println("服务已关闭")
+}