新增数据库主键处理

api/model/area.go

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