修改1
This commit is contained in:
@@ -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,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
@@ -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
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user