package models
import (
"errors"
"fmt"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/bsontype"
"go.mongodb.org/mongo-driver/bson/primitive"
"go.mongodb.org/mongo-driver/x/bsonx/bsoncore"
"time"
)
type JsonTime time.Time
const (
timeFormat = "2006-01-02 15:04:05"
)
//实现json反序列化,从传递的字符串中解析成时间对象
func (t *JsonTime) UnmarshalJSON(data []byte) (err error) {
now, err := time.ParseInLocation(`"`+timeFormat+`"`, string(data), time.Local)
*t = JsonTime(now)
return
}
//实现json序列化,将时间转换成字符串byte数组
func (t JsonTime) MarshalJSON() ([]byte, error) {
b := make([]byte, 0, len(timeFormat)+2)
b = append(b, '"')
b = time.Time(t).AppendFormat(b, timeFormat)
b = append(b, '"')
return b, nil
}
//mongodb是存储bson格式,因此需要实现序列化bsonvalue(这里不能实现MarshalBSON,MarshalBSON是处理Document的),将时间转换成mongodb能识别的primitive.DateTime
func (t *JsonTime) MarshalBSONValue() (bsontype.Type, []byte, error) {
targetTime := primitive.NewDateTimeFromTime(time.Time(*t))
return bson.MarshalValue(targetTime)
}
//实现bson反序列化,从mongodb中读取数据转换成time.Time格式,这里用到了bsoncore中的方法读取数据转换成datetime然后再转换成time.Time
func (t *JsonTime) UnmarshalBSONValue(t2 bsontype.Type, data []byte) error {
v, _, valid := bsoncore.ReadValue(data, t2)
if valid == false {
return errors.New(fmt.Sprintf("%s, %s, %s", "读取数据失败:", t2, data))
}
*t = JsonTime(v.Time())
return nil
}
type News struct {
Id primitive.ObjectID `json:"id,omitempty"`
CreateTime JsonTime `bson:"create_time" json:"create_time,omitempty" validate:"required"`
Title string `json:"title,omitempty" validate:"required"`
}
Go mongo TZ时间转换
9 min read