一尘不染

在插入带有时间的文档时设置默认日期。时间字段

go

mongoosenode.js)中,我可以定义一个默认的模型架构,Date.now如下所示:

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

如何实现相同的目的而不必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
}

阅读 199

收藏
2020-07-02

共1个答案

一尘不染

在Go中,您无法定义字段的默认值,当创建新的struct值时,它们将始终是其类型的零值(除非您使用复合文字,可以在其中显式给出其他值)。

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

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

当然,不能强制这样做,这也将保留User结构值创建的时间戳,而不是保存它的时间。

另一种更好的方法是使用自定义封送处理逻辑。

您可以通过实现编写自定义封送处理逻辑bson.GetterGetBSON()负责提供将实际保存的值。我们希望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创建并返回一个新类型。这样做的原因是为了避免堆栈溢出。简单地返回type的值*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
}
2020-07-02