GO Mongodb时间处理

go在操作monogdb时间的时候如果使用time.Time时间格式，则前后端的时间显示是这样的：2021-01-28T06:38:43.622Z；这种格式很不友好（时区还不对），更多时候我们可能需要的是这种格式：2021-01-28 14:38:43；在网上查了一些资料，发现解决方案都不太好，有的是要手动做转换，有的是改变了mongodb的存储内容；我这里综合了一些资料给出以下解决方法（通过申明time.Time的扩展，重写json和bson序列化代码实现）：

序列化处理代码：

package common

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 (
	timeFormart = "2006-01-02 15:04:05"
)

//实现json反序列化，从传递的字符串中解析成时间对象
func (t *JsonTime) UnmarshalJSON(data []byte) (err error) {
	now, err := time.ParseInLocation(`"`+timeFormart+`"`, string(data), time.Local)
	*t = JsonTime(now)
	return
}

//实现json序列化，将时间转换成字符串byte数组
func (t JsonTime) MarshalJSON() ([]byte, error) {
	b := make([]byte, 0, len(timeFormart)+2)
	b = append(b, '"')
	b = time.Time(t).AppendFormat(b, timeFormart)
	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
}

在需要进行时间转换的struct中申明common.JsonTime

type User struct {
	Name string `bson:"name" json:"name"` 
	Type int `bson:"type" json:"type"` 
	Createtime common.JsonTime `bson:"createtime" json:"createtime"`
}

 

 

来源：oschina

链接：https://my.oschina.net/u/184206/blog/4931673