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