使用 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 of their type when a new struct value is created (unless you use a composite literal，您可以在其中明确指定不同的值）。

所以一个选择是创建一个 constructor-like 函数 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类型没有这个方法，所以不会发生无穷无尽的"recursion"（type关键字创建了一个新类型，它没有底层的"inherit"方法类型）。

请注意，此解决方案也会覆盖（re-set）CreatedAt 字段，即使您只想 re-save 一个 User。所以我们应该添加一个检查 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

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

Set default date when inserting document with time.Time field

time

struct

go

mongodb

mgo