golang在嵌套结构中解码JSON请求并将其作为blob插入DB

一尘不染

golang在嵌套结构中解码JSON请求并将其作为blob插入DB

我有一个用于解码JSON请求的嵌套结构。

type Service struct {
    ID       string `json:"id,omitempty" db:"id"`
    Name     string `json:"name" db:"name"`
    Contract struct {
        ServiceTime int    `json:"service_time"`
        Region      string `json:"region"`
    } `json:"contract" db:"contract"`
}

我正在使用Blob类型将Contract存储在MySQL中。为了使其工作，我将需要使用Contract作为字符串创建一个不同的结构以插入DB中。可以仅通过使用单个结构以更好的方式完成此操作，还是有其他优雅的方法？

阅读 215

2020-07-02

共1个答案

一尘不染

这取决于您用来与数据库进行对话的内容，但是如果您正在使用database/sql并且提供了对此提供支持的驱动程序，则可以让您的Contract类型实现Valuer接口。

type Service struct {
    ID       string    `json:"id,omitempty" db:"id"`
    Name     string    `json:"name" db:"name"`
    Contract Contract `json:"contract" db:"contract"`
}

type Contract struct {
    ServiceTime int    `json:"service_time"`
    Region      string `json:"region"`
}

func (c *Contract) Value() (driver.Value, error) {
    if c != nil {
        b, err := json.Marshal(c)
        if err != nil {
            return nil, err
        }
        return string(b), nil
    }
    return nil, nil
}

然后，当您执行查询时，只需在Service.Contract字段中传递，驱动程序应调用该Value方法。

sql := // INSERT INTO ...
svc := // &Service{ ...
db.Exec(sql, svc.ID, svc.Name, svc.Contract)

为了能够从db中检索blob并将其解组，Contract您可以再次实现Scanner接口，如果类型实现了该目标，则驱动程序应该调用它。

func (c *Contract) Scan(src interface{}) error {
    var data []byte
    if b, ok := src.([]byte); ok {
        data = b
    } else if s, ok := src.(string); ok {
        data = []byte(s)
    }
    return json.Unmarshal(data, c)
}

// ...

sql := // SELECT * FROM ...
svc := // &Service{ ...
db.QueryRow(sql, 123).Scan(&svc.ID, &svc.Name, &svc.Contract)

2020-07-02