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了