修改1

api/model/model.go

@@ -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
+}