catface_backend_go/app/model/base_model.go

91 lines
3.5 KiB
Go
Raw Normal View History

2024-10-14 13:49:16 +08:00
package model
import (
"catface/app/global/my_errors"
"catface/app/global/variable"
"fmt"
"strings"
"time"
"gorm.io/gorm"
2024-10-23 02:54:56 +08:00
"gorm.io/plugin/soft_delete"
2024-10-14 13:49:16 +08:00
)
type BaseModel struct {
*gorm.DB `gorm:"-" json:"-"`
Id int64 `gorm:"primaryKey" json:"id"`
CreatedAt *time.Time `json:"created_at,omitempty" gorm:""` //日期时间字段统一设置为字符串即可
UpdatedAt *time.Time `json:"updated_at,omitempty" gorm:""` // TIP 设为 *time.Timeomitempty 和 autoUpdated 就都可以生效
//DeletedAt gorm.DeletedAt `json:"deleted_at"` // 如果开发者需要使用软删除功能打开本行注释掉的代码即可同时需要在数据库的所有表增加字段deleted_at 类型为 datetime
}
type BriefModel struct {
Id int64 `json:"id"`
NameZh string `json:"name_zh"`
NameEn string `json:"name_en"`
}
2024-10-23 02:54:56 +08:00
type Color struct {
ColorFont string `json:"color_font" gorm:"type:char(10)"`
ColorBackground string `json:"color_background" gorm:"type:char(10)"`
}
type DeletedAt struct {
DeletedAt *time.Time `json:"deleted_at" gorm:"defalt:NULL"`
2024-10-23 12:08:05 +08:00
IsDel soft_delete.DeletedAt `gorm:"softDelete:flag,DeletedAtField:DeletedAt;default:0"`
2024-10-23 02:54:56 +08:00
}
2024-10-14 13:49:16 +08:00
func UseDbConn(sqlType string) *gorm.DB {
var db *gorm.DB
sqlType = strings.Trim(sqlType, " ")
if sqlType == "" {
sqlType = variable.ConfigGormv2Yml.GetString("Gormv2.UseDbType")
}
switch strings.ToLower(sqlType) {
case "mysql":
if variable.GormDbMysql == nil {
variable.ZapLog.Fatal(fmt.Sprintf(my_errors.ErrorsGormNotInitGlobalPointer, sqlType, sqlType))
}
db = variable.GormDbMysql
case "sqlserver":
if variable.GormDbSqlserver == nil {
variable.ZapLog.Fatal(fmt.Sprintf(my_errors.ErrorsGormNotInitGlobalPointer, sqlType, sqlType))
}
db = variable.GormDbSqlserver
case "postgres", "postgre", "postgresql":
if variable.GormDbPostgreSql == nil {
variable.ZapLog.Fatal(fmt.Sprintf(my_errors.ErrorsGormNotInitGlobalPointer, sqlType, sqlType))
}
db = variable.GormDbPostgreSql
default:
variable.ZapLog.Error(my_errors.ErrorsDbDriverNotExists + sqlType)
}
return db
}
// 在 ginskeleton项目中如果在业务 model 设置了回调函数,请看以下说明
// 注意gorm 的自动回调函数BeforeCreate、BeforeUpdate 等),不是由本项目的 Create ... 函数先初始化然后调用的而是gorm自动直接调用的
// 所以 接收器 b 的所有参数都是没有赋值的,因此这里需要给 b.DB 赋予回调的 gormDb
// baseModel 的代码执行顺序晚于其他业务 model 的回调函数如果回调函数名称相同会被普通业务model的同名回调函数覆盖
// gorm 支持的自动回调函数清单https://github.com/go-gorm/gorm/blob/master/callbacks/interfaces.go
//func (b *BaseModel) BeforeCreate(gormDB *gorm.DB) error {
// 第一步必须反向将 gormDB 赋值给 b.DB
// b.DB = gormDB
// 后续的代码就可以像普通业务 model 一样操作,
// b.Exec(sql,参数1参数2...)
// b.Raw(sql,参数1参数2...)
// return nil
//}
// BeforeUpdate、BeforeSave 函数都会因为 更新类的操作而被触发
// 如果baseModel 和 普通业务 model 都想使用回调函数,那么请设置不同的回调函数名,例如:这里设置 BeforeUpdate、普通业务model 设置 BeforeSave 即可
//func (b *BaseModel) BeforeUpdate(gormDB *gorm.DB) error {
// 第一步必须反向将 gormDB 赋值给 b.DB
// b.DB = gormDB
// 后续的代码就可以像普通业务 model 一样操作,
// b.Exec(sql,参数1参数2...)
// b.Raw(sql,参数1参数2...)
// return nil
//}