refactor(db): 重命名Init为InitDB并集中初始化日志
This commit is contained in:
1 parent
1bca4808e9
commit
17d6669acb
5 files changed
+9
-7
No files matched your search
@@ -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))
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -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
@@ -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
@@ -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
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -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))
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in new issue
Block a user