146 lines
4.0 KiB
Go
146 lines
4.0 KiB
Go
package mixin
|
|
|
|
import (
|
|
"time"
|
|
|
|
"entgo.io/ent"
|
|
"entgo.io/ent/schema/field"
|
|
"entgo.io/ent/schema/mixin"
|
|
)
|
|
|
|
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
var _ ent.Mixin = (*CreateTimestamp)(nil)
|
|
|
|
type CreateTimestamp struct{ mixin.Schema }
|
|
|
|
func (CreateTimestamp) Fields() []ent.Field {
|
|
return []ent.Field{
|
|
// 创建时间，毫秒
|
|
field.Int64("create_time").
|
|
Comment("创建时间").
|
|
Immutable().
|
|
Optional().
|
|
Nillable().
|
|
DefaultFunc(time.Now().UnixMilli),
|
|
}
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
var _ ent.Mixin = (*UpdateTimestamp)(nil)
|
|
|
|
type UpdateTimestamp struct{ mixin.Schema }
|
|
|
|
func (UpdateTimestamp) Fields() []ent.Field {
|
|
return []ent.Field{
|
|
// 更新时间，毫秒
|
|
// 需要注意的是，如果不是程序自动更新，那么这个字段不会被更新，除非在数据库里面下触发器更新。
|
|
field.Int64("update_time").
|
|
Comment("更新时间").
|
|
Optional().
|
|
Nillable().
|
|
UpdateDefault(time.Now().UnixMilli),
|
|
}
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
var _ ent.Mixin = (*DeleteTimestamp)(nil)
|
|
|
|
type DeleteTimestamp struct{ mixin.Schema }
|
|
|
|
func (DeleteTimestamp) Fields() []ent.Field {
|
|
return []ent.Field{
|
|
// 删除时间，毫秒
|
|
field.Int64("delete_time").
|
|
Comment("删除时间").
|
|
Optional().
|
|
Nillable(),
|
|
}
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
var _ ent.Mixin = (*Timestamp)(nil)
|
|
|
|
type Timestamp struct{ mixin.Schema }
|
|
|
|
func (Timestamp) Fields() []ent.Field {
|
|
var fields []ent.Field
|
|
fields = append(fields, CreateTimestamp{}.Fields()...)
|
|
fields = append(fields, UpdateTimestamp{}.Fields()...)
|
|
fields = append(fields, DeleteTimestamp{}.Fields()...)
|
|
return fields
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
var _ ent.Mixin = (*CreatedAtTimestamp)(nil)
|
|
|
|
type CreatedAtTimestamp struct{ mixin.Schema }
|
|
|
|
func (CreatedAtTimestamp) Fields() []ent.Field {
|
|
return []ent.Field{
|
|
// 创建时间，毫秒
|
|
field.Int64("created_at").
|
|
Comment("创建时间").
|
|
Immutable().
|
|
Optional().
|
|
Nillable().
|
|
DefaultFunc(time.Now().UnixMilli),
|
|
}
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
var _ ent.Mixin = (*UpdatedAtTimestamp)(nil)
|
|
|
|
type UpdatedAtTimestamp struct{ mixin.Schema }
|
|
|
|
func (UpdatedAtTimestamp) Fields() []ent.Field {
|
|
return []ent.Field{
|
|
// 更新时间，毫秒
|
|
// 需要注意的是，如果不是程序自动更新，那么这个字段不会被更新，除非在数据库里面下触发器更新。
|
|
field.Int64("updated_at").
|
|
Comment("更新时间").
|
|
Optional().
|
|
Nillable().
|
|
UpdateDefault(time.Now().UnixMilli),
|
|
}
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
var _ ent.Mixin = (*DeletedAtTimestamp)(nil)
|
|
|
|
type DeletedAtTimestamp struct{ mixin.Schema }
|
|
|
|
func (DeletedAtTimestamp) Fields() []ent.Field {
|
|
return []ent.Field{
|
|
// 删除时间，毫秒
|
|
field.Int64("deleted_at").
|
|
Comment("删除时间").
|
|
Optional().
|
|
Nillable(),
|
|
}
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
var _ ent.Mixin = (*TimestampAt)(nil)
|
|
|
|
type TimestampAt struct{ mixin.Schema }
|
|
|
|
func (TimestampAt) Fields() []ent.Field {
|
|
var fields []ent.Field
|
|
fields = append(fields, CreatedAtTimestamp{}.Fields()...)
|
|
fields = append(fields, UpdatedAtTimestamp{}.Fields()...)
|
|
fields = append(fields, DeletedAtTimestamp{}.Fields()...)
|
|
return fields
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|