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

时间:2017-01-28 08:02:05

标签: mongodb go struct time mgo

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
}

1 个答案:

答案 0 :(得分:2)

在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类型没有这种方法,所以无限的#34;递归"不会发生(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
}

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