新增数据库主键处理

api/model/user.go

@@ -1,5 +1,11 @@
 package model
 
+import (
+	"github.com/bwmarrin/snowflake"
+	"gorm.io/gorm"
+	"hospital-admin-api/config"
+)
+
 // User 用户主表
 type User struct {
 	UserId         int64     `gorm:"column:user_id;type:bigint(19);primary_key;comment:主键" json:"user_id"`
@@ -24,3 +30,17 @@ type User struct {
 func (m *User) TableName() string {
 	return "gdxz_user"
 }
+
+func (m *User) BeforeCreate(tx *gorm.DB) error {
+	if m.UserId == 0 {
+		// 创建雪花算法实例
+		sf, err := snowflake.NewNode(config.C.Snowflake)
+		if err != nil {
+			return err
+		}
+
+		// 生成新的雪花算法 ID
+		m.UserId = sf.Generate().Int64()
+	}
+	return nil
+}