插入带有 time.Time 字段的文档时设置默认日期

bor*_*mke 4 time struct go mongodb mgo

mongoose( node.js) 中,我可以使用默认值定义模型架构,Date.now如下所示:

...
type: Date,
default: Date.now
...
Run Code Online (Sandbox Code Playgroud)

如何实现相同的效果,而不必在time.Time每次创建文档时插入mgo

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
}
Run Code Online (Sandbox Code Playgroud)

icz*_*cza 5

在 Go 中,您不能为字段定义默认值,当创建新的结构值时,它们将始终是其类型的零值(除非您使用复合文字,您可以明确给出不同的值)。

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

func NewUser() *User {
    return &User{
        CreatedAt: time.Now(),
    }
}
Run Code Online (Sandbox Code Playgroud)

当然这不能被强制,而且这将保存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
}
Run Code Online (Sandbox Code Playgroud)

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

请注意,CreatedAt即使您只想重新保存User. 所以我们应该添加一个检查CreatedAt字段是否被填充,并且只有在它是零值时才设置它:

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

另请参阅相关/类似问题:从 Go 访问 MongoDB