update time

.gitignore

@@ -4,5 +4,5 @@
 .idea
 .vscode
 *.log
-gitmod.sh
+gomod.sh
 *_test.go

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 // 驱动

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
+}

gomod.sh

@@ -1,3 +0,0 @@
-go get -u && go mod tidy
-go get -u all
-go mod vendor