新增添加角色

api/model/adminRole.go

@@ -12,15 +12,3 @@ 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
-// }

api/model/model.go

@@ -4,7 +4,10 @@ import (
 	"database/sql/driver"
 	"errors"
 	"fmt"
+	"github.com/bwmarrin/snowflake"
 	"gorm.io/gorm"
+	"hospital-admin-api/config"
+	"reflect"
 	"strings"
 	"time"
 )
@@ -88,27 +91,28 @@ func (m *Model) BeforeUpdate(tx *gorm.DB) (err error) {
 
 // BeforeCreate 注册 BeforeCreate 回调函数
 func (m *Model) BeforeCreate(tx *gorm.DB) (err error) {
-	IdName := ""
+	// 动态添加表主键
+	primaryName := ""
 	field := tx.Statement.Schema.Fields
 	for _, v := range field {
 		if v.PrimaryKey {
-			IdName = v.Name
+			primaryName = v.Name
 		}
 	}
 
-	if IdName == "" {
-		return errors.New("主键错误")
+	if primaryName != "" {
+		// 动态访问 YourModel 结构体本身
+		model := tx.Statement.Dest
+		primaryValue := reflect.ValueOf(model).Elem().FieldByName(primaryName)
+		if primaryValue.IsZero() {
+			node, err := snowflake.NewNode(config.C.Snowflake)
+			if err != nil {
+				return errors.New("服务器错误")
+			}
+			tx.Statement.SetColumn(primaryName, node.Generate().Int64())
+		}
 	}
 
-	// 动态访问 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)