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}
}