This commit is contained in:
2023-06-13 17:08:48 +08:00
parent 588ae8ebdc
commit 71f682d99a
17 changed files with 202 additions and 58 deletions
+2 -2
View File
@@ -2,8 +2,8 @@ package model
// AdminMenuApi 后台-菜单-接口表
type AdminMenuApi struct {
MenuId int64 `gorm:"column:menu_id;type:bigint(19);primary_key;comment:菜单id" json:"menu_id"`
ApiId int64 `gorm:"column:api_id;type:bigint(19);primary_key;comment:接口id" json:"api_id"`
MenuId int64 `gorm:"column:menu_id;type:bigint(19);comment:菜单id" json:"menu_id"`
ApiId int64 `gorm:"column:api_id;type:bigint(19);comment:接口id" json:"api_id"`
Menu AdminMenu `gorm:"foreignkey:MenuId;association_foreignkey:MenuID"`
API AdminAPI `gorm:"foreignkey:ApiId;association_foreignkey:APIID"`
}
+12
View File
@@ -12,3 +12,15 @@ type AdminRole struct {
func (m *AdminRole) TableName() string {
return "gdxz_admin_role"
}
// func (a *AdminRole) BeforeCreate(tx *gorm.DB) (err error) {
// node, err := snowflake.NewNode(config.C.Snowflake)
// if err != nil {
// return err
// }
// tx.Statement.SetColumn("RoleId", node.Generate().Int64())
//
// a.Model.CreatedAt = LocalTime(time.Now())
// tx.Statement.SetColumn("CreatedAt", a.Model.CreatedAt)
// return
// }
+52 -21
View File
@@ -5,7 +5,6 @@ import (
"errors"
"fmt"
"gorm.io/gorm"
"reflect"
"strings"
"time"
)
@@ -59,26 +58,26 @@ func (t *LocalTime) String() string {
}
// BeforeCreate 注册 BeforeCreate 回调函数
func (m *Model) BeforeCreate(tx *gorm.DB) (err error) {
// 动态访问 YourModel 结构体本身
model := tx.Statement.Dest
// 设置创建时间
layout := "2006-01-02 15:04:05"
strTime := "2019-08-09 11:35:52"
parsedTime, err := time.Parse(layout, strTime)
if err != nil {
return err
}
// 使用反射设置创建时间字段
createdAtField := reflect.ValueOf(model).Elem().FieldByName("CreatedAt")
if createdAtField.CanSet() {
createdAtField.Set(reflect.ValueOf(parsedTime))
}
return nil
}
// func (m *Model) BeforeCreate(tx *gorm.DB) (err error) {
// // 动态访问 YourModel 结构体本身
// model := tx.Statement.Dest
//
// // 设置创建时间
// layout := "2006-01-02 15:04:05"
// strTime := "2019-08-09 11:35:52"
// parsedTime, err := time.Parse(layout, strTime)
// if err != nil {
// return err
// }
//
// // 使用反射设置创建时间字段
// createdAtField := reflect.ValueOf(model).Elem().FieldByName("CreatedAt")
// if createdAtField.CanSet() {
// createdAtField.Set(reflect.ValueOf(parsedTime))
// }
//
// return nil
// }
func (m *Model) BeforeUpdate(tx *gorm.DB) (err error) {
m.UpdatedAt = LocalTime(time.Now())
@@ -86,3 +85,35 @@ func (m *Model) BeforeUpdate(tx *gorm.DB) (err error) {
return nil
}
// BeforeCreate 注册 BeforeCreate 回调函数
func (m *Model) BeforeCreate(tx *gorm.DB) (err error) {
IdName := ""
field := tx.Statement.Schema.Fields
for _, v := range field {
if v.PrimaryKey {
IdName = v.Name
}
}
if IdName == "" {
return errors.New("主键错误")
}
// 动态访问 YourModel 结构体本身
model := tx.Statement.Dest
// primaryField := reflect.ValueOf(model).Elem().FieldByName(IdName)
//
// if primaryField.CanSet() {
// primaryField.SetColumn(reflect.ValueOf(parsedTime))
// }
fmt.Println(model)
m.CreatedAt = LocalTime(time.Now())
tx.Statement.SetColumn("CreatedAt", m.CreatedAt)
m.UpdatedAt = LocalTime(time.Now())
tx.Statement.SetColumn("UpdatedAt", m.UpdatedAt)
return nil
}