refactor(db): 重命名Init为InitDB并集中初始化日志

This commit is contained in:
unclesam committed 2026-07-07 23:50:38 +08:00
1 parent 1bca4808e9
commit 17d6669acb
5 files changed
+9 -7

No files matched your search

+1 -1
View File
@@ -13,7 +13,7 @@ var migrateCmd = &cobra.Command{
Use: "migrate", Use: "migrate",
Short: "执行数据库 Schema 自动迁移", Short: "执行数据库 Schema 自动迁移",
Run: func(cmd *cobra.Command, args []string) { Run: func(cmd *cobra.Command, args []string) {
err := db.Init(global.Config.Database.Driver, global.Config.Database.Dsn()) err := db.InitDB(global.Config.Database.Driver, global.Config.Database.Dsn())
if err != nil { if err != nil {
global.Log.Fatal("连接数据库失败", zap.Error(err)) global.Log.Fatal("连接数据库失败", zap.Error(err))
} }
+1 -2
View File
@@ -14,7 +14,7 @@ var serverCmd = &cobra.Command{
Short: "启动 Web API 接口服务", Short: "启动 Web API 接口服务",
Run: func(cmd *cobra.Command, args []string) { Run: func(cmd *cobra.Command, args []string) {
// 初始化 db 包 // 初始化 db 包
err := db.Init(global.Config.Database.Driver, global.Config.Database.Dsn()) err := db.InitDB(global.Config.Database.Driver, global.Config.Database.Dsn())
if err != nil { if err != nil {
global.Log.Fatal("数据库连接失败", zap.Error(err)) global.Log.Fatal("数据库连接失败", zap.Error(err))
} }
@@ -33,7 +33,6 @@ var serverCmd = &cobra.Command{
global.Log.Fatal("初始化 Casbin 失败", zap.Error(err)) global.Log.Fatal("初始化 Casbin 失败", zap.Error(err))
} }
global.Log.Info("数据库初始化成功,正在启动 Web 服务...")
// 启动 Gin 服务 // 启动 Gin 服务
server.Run() server.Run()
}, },
+1 -1
View File
@@ -25,7 +25,7 @@ var userCreateCmd = &cobra.Command{
Use: "create", Use: "create",
Short: "交互式创建一个管理员用户", Short: "交互式创建一个管理员用户",
Run: func(cmd *cobra.Command, args []string) { Run: func(cmd *cobra.Command, args []string) {
err := db.Init(global.Config.Database.Driver, global.Config.Database.Dsn()) err := db.InitDB(global.Config.Database.Driver, global.Config.Database.Dsn())
if err != nil { if err != nil {
global.Log.Fatal("连接数据库失败", zap.Error(err)) global.Log.Fatal("连接数据库失败", zap.Error(err))
} }
+5 -2
View File
@@ -1,6 +1,7 @@
package db package db
import ( import (
"EnGin/internal/global"
"database/sql" "database/sql"
"fmt" "fmt"
"time" "time"
@@ -16,8 +17,8 @@ import (
// Client 跨包调用的统一数据库客户端实例 // Client 跨包调用的统一数据库客户端实例
var Client *ent.Client var Client *ent.Client
// Init 根据配置初始化数据库连接 // InitDB 根据配置初始化数据库连接
func Init(driver string, dsn string) error { func InitDB(driver string, dsn string) error {
// 先创建标准 sql.DB 以配置连接池 // 先创建标准 sql.DB 以配置连接池
db, err := sql.Open(driver, dsn) db, err := sql.Open(driver, dsn)
if err != nil { if err != nil {
@@ -37,6 +38,8 @@ func Init(driver string, dsn string) error {
// 将 sql.DB 包装为 ent 客户端 // 将 sql.DB 包装为 ent 客户端
drv := entsql.OpenDB(driver, db) drv := entsql.OpenDB(driver, db)
Client = ent.NewClient(ent.Driver(drv)) Client = ent.NewClient(ent.Driver(drv))
global.Log.Info("数据库初始化成功")
return nil return nil
} }
+1 -1
View File
@@ -21,7 +21,7 @@ func Run() {
// 注册路由 // 注册路由
router.RegisterRoutes(r) router.RegisterRoutes(r)
port := global.Config.Server.Port port := global.Config.Server.Port
global.Log.Info(fmt.Sprintf("Server is running on port %d", port)) global.Log.Info(fmt.Sprintf("服务运行在端口 %d", port))
if err := r.Run(fmt.Sprintf(":%d", port)); err != nil { if err := r.Run(fmt.Sprintf(":%d", port)); err != nil {
global.Log.Fatal("Web 服务启动失败", zap.Error(err)) global.Log.Fatal("Web 服务启动失败", zap.Error(err))
} }