算了，后端我自己写吧

2026-04-01 12:09:02 +08:00
parent 4138340f53
commit 1a0a01a56d
69 changed files with 1949 additions and 102 deletions
@@ -0,0 +1,139 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"log"
+	"net/http"
+	"os"
+	"os/signal"
+	"syscall"
+	"time"
+
+	"github.com/gin-gonic/gin"
+	"go.uber.org/zap"
+
+	"ops/api"
+	"ops/internal/config"
+	"ops/internal/database"
+	"ops/internal/middleware"
+)
+
+func main() {
+	// 创建日志记录器
+	logger, err := createLogger()
+	if err != nil {
+		log.Fatalf("Failed to create logger: %v", err)
+	}
+	defer logger.Sync()
+
+	// 加载配置
+	configPath := "./data/config.yaml"
+	if err := config.Load(configPath); err != nil {
+		logger.Fatal("Failed to load config", zap.Error(err))
+	}
+
+	// 初始化数据库
+	if err := database.Init(); err != nil {
+		logger.Fatal("Failed to connect database", zap.Error(err))
+	}
+	defer database.Close()
+
+	// 自动迁移数据库表
+	if err := database.AutoMigrate(); err != nil {
+		logger.Warn("Auto migration failed", zap.Error(err))
+	}
+
+	// 设置Gin模式
+	if config.Current.Web.Host == "127.0.0.1" || config.Current.Web.Host == "localhost" {
+		gin.SetMode(gin.DebugMode)
+	} else {
+		gin.SetMode(gin.ReleaseMode)
+	}
+
+	// 创建Gin实例
+	r := gin.New()
+
+	// 注册中间件
+	r.Use(middleware.CORS())
+
+	// 根据环境选择日志中间件
+	if config.Current.Web.Host == "127.0.0.1" || config.Current.Web.Host == "localhost" {
+		r.Use(middleware.SimpleLogger())
+	} else {
+		r.Use(middleware.Logger(logger))
+	}
+
+	r.Use(middleware.Recovery(logger))
+
+	// 注册API路由
+	api.RegisterAllRoutes(r)
+
+	// 健康检查端点
+	r.GET("/health", func(c *gin.Context) {
+		c.JSON(200, gin.H{
+			"code":    "0",
+			"message": "Server is healthy",
+			"data": gin.H{
+				"timestamp": time.Now().Unix(),
+				"status":    "running",
+				"version":   "1.0.0",
+			},
+		})
+	})
+
+	// 确保dist目录存在
+	ensureDistDirectory(logger)
+
+	// 启动HTTP服务器
+	addr := fmt.Sprintf("%s:%s", config.Current.Web.Host, config.Current.Web.Port)
+	srv := &http.Server{
+		Addr:         addr,
+		Handler:      r,
+		ReadTimeout:  15 * time.Second,
+		WriteTimeout: 15 * time.Second,
+		IdleTimeout:  60 * time.Second,
+	}
+
+	go func() {
+		logger.Info("Server starting", zap.String("addr", addr))
+		if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
+			logger.Fatal("Failed to start server", zap.Error(err))
+		}
+	}()
+
+	// 优雅关机
+	quit := make(chan os.Signal, 1)
+	signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
+	<-quit
+
+	logger.Info("Shutting down server...")
+
+	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
+	defer cancel()
+
+	if err := srv.Shutdown(ctx); err != nil {
+		logger.Fatal("Server forced to shutdown", zap.Error(err))
+	}
+
+	logger.Info("Server exited")
+}
+
+// createLogger 创建日志记录器
+func createLogger() (*zap.Logger, error) {
+	if gin.Mode() == gin.DebugMode {
+		return zap.NewDevelopment()
+	}
+	return zap.NewProduction()
+}
+
+// ensureDistDirectory 确保dist目录存在
+func ensureDistDirectory(logger *zap.Logger) {
+	if _, err := os.Stat("./dist"); os.IsNotExist(err) {
+		if err := os.MkdirAll("./dist", 0755); err != nil {
+			logger.Warn("Failed to create dist directory", zap.Error(err))
+		} else {
+			logger.Info("Created empty dist directory for static files")
+		}
+	}
+}