120 lines
2.8 KiB
Go
120 lines
2.8 KiB
Go
package model
|
|
|
|
import (
|
|
"database/sql/driver"
|
|
"errors"
|
|
"fmt"
|
|
"gorm.io/gorm"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
type Model struct {
|
|
CreatedAt LocalTime `gorm:"column:created_at;type:datetime;comment:创建时间" json:"created_at"`
|
|
UpdatedAt LocalTime `gorm:"column:updated_at;type:datetime;comment:修改时间" json:"updated_at"`
|
|
}
|
|
|
|
// LocalTime 自定义数据类型
|
|
type LocalTime time.Time
|
|
|
|
func (t *LocalTime) UnmarshalJSON(data []byte) error {
|
|
if string(data) == "null" {
|
|
return nil
|
|
}
|
|
var err error
|
|
// 前端接收的时间字符串
|
|
str := string(data)
|
|
// 去除接收的str收尾多余的"
|
|
timeStr := strings.Trim(str, "\"")
|
|
t1, err := time.Parse("2006-01-02 15:04:05", timeStr)
|
|
*t = LocalTime(t1)
|
|
return err
|
|
}
|
|
|
|
func (t LocalTime) MarshalJSON() ([]byte, error) {
|
|
formatted := fmt.Sprintf("\"%v\"", time.Time(t).Format("2006-01-02 15:04:05"))
|
|
return []byte(formatted), nil
|
|
}
|
|
|
|
func (t LocalTime) Value() (driver.Value, error) {
|
|
// MyTime 转换成 time.Time 类型
|
|
tTime := time.Time(t)
|
|
return tTime.Format("2006-01-02 15:04:05"), nil
|
|
}
|
|
|
|
func (t *LocalTime) Scan(v interface{}) error {
|
|
switch vt := v.(type) {
|
|
case time.Time:
|
|
// 字符串转成 time.Time 类型
|
|
*t = LocalTime(vt)
|
|
default:
|
|
return errors.New("类型处理错误")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (t *LocalTime) String() string {
|
|
return fmt.Sprintf("hhh:%s", time.Time(*t).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) BeforeUpdate(tx *gorm.DB) (err error) {
|
|
m.UpdatedAt = LocalTime(time.Now())
|
|
tx.Statement.SetColumn("UpdatedAt", m.UpdatedAt)
|
|
|
|
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
|
|
}
|