beego 时间类型json格式问题

beego 时间类型json格式化问题
再使用beego的orm框架时,如果希望重新定义time.Time转json的格式,我们可以重写time.Time类型,这样,就不用每次返回给客户端的时候去手动转了,代码如下

package lib

import (
    "encoding/json"
    "fmt"
    "time"

    "github.com/astaxie/beego/orm"
)

type Time struct {
    time.Time
}

// MarshalJSON 序列化为JSON
func (t Time) MarshalJSON() ([]byte, error) {
    if t.IsZero() {
        return []byte("\"\""), nil
    }
    stamp := fmt.Sprintf("\"%s\"", t.Format("2006-01-02 15:04:05"))
    return []byte(stamp), nil
}

// UnmarshalJSON 反序列化为JSON
func (t *Time) UnmarshalJSON(data []byte) error {
    var err error
    t.Time, err = time.Parse("2006-01-02 15:04:05", string(data)[1:20])
    return err
}

// String 重写String方法
func (t *Time) String() string {
    data, _ := json.Marshal(t)
    return string(data)
}

// FieldType 数据类型
func (t *Time) FieldType() int {
    return orm.TypeDateTimeField

}

// SetRaw 读取数据库值
func (t *Time) SetRaw(value interface{}) error {
    switch value.(type) {
    case time.Time:
        t.Time = value.(time.Time)
    }
    return nil
}

// RawValue 写入数据库
func (t *Time) RawValue() interface{} {
    str := t.Format("2006-01-02 15:04:05")
    if str == "0001-01-01 00:00:00" {
        return nil
    }
    return str
}

你只需要将默认的类型time.Time类型改为lib.Time就OK了

你可能感兴趣的:(beego 时间类型json格式问题)