129 lines
4.8 KiB
Go
129 lines
4.8 KiB
Go
package datasource
|
||
|
||
import (
|
||
"servicebase/pkg/log"
|
||
"servicebase/pkg/repo"
|
||
"servicebase/pkg/utils"
|
||
"strings"
|
||
"time"
|
||
|
||
"github.com/spf13/viper"
|
||
"gorm.io/driver/mysql"
|
||
"gorm.io/gorm"
|
||
"gorm.io/gorm/schema"
|
||
"gorm.io/plugin/dbresolver"
|
||
)
|
||
|
||
type BaseModel struct {
|
||
ID int `gorm:"primaryKey" json:"id"`
|
||
CreateTime utils.Time `gorm:"type:datetime;index;not null" json:"createTime"`
|
||
}
|
||
|
||
func (model *BaseModel) BeforeCreate(tx *gorm.DB) (err error) {
|
||
model.CreateTime.Time = time.Now()
|
||
return
|
||
}
|
||
|
||
var DB *gorm.DB
|
||
|
||
type MySQLStarter struct {
|
||
}
|
||
|
||
func (s *MySQLStarter) Init() error {
|
||
InitMySQl()
|
||
return nil
|
||
}
|
||
|
||
func InitMySQl() {
|
||
dsn := viper.GetString("db.connectString")
|
||
log.InfoF("init db config with: %s", dsn)
|
||
var e error
|
||
if DB, e = gorm.Open(mysql.New(mysql.Config{
|
||
DriverName: "",
|
||
ServerVersion: "",
|
||
DSN: dsn,
|
||
Conn: nil,
|
||
SkipInitializeWithVersion: false,
|
||
DefaultStringSize: 255,
|
||
DefaultDatetimePrecision: nil,
|
||
DisableDatetimePrecision: true,
|
||
DontSupportRenameIndex: false,
|
||
DontSupportRenameColumn: true,
|
||
DontSupportForShareClause: false,
|
||
// DontSupportNullAsDefaultValue: false,
|
||
}), &gorm.Config{
|
||
SkipDefaultTransaction: false,
|
||
NamingStrategy: schema.NamingStrategy{
|
||
TablePrefix: "", // table name prefix, table for `User` would be `t_users`
|
||
SingularTable: true, // use singular table name, table for `User` would be `user` with this option enabled
|
||
NoLowerCase: false, // skip the snake_casing of names
|
||
NameReplacer: strings.NewReplacer("PID", "pid"), // use name replacer to change struct/field name before convert it to db name
|
||
},
|
||
FullSaveAssociations: false,
|
||
Logger: &log.GLog{}, //
|
||
NowFunc: func() time.Time {
|
||
return time.Now().Local()
|
||
},
|
||
DryRun: false,
|
||
PrepareStmt: false, // 执行任何 SQL 时都会创建一个 prepared statement 并将其缓存，以提高后续的效率
|
||
DisableAutomaticPing: false, // 在完成初始化后，GORM 会自动 ping 数据库以检查数据库的可用性
|
||
DisableForeignKeyConstraintWhenMigrating: true, // 在 AutoMigrate 或 CreateTable 时，GORM 会自动创建外键约束，若要禁用该特性，可将其设置为 true
|
||
DisableNestedTransaction: false, // 嵌套事务
|
||
AllowGlobalUpdate: false, // true = 启用全局 update/delete
|
||
QueryFields: false,
|
||
CreateBatchSize: 0,
|
||
ClauseBuilders: nil,
|
||
ConnPool: nil,
|
||
Dialector: nil,
|
||
Plugins: nil,
|
||
}); e != nil {
|
||
panic("failed to connect database")
|
||
}
|
||
|
||
_ = DB.Use(dbresolver.Register(dbresolver.Config{
|
||
Replicas: []gorm.Dialector{mysql.Open(dsn)},
|
||
Policy: dbresolver.RandomPolicy{},
|
||
}).Register(dbresolver.Config{
|
||
Replicas: []gorm.Dialector{mysql.Open(dsn)},
|
||
} /* &model.User{} */))
|
||
|
||
sqlDB, _ := DB.DB()
|
||
// SetMaxIdleConns 用于设置连接池中空闲连接的最大数量。
|
||
sqlDB.SetMaxIdleConns(20)
|
||
// SetMaxOpenConns 设置打开数据库连接的最大数量。
|
||
sqlDB.SetMaxOpenConns(100)
|
||
// SetConnMaxLifetime 设置了连接可复用的最大时间。
|
||
sqlDB.SetConnMaxLifetime(time.Hour * 24)
|
||
|
||
// log.InfoF("gorm metrics at: %d", viper.GetUint32("db.metrics.port"))
|
||
// _ = DB.Use(prometheus.New(prometheus.Config{
|
||
// DBName: "db_data-front", // 使用 `DBName` 作为指标 label
|
||
// RefreshInterval: 10, // 指标刷新频率（默认为 15 秒）
|
||
// PushAddr: "", // 如果配置了 `PushAddr`，则推送指标
|
||
// StartServer: false, // 启用一个 http 服务来暴露指标
|
||
// //HTTPServerPort: viper.GetUint32("db.metrics.port"), // 配置 http 服务监听端口，默认端口为 8080 （如果您配置了多个，只有第一个 `HTTPServerPort` 会被使用）
|
||
// MetricsCollector: []prometheus.MetricsCollector{
|
||
// &prometheus.MySQL{
|
||
// // 指标名前缀，默认为 `gorm_status_`
|
||
// // 例如： Threads_running 的指标名就是 `gorm_status_Threads_running`
|
||
// Prefix: "gorm_status_",
|
||
// // 拉取频率，默认使用 Prometheus 的 RefreshInterval
|
||
// Interval: 30,
|
||
// // 从 SHOW STATUS 选择变量变量，如果不设置，则使用全部的状态变量
|
||
// VariableNames: []string{"Threads_running"},
|
||
// },
|
||
// }, // 用户自定义指标
|
||
// }))
|
||
|
||
//_ = DB.Use(tracing.NewPlugin(tracing.WithoutMetrics()))
|
||
|
||
autoMigrate()
|
||
|
||
repo.SetDefault(DB)
|
||
}
|
||
|
||
func autoMigrate() {
|
||
//_ = DB.AutoMigrate(&model.User{})
|
||
//_ = DB.AutoMigrate(&model.LeaseType{})
|
||
}
|