更新底层架构

cmd/server.go

@@ -3,19 +3,19 @@ package cmd
 import (
 	"context"
 	"fmt"
-	"io"
 	"net/http"
 	"os"
 	"os/signal"
 	"syscall"
 	"time"
 
-	"networkDev/database"
-	"networkDev/middleware"
-	"networkDev/server"
-	"networkDev/utils"
-	"networkDev/utils/logger"
-	"networkDev/web"
+	"NetworkAuth/database"
+	"NetworkAuth/middleware"
+	"NetworkAuth/server"
+	"NetworkAuth/services"
+	"NetworkAuth/utils"
+	"NetworkAuth/utils/logger"
+	"NetworkAuth/web"
 
 	"github.com/gin-gonic/gin"
 	"github.com/sirupsen/logrus"
@@ -23,22 +23,14 @@ import (
 	"github.com/spf13/viper"
 )
 
-// ============================================================================
-// 命令定义
-// ============================================================================
-
 // serverCmd 代表服务器命令
 var serverCmd = &cobra.Command{
 	Use:   "server",
-	Short: "启动HTTP服务器",
-	Long:  `启动一个简单的HTTP服务器，监听配置文件中指定的端口。`,
+	Short: "启动网络授权服务",
+	Long:  `启动 NetworkAuth HTTP 服务器，监听配置文件中指定的端口，提供 Web 管理界面和 API 服务。`,
 	Run:   runServer,
 }
 
-// ============================================================================
-// 初始化函数
-// ============================================================================
-
 func init() {
 	// 将服务器命令添加到根命令
 	rootCmd.AddCommand(serverCmd)
@@ -48,10 +40,6 @@ func init() {
 	serverCmd.Flags().IntP("port", "p", 0, "服务器监听端口 (覆盖配置文件)")
 }
 
-// ============================================================================
-// 主要函数
-// ============================================================================
-
 // runServer 运行HTTP服务器
 func runServer(cmd *cobra.Command, args []string) {
 	// 获取配置
@@ -63,21 +51,47 @@ func runServer(cmd *cobra.Command, args []string) {
 	logger := logger.GetLogger()
 	logger.LogServerStart(host, port)
 
+	// 重定向 Gin 框架内部日志到 Logrus
+	// 这将捕获 [GIN-debug] 路由注册日志和其他框架级输出
+	gin.DefaultWriter = logger.WriterLevel(logrus.DebugLevel)
+	gin.DefaultErrorWriter = logger.WriterLevel(logrus.ErrorLevel)
+
+	// 设置 Gin 模式
+	if !viper.GetBool("server.dev_mode") {
+		gin.SetMode(gin.ReleaseMode)
+	}
+
 	// 初始化Redis（如果配置存在，失败不致命）
 	utils.InitRedis()
 
 	// 初始化数据库（根据 viper 配置选择 SQLite 或 MySQL）
 	// 如果初始化失败则回退并退出
-	if _, err := database.Init(); err != nil {
+	db, err := database.Init()
+	if err != nil {
 		logrus.WithError(err).Fatal("数据库初始化失败")
 	}
-	// 执行自动迁移（确保表结构存在）
-	if err := database.AutoMigrate(); err != nil {
-		logrus.WithError(err).Fatal("数据库自动迁移失败")
-	}
-	// 初始化默认系统设置（包含管理员账号）
-	if err := database.SeedDefaultSettings(); err != nil {
-		logrus.WithError(err).Fatal("默认系统设置初始化失败")
+
+	if db != nil {
+		// 执行自动迁移（确保表结构存在）
+		if err := database.AutoMigrate(); err != nil {
+			logrus.WithError(err).Fatal("数据库自动迁移失败")
+		}
+		// 初始化默认系统设置
+		if err := database.SeedDefaultSettings(); err != nil {
+			logrus.WithError(err).Fatal("默认系统设置初始化失败")
+		}
+
+		// 初始化加密管理器
+		// 从数据库设置中获取加密密钥
+		encryptionKey := services.GetSettingsService().GetEncryptionKey()
+		if err := utils.InitEncryption(encryptionKey); err != nil {
+			logrus.WithError(err).Fatal("加密管理器初始化失败")
+		}
+
+		// 启动日志清理定时任务
+		services.StartLogCleanupTask()
+	} else {
+		logrus.Info("系统处于未初始化状态，跳过数据库自动迁移和设置加载")
 	}
 
 	// 创建HTTP服务器
@@ -87,10 +101,6 @@ func runServer(cmd *cobra.Command, args []string) {
 	startServer(server)
 }
 
-// ============================================================================
-// 辅助函数
-// ============================================================================
-
 // getServerHost 获取服务器监听地址
 func getServerHost(cmd *cobra.Command) string {
 	if host, _ := cmd.Flags().GetString("host"); host != "" {
@@ -109,49 +119,51 @@ func getServerPort(cmd *cobra.Command) int {
 
 // createHTTPServer 创建HTTP服务器
 func createHTTPServer(addr string) *http.Server {
-	// 配置Gin模式和日志
-	configureGin()
+	// 创建 Gin 引擎
+	r := gin.New()
 
-	// 创建Gin引擎
-	router := gin.New()
-
-	// 添加恢复中间件
-	router.Use(gin.Recovery())
+	// 使用默认的 Recovery 中间件
+	r.Use(gin.Recovery())
 
 	// 添加日志中间件
-	router.Use(middleware.WrapHandler())
+	// 默认为 true，只有显式设置为 false 才关闭
+	enableAccessLog := true
+	if viper.IsSet("server.access_log") {
+		enableAccessLog = viper.GetBool("server.access_log")
+	}
+	if enableAccessLog {
+		r.Use(middleware.WrapHandler())
+	}
 
-	// 添加开发模式中间件（统一管理开发模式功能）
-	router.Use(middleware.DevModeMiddleware(router))
+	// 添加安装检查中间件
+	r.Use(middleware.InstallCheckMiddleware())
 
-	// 加载模板
-	if err := loadTemplates(router); err != nil {
-		logrus.WithError(err).Fatal("模板加载失败")
+	// 添加维护模式中间件
+	r.Use(middleware.MaintenanceMiddleware())
+
+	// 添加开发模式中间件（统一管理开发模式功能：模板热重载等）
+	r.Use(middleware.DevModeMiddleware(r))
+
+	// 加载并设置 HTML 模板
+	if tmpl, err := web.ParseTemplates(); err == nil {
+		r.SetHTMLTemplate(tmpl)
+	} else {
+		logrus.WithError(err).Error("HTML模板加载失败")
 	}
 
 	// 注册路由
-	registerRoutes(router)
+	registerRoutes(r)
 
 	return &http.Server{
 		Addr:    addr,
-		Handler: router,
+		Handler: r,
 	}
 }
 
-// loadTemplates 加载模板到Gin引擎
-func loadTemplates(router *gin.Engine) error {
-	tmpl, err := web.ParseTemplates()
-	if err != nil {
-		return err
-	}
-	router.SetHTMLTemplate(tmpl)
-	return nil
-}
-
 // registerRoutes 注册HTTP路由
-func registerRoutes(router *gin.Engine) {
+func registerRoutes(r *gin.Engine) {
 	// 使用server包中的路由注册函数
-	server.RegisterRoutes(router)
+	server.RegisterRoutes(r)
 }
 
 // startServer 启动服务器并处理优雅关闭
@@ -174,8 +186,6 @@ func startServer(server *http.Server) {
 
 	// 等待中断信号
 	<-sigChan
-	// 清除终端上的 ^C 字符并移动光标到行首
-	fmt.Print("\r\033[K")
 	logger.Info("收到关闭信号，正在优雅关闭服务器...")
 
 	// 创建一个带超时的上下文
@@ -189,20 +199,3 @@ func startServer(server *http.Server) {
 		logger.LogServerStop()
 	}
 }
-
-// configureGin 配置Gin的全局设置
-func configureGin() {
-	// 禁用Gin的颜色输出，提高控制台兼容性
-	gin.DisableConsoleColor()
-
-	// 设置Gin的输出为丢弃，因为我们使用自定义日志中间件
-	gin.DefaultWriter = io.Discard
-	gin.DefaultErrorWriter = io.Discard
-
-	// 根据配置设置Gin模式
-	if viper.GetString("app.mode") == "production" {
-		gin.SetMode(gin.ReleaseMode)
-	} else {
-		gin.SetMode(gin.DebugMode)
-	}
-}