时间戳

首页 > 解决方案 > 使用自定义格式将结构的 time.Time 字段编组为 JSON 的通用方法

json - 使用自定义格式将结构的 time.Time 字段编组为 JSON 的通用方法

问题描述

我的数据模型定义了多个结构，它们都有两个共同的字段： aStartDate和 an EndDate。我需要将这两个字段格式化为2018-09-21编组 JSON 中的格式，因此结构实现了Marshaller接口：

type Results struct {
    Source     string    `json:"source"`
    StartDate  time.Time 
    EndDate    time.Time 
}

type WeightedResults struct {
    Source          string           `json:"source"`
    StartDate       time.Time        
    EndDate         time.Time        
}

func (r Results) MarshalJSON() ([]byte, error) {
    type Alias Results
    if equalDate(r.StartDate, r.EndDate) {
        return json.Marshal(&struct {
            Date string `json:"date"`
            Alias
        }{
            Date:  r.StartDate.Format(dateFormat),
            Alias: (Alias)(r),
        })
    }    
    return json.Marshal(&struct {
        StartDate string `json:"start_date"`
        EndDate   string `json:"end_date"`
        Alias
    }{
        StartDate: r.StartDate.Format("2006-01-02"),
        EndDate:   r.EndDate.Format("2006-01-02"),
        Alias:     (Alias)(r),
    })
}

func (r WeightedResults) MarshalJSON() ([]byte, error) {
    type Alias WeightedResults
    if equalDate(r.StartDate, r.EndDate) {
        return json.Marshal(&struct {
            Date string `json:"date"`
            Alias
        }{
            Date:  r.StartDate.Format(dateFormat),
            Alias: (Alias)(r),
        })
    } 
    return json.Marshal(&struct {
        StartDate string `json:"start_date"`
        EndDate   string `json:"end_date"`
        Alias
    }{
        StartDate: r.StartDate.Format("2006-01-02"),
        EndDate:   r.EndDate.Format("2006-01-02"),
        Alias:     (Alias)(r),
    })
}

上面的解决方案工作正常，但会产生大量代码重复。有没有办法重构两个实现MarshalJSON以使用相同的逻辑/代码？我很清楚 Go 不提供泛型（还），但必须有另一种方法来解决这个问题，对吧？

标签： jsongenericsgomarshalling

解决方案

您的自定义封送拆收器不应位于结构上，而应位于嵌入的自定义类型上time.Time：

type MyTime struct {
    time.Time
}

func (t MyTime) MarshalJSON() ([]byte, error) {
    return json.Marshal(t.Format("2006-01-02"))
}

然后在任何你想要的地方使用这种类型。

type Results struct {
    Source     string    `json:"source"`
    StartDate  MyTime
    EndDate    MyTime
}

推荐阅读