在推入 mongodb 时自动填充 golang struct 中的 created_at 和 updated_at

发布于04月17日

type User struct {
    ID           primitive.ObjectID `bson:"_id,omitempty"`
    CreatedAt    time.Time          `bson:"created_at"`
    UpdatedAt    time.Time          `bson:"updated_at"`
    Name         string             `bson:"name"`
}

user := User{Name: "username"}

client.Database("db").Collection("collection").InsertOne(context.Background(), user)

如何在golang的mongodb(仅限mongodb驱动程序)中使用上述代码中自动创建的_at和更新的_at？目前，它将为在创建的_和在更新的_设置零时间(0001-01-01T00:00:00.000+00:00).

推荐答案

MongoDB服务器不支持这一点.

您可以实现一个自定义封送拆收器，在其中可以根据自己的喜好更新这些字段.实现bson.Marshaler，当保存*User类型的值时，将调用MarshalBSON()函数.

这就是它的样子:

func (u *User) MarshalBSON() ([]byte, error) {
    if u.CreatedAt.IsZero() {
        u.CreatedAt = time.Now()
    }
    u.UpdatedAt = time.Now()
    
    type my User
    return bson.Marshal((*my)(u))
}

请注意，该方法具有指针接收器，因此请使用指向您的值的指针:

user := &User{Name: "username"}


c := client.Database("db").Collection("collection")
if _, err := c.InsertOne(context.Background(), user); err != nil {
    // handle error
}

my类型的目的是避免堆栈溢出.