优化数据库,日志,命令模块

internal/cli/init/db.go

@@ -3,7 +3,6 @@ package initcmd
 import (
 	"fmt"
 	"sunhpc/internal/middler/auth"
-	"sunhpc/pkg/config"
 	"sunhpc/pkg/database"
 	"sunhpc/pkg/logger"
 
@@ -28,26 +27,29 @@ func NewInitDBCmd() *cobra.Command {
 
 			logger.Debug("执行数据库初始化...")
 
-			cfg, err := config.LoadConfig()
-			if err != nil {
-				return fmt.Errorf("加载配置失败: %w", err)
-			}
-
 			// 初始化数据库
-			db, err := database.GetInstance(&cfg.Database, nil)
+			db, err := database.GetDB()
 			if err != nil {
 				return fmt.Errorf("数据库连接失败: %w", err)
 			}
 			defer db.Close()
 
-			if err := db.InitTables(force); err != nil {
+			if err := database.InitTables(db, force); err != nil {
 				return fmt.Errorf("数据库初始化失败: %w", err)
 			}
 
+			// 测试数据库连接
+			if err := database.TestNodeInsert(db); err != nil {
+				return fmt.Errorf("数据库测试失败: %w", err)
+			}
+
 			return nil
 		},
 	}
 
-	cmd.Flags().BoolVarP(&force, "force", "f", false, "强制重新初始化")
+	cmd.Flags().BoolVarP(
+		&force, "force", "f", false,
+		"强制重新初始化")
+
 	return cmd
 }