119 lines
3.3 KiB
Go
119 lines
3.3 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"kefu-cloud/server/internal/config"
|
|
"kefu-cloud/server/internal/geoip"
|
|
"kefu-cloud/server/internal/handler"
|
|
"kefu-cloud/server/internal/metrics"
|
|
"kefu-cloud/server/internal/middleware"
|
|
"kefu-cloud/server/internal/model"
|
|
"kefu-cloud/server/internal/service"
|
|
"kefu-cloud/server/internal/storage"
|
|
"kefu-cloud/server/internal/ws"
|
|
)
|
|
|
|
func main() {
|
|
cfg := config.Load()
|
|
|
|
// 初始化数据库
|
|
model.InitDB(cfg.Database.DSN())
|
|
|
|
// 可选 GeoIP 离线库；未配置或文件不存在时自动降级到 CDN 国家头/未知。
|
|
if err := geoip.Init(cfg.GeoIP.DBPath); err != nil {
|
|
log.Printf("警告: GeoIP 数据库未启用: %v", err)
|
|
} else if geoip.Enabled() {
|
|
log.Printf("GeoIP 数据库已启用: %s", cfg.GeoIP.DBPath)
|
|
defer geoip.Close()
|
|
go func() {
|
|
updated, err := service.BackfillVisitorGeoIP(context.Background())
|
|
if err != nil {
|
|
log.Printf("警告: 历史访客地区回填失败: %v", err)
|
|
return
|
|
}
|
|
if updated > 0 {
|
|
log.Printf("历史访客地区回填完成: %d 条", updated)
|
|
}
|
|
}()
|
|
}
|
|
|
|
// 初始化 JWT（有效期可用 JWT_EXPIRE_HOURS 配置，默认 24 小时）
|
|
middleware.InitJWT(cfg.JWT.Secret, cfg.JWT.ExpireTime)
|
|
|
|
// 对象存储（MinIO / S3 兼容）
|
|
store, err := storage.NewS3(cfg.Storage)
|
|
if err != nil {
|
|
log.Fatalf("对象存储初始化失败: %v", err)
|
|
}
|
|
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
|
defer cancel()
|
|
if err := store.EnsureBucket(ctx); err != nil {
|
|
log.Printf("警告: 确保存储桶失败（可稍后启动 MinIO）: %v", err)
|
|
} else {
|
|
log.Printf("对象存储就绪: %s bucket=%s public=%s", cfg.Storage.Endpoint, cfg.Storage.Bucket, cfg.Storage.PublicBase)
|
|
}
|
|
|
|
// 启动 WebSocket Hub
|
|
ws.SetAllowedOrigins(cfg.AllowedOrigins)
|
|
// 多 API 副本部署时通过 REDIS_ADDR 启用跨节点广播；单实例保持进程内投递
|
|
if cfg.Redis.Addr != "" {
|
|
bus := ws.NewRedisBus(cfg.Redis.Addr, cfg.Redis.Password, cfg.Redis.DB)
|
|
ws.DefaultHub.SetEventBus(bus)
|
|
log.Printf("WebSocket 跨节点广播已启用: redis=%s db=%d", cfg.Redis.Addr, cfg.Redis.DB)
|
|
defer func() { _ = bus.Close() }()
|
|
}
|
|
go ws.DefaultHub.Run()
|
|
|
|
// 设置 Gin 模式
|
|
gin.SetMode(cfg.Server.Mode)
|
|
|
|
r := gin.Default()
|
|
// 全站请求统计（运维监控 24h 曲线）
|
|
r.Use(metrics.Middleware())
|
|
|
|
// CORS
|
|
r.Use(func(c *gin.Context) {
|
|
origin := c.GetHeader("Origin")
|
|
allowed := false
|
|
for _, candidate := range cfg.AllowedOrigins {
|
|
if candidate == "*" || candidate == origin {
|
|
allowed = true
|
|
break
|
|
}
|
|
}
|
|
if allowed && origin != "" {
|
|
c.Header("Access-Control-Allow-Origin", origin)
|
|
c.Header("Vary", "Origin")
|
|
}
|
|
c.Header("Access-Control-Allow-Methods", "GET,POST,PUT,DELETE,OPTIONS")
|
|
c.Header("Access-Control-Allow-Headers", "Content-Type,Authorization,X-Visitor-Token")
|
|
if c.Request.Method == "OPTIONS" {
|
|
if origin != "" && !allowed {
|
|
c.AbortWithStatus(http.StatusForbidden)
|
|
return
|
|
}
|
|
c.AbortWithStatus(204)
|
|
return
|
|
}
|
|
c.Next()
|
|
})
|
|
|
|
// 健康检查
|
|
r.GET("/health", func(c *gin.Context) {
|
|
c.JSON(200, gin.H{"status": "ok"})
|
|
})
|
|
|
|
// 注册路由
|
|
handler.SetupRoutes(r, store, cfg.Storage)
|
|
|
|
log.Printf("客服云服务启动在 :%s", cfg.Server.Port)
|
|
if err := r.Run(":" + cfg.Server.Port); err != nil {
|
|
log.Fatalf("启动失败: %v", err)
|
|
}
|
|
}
|