diff --git a/.gitignore b/.gitignore index 34509e3..23a3a17 100644 --- a/.gitignore +++ b/.gitignore @@ -4,5 +4,5 @@ .idea .vscode *.log -gitmod.sh +gomod.sh *_test.go \ No newline at end of file diff --git a/app.go b/app.go index 01472d4..0fd6217 100644 --- a/app.go +++ b/app.go @@ -8,7 +8,7 @@ import ( "go.mongodb.org/mongo-driver/mongo/options" ) -const Version = "1.0.1" +const Version = "1.0.2" type Client struct { Db *mongo.Client // 驱动 diff --git a/common.go b/common.go new file mode 100644 index 0000000..4375888 --- /dev/null +++ b/common.go @@ -0,0 +1,51 @@ +package gomongo + +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" +) + +// https://chowdera.com/2021/01/20210128154417121y.html + +type JsonTime time.Time + +const ( + timeFormart = "2006-01-02 15:04:05" +) + +// UnmarshalJSON 实现json反序列化,从传递的字符串中解析成时间对象 +func (t *JsonTime) UnmarshalJSON(data []byte) (err error) { + now, err := time.ParseInLocation(`"`+timeFormart+`"`, string(data), time.Local) + *t = JsonTime(now) + return +} + +// MarshalJSON 实现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 +} + +// MarshalBSONValue 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) +} + +// UnmarshalBSONValue 实现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 +} diff --git a/gomod.sh b/gomod.sh deleted file mode 100644 index c03f9f0..0000000 --- a/gomod.sh +++ /dev/null @@ -1,3 +0,0 @@ -go get -u && go mod tidy -go get -u all -go mod vendor \ No newline at end of file