新增数据库主键处理

api/model/adminRole.go

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