新增数据库主键处理

api/model/adminUser.go

@@ -1,5 +1,11 @@
 package model
 
+import (
+	"github.com/bwmarrin/snowflake"
+	"gorm.io/gorm"
+	"hospital-admin-api/config"
+)
+
 // AdminUser 后台-用户表
 type AdminUser struct {
 	Model
@@ -28,3 +34,17 @@ type AdminUser struct {
 func (m *AdminUser) TableName() string {
 	return "gdxz_admin_user"
 }
+
+func (m *AdminUser) 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
+}