95 lines
2.3 KiB
Go
95 lines
2.3 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"log"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
"time"
|
|
|
|
"gitea.loveuer.com/loveuer/cluster/api"
|
|
"gitea.loveuer.com/loveuer/cluster/internal/config"
|
|
"gitea.loveuer.com/loveuer/cluster/internal/database"
|
|
"gitea.loveuer.com/loveuer/cluster/internal/middleware"
|
|
"gitea.loveuer.com/loveuer/cluster/internal/registry/storage"
|
|
|
|
"github.com/gofiber/fiber/v3"
|
|
)
|
|
|
|
func main() {
|
|
// 解析命令行参数
|
|
var (
|
|
debug = flag.Bool("debug", false, "Enable debug mode")
|
|
address = flag.String("address", ":8080", "API server listen address")
|
|
dataDir = flag.String("data-dir", "./storage", "Data directory for storing all data")
|
|
)
|
|
flag.Parse()
|
|
|
|
// 加载配置
|
|
cfg, err := config.LoadFromFlags(*debug, *address, *dataDir)
|
|
if err != nil {
|
|
log.Fatalf("Failed to load config: %v", err)
|
|
}
|
|
|
|
log.Printf("Starting server with config: debug=%v, address=%s, data-dir=%s",
|
|
cfg.Server.Debug, cfg.Server.Address, cfg.Storage.RootPath)
|
|
|
|
// 初始化数据库
|
|
if err := database.Init(cfg.Storage.RootPath); err != nil {
|
|
log.Fatalf("Failed to initialize database: %v", err)
|
|
}
|
|
|
|
// 初始化存储（使用 GORM 元数据 + 文件系统）
|
|
store, err := storage.NewGormStorage(cfg.Storage.RootPath)
|
|
if err != nil {
|
|
log.Fatalf("Failed to initialize storage: %v", err)
|
|
}
|
|
|
|
// 创建 Fiber 应用
|
|
config := fiber.Config{}
|
|
if !cfg.Server.Debug {
|
|
// Fiber v3 默认不显示启动消息
|
|
}
|
|
app := fiber.New(config)
|
|
|
|
// 全局中间件
|
|
app.Use(middleware.Logger())
|
|
app.Use(middleware.Recovery())
|
|
app.Use(middleware.CORS())
|
|
|
|
// 健康检查
|
|
app.Get("/health", func(c fiber.Ctx) error {
|
|
return c.JSON(fiber.Map{"status": "ok"})
|
|
})
|
|
|
|
// 设置路由
|
|
api.SetupRoutes(app, store)
|
|
|
|
// 启动服务器（在 goroutine 中）
|
|
go func() {
|
|
log.Printf("Server listening on %s", cfg.Server.Address)
|
|
if err := app.Listen(cfg.Server.Address); err != nil {
|
|
log.Fatalf("Failed to start server: %v", err)
|
|
}
|
|
}()
|
|
|
|
// 等待中断信号以优雅关闭服务器
|
|
quit := make(chan os.Signal, 1)
|
|
signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
|
|
<-quit
|
|
|
|
log.Println("Shutting down server...")
|
|
|
|
// 优雅关闭，等待 5 秒
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
|
|
if err := app.ShutdownWithContext(ctx); err != nil {
|
|
log.Fatalf("Server forced to shutdown: %v", err)
|
|
}
|
|
|
|
log.Println("Server exited")
|
|
}
|