25 lines
676 B
Go
25 lines
676 B
Go
package db
|
|
|
|
import (
|
|
bundleConfig "micro-bundle/config"
|
|
"strings"
|
|
|
|
"github.com/google/wire"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// 定义一个新的类型来区分TaskBenchDB
|
|
type TaskBenchDB struct {
|
|
*gorm.DB
|
|
}
|
|
|
|
var TaskBenchProvider = wire.NewSet(NewTaskBenchDB)
|
|
|
|
func NewTaskBenchDB() *TaskBenchDB {
|
|
connTaskBenchDB := strings.Join([]string{bundleConfig.Data.TaskBenchDB.User, ":", bundleConfig.Data.TaskBenchDB.Password,
|
|
"@tcp(", bundleConfig.Data.TaskBenchDB.Host, ":", bundleConfig.Data.TaskBenchDB.Port, ")/",
|
|
bundleConfig.Data.TaskBenchDB.DbName, "?charset=utf8mb4&parseTime=true&loc=Local"}, "")
|
|
db := loadTaskBenchMysqlConn(connTaskBenchDB)
|
|
return &TaskBenchDB{DB: db}
|
|
}
|