service/vendor/github.com/qiniu/qmgo/field/default_field.go

55 lines
1.5 KiB
Go
Raw Normal View History

2023-12-21 22:17:40 +08:00
/*
Copyright 2020 The Qmgo Authors.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package field
import (
"time"
"go.mongodb.org/mongo-driver/bson/primitive"
)
// DefaultFieldHook defines the interface to change default fields by hook
type DefaultFieldHook interface {
DefaultUpdateAt()
DefaultCreateAt()
DefaultId()
}
// DefaultField defines the default fields to handle when operation happens
// import the DefaultField in document struct to make it working
type DefaultField struct {
Id primitive.ObjectID `bson:"_id"`
CreateAt time.Time `bson:"createAt"`
UpdateAt time.Time `bson:"updateAt"`
}
// DefaultUpdateAt changes the default updateAt field
func (df *DefaultField) DefaultUpdateAt() {
df.UpdateAt = time.Now().Local()
}
// DefaultCreateAt changes the default createAt field
func (df *DefaultField) DefaultCreateAt() {
if df.CreateAt.IsZero() {
df.CreateAt = time.Now().Local()
}
}
// DefaultId changes the default _id field
func (df *DefaultField) DefaultId() {
if df.Id.IsZero() {
df.Id = primitive.NewObjectID()
}
}