mongodb - 使用 time.Time 字段插入文档时设置默认日期

Question

在 mongoose(node.js) 中，我可以定义一个带有默认 Date.now 的模型架构，如下所示:

...
type: Date,
default: Date.now
...

如何在每次使用 mgo 创建文档时都不必插入 time.Time 来实现相同的目的？

type User struct {
    CreatedAt   time.Time `json:"created_at" bson:"created_at"` // Make this field filled automatically with time.Now() every time a document of this `struct` is inserted
}

Answer 1

在 Go 中你不能为字段定义默认值，它们总是 zero-value在创建新结构值时它们的类型(除非您使用 composite literal，您可以在其中显式提供不同的值)。

因此，一种选择是创建一个类似构造函数的函数 NewUser() 来设置该字段，并始终使用该函数来创建新用户:

func NewUser() *User {
    return &User{
        CreatedAt: time.Now(),
    }
}

当然这不能强制，而且这将保存 User 结构值创建的时间戳，而不是保存时的时间戳。

另一种更好的方法是使用自定义编码逻辑。

您可以通过实现 bson.Getter 来编写自定义编码逻辑. GetBSON() 负责提供实际保存的值。我们希望保存相同的 User 实例，只是之前设置的 CreatedAt 字段:

type User struct {
    CreatedAt time.Time `json:"created_at" bson:"created_at"`
}

func (u *User) GetBSON() (interface{}, error) {
    u.CreatedAt = time.Now()
    type my *User
    return my(u), nil
}

请注意，创建并返回了一个新的 my 类型。这样做的原因是为了避免堆栈溢出。简单地返回一个 *User 类型的值是不好的，因为它实现了 bson.Getter，所以 GetBSON() 会被无休止地调用。新的 my 类型没有这个方法，因此不会发生无休止的“递归”( type 关键字创建一个新类型，它不会“继承”底层类型)。

请注意，即使您只想重新保存一个 User，此解决方案也会覆盖(重新设置)CreatedAt 字段)。所以我们应该添加一个检查 CreatedAt 字段是否被填充，并且只有当它是零值时才设置它:

func (u *User) GetBSON() (interface{}, error) {
    if u.CreatedAt.IsZero() {
        u.CreatedAt = time.Now()
    }
    type my *User
    return my(u), nil
}

另见相关/类似问题:Accesing MongoDB from Go