micro-bundle/pkg/db/bundleDB.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}
}