snowflake的id传参换成字符串

zhangqian

2023-12-08 8e941bd782f574597f19c980982afade4dc11748

 model/model.go

@@ -2,53 +2,18 @@

import (
   "apsClient/pkg/snowflake"
   "encoding/json"
   "github.com/jinzhu/gorm"
   "strconv"
   "time"
)

type CommonModel struct {
   ID        uint   `gorm:"primary_key" json:"-"`
   IDStr     string `json:"ID" gorm:"-"`
   ID        uint `gorm:"primary_key" json:"ID,string"`
   CreatedAt time.Time
   UpdatedAt time.Time
   DeletedAt *time.Time `sql:"index"`
}

func (c *CommonModel) BeforeCreate(db *gorm.DB) {
func (c *CommonModel) BeforeCreate() {
   if c.ID == 0 {
      id := snowflake.GenerateID()
      if id < 0 {
         // 处理 ID 为负数的情况（可选）
         id = snowflake.GenerateID()
      }
      c.ID = uint(id)
      c.ID = uint(snowflake.GenerateID())
   }
}

func (c *CommonModel) UnmarshalJSON(b []byte) error {
   var data map[string]interface{}
   if err := json.Unmarshal(b, &data); err != nil {
      return err
   }

   if idStr, ok := data["ID"].(string); ok {
      id, err := strconv.ParseUint(idStr, 10, 64)
      if err != nil {
         return err
      }
      c.ID = uint(id)
      c.IDStr = idStr
   }

   return nil
}

func (c *CommonModel) MarshalJSON() ([]byte, error) {
   if c.ID != 0 && c.IDStr == "" {
      c.IDStr = strconv.FormatUint(uint64(c.ID), 10)
   }

   return json.Marshal(c)
}