fix

zhangqian

2023-12-08 2dea693b0d78246e0db41e8e2daaf6ad9cc3ed8c

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
package model
 
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:"-"`
    CreatedAt time.Time
    UpdatedAt time.Time
    DeletedAt *time.Time `sql:"index"`
}
 
func (c *CommonModel) BeforeCreate(db *gorm.DB) {
    if c.ID == 0 {
        id := snowflake.GenerateID()
        if id < 0 {
            // 处理 ID 为负数的情况（可选）
            id = snowflake.GenerateID()
        }
        c.ID = uint(id)
    }
}
 
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)
}