debug

zhangqian

2023-12-08 c68a2d376f67d60277001e3f748fbd71c5201af3

 model/model.go

@@ -8,9 +8,10 @@
   "time"
)

type BigID uint

type CommonModel struct {
   ID        uint   `gorm:"primary_key;autoIncrement:false" json:"-"`
   IDStr     string `json:"ID" gorm:"-"`
   ID        BigID `gorm:"primary_key"`
   CreatedAt time.Time
   UpdatedAt time.Time
   DeletedAt *time.Time `sql:"index"`
@@ -20,21 +21,29 @@
   if c.ID == 0 {
      id := snowflake.GenerateID()
      if id < 0 {
         // 处理 ID 为负数的情况（可选）
         id = snowflake.GenerateID()
      }
      c.ID = uint(id)
      c.ID = BigID(id)
   }
}
func (c CommonModel) UnmarshalJSON(b []byte) (err error) {
   id, err := strconv.ParseUint(c.IDStr, 10, 64)

func (id *BigID) UnmarshalJSON(b []byte) error {
   var idString string
   if err := json.Unmarshal(b, &idString); err != nil {
      return err
   }

   idValue, err := strconv.ParseUint(idString, 10, 64)
   if err != nil {
      return err
   }
   c.ID = uint(id)
   return

   *id = BigID(idValue)
   return nil
}

func (c CommonModel) MarshalJSON() ([]byte, error) {
   c.IDStr = strconv.FormatUint(uint64(c.ID), 10)
   return json.Marshal(c)
func (id *BigID) MarshalJSON() ([]byte, error) {
   idString := strconv.FormatUint(uint64(*id), 10)
   return []byte(idString), nil
}