dorm/mongo_time.go

package dorm

import (
	"go.dtapp.net/gotime"
	"go.mongodb.org/mongo-driver/bson"
	"go.mongodb.org/mongo-driver/bson/bsontype"
	"time"
)

// BsonTime 时间类型
type BsonTime time.Time

// MarshalJSON 实现json序列化
func (bt BsonTime) MarshalJSON() ([]byte, error) {

	b := make([]byte, 0)
	b = append(b, '"')
	b = append(b, gotime.SetCurrent(time.Time(bt)).Bson()...)
	b = append(b, '"')

	return b, nil
}

// UnmarshalJSON 实现json反序列化
func (bt *BsonTime) UnmarshalJSON(data []byte) (err error) {

	if string(data) == "null" {
		return nil
	}

	bsonTime := gotime.SetCurrentParse(string(data))
	*bt = BsonTime(bsonTime.Time)

	return nil
}

// Time 转时间
func (bt BsonTime) Time() time.Time {
	return gotime.SetCurrent(time.Time(bt)).Time
}

// Format 转时间字符串
func (bt BsonTime) Format() string {
	return gotime.SetCurrent(time.Time(bt)).Format()
}

// TimePro 转时间操作
func (bt BsonTime) TimePro() gotime.Pro {
	return gotime.SetCurrent(time.Time(bt))
}

// NewBsonTimeCurrent 创建当前时间
func NewBsonTimeCurrent() BsonTime {
	return BsonTime(gotime.Current().Time)
}

// NewBsonTimeFromTime 创建某个时间
func NewBsonTimeFromTime(t time.Time) BsonTime {
	return BsonTime(t)
}

// NewBsonTimeFromString 创建某个时间 字符串
func NewBsonTimeFromString(t string) BsonTime {
	return BsonTime(gotime.SetCurrentParse(t).Time)
}

// Value 时间类型
func (bt BsonTime) Value() string {
	return gotime.SetCurrent(time.Time(bt)).Bson()
}

// MarshalBSONValue 实现bson序列化
func (bt BsonTime) MarshalBSONValue() (bsontype.Type, []byte, error) {
	return bson.MarshalValue(gotime.SetCurrent(time.Time(bt)).Bson())
}

// UnmarshalBSONValue 实现bson反序列化
func (bt *BsonTime) UnmarshalBSONValue(t bsontype.Type, data []byte) error {
	t1 := gotime.SetCurrentParse(string(data))
	//if string(data) == "" {
	//	return errors.New(fmt.Sprintf("%s, %s, %s", "读取数据失败:", t, data))
	//}
	*bt = BsonTime(t1.Time)
	return nil
}