basic.go
851 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
package models
import (
"pro2d/components/db"
)
type ModelBaseMgo struct {
MonGo *db.MgoPool
pri interface{}
schema interface{}
}
func NewModelBaseMgo(mgo *db.Database, collection string, pri interface{}, schema interface{}) *ModelBaseMgo{
return &ModelBaseMgo{
MonGo: db.NewMongoPool(mgo, collection),
pri: pri,
schema: schema,
}
}
func (base *ModelBaseMgo) Load() error{
r := base.MonGo.FindOne(base.pri)
err := r.Decode(base.schema)
if err != nil {
return err
}
return nil
}
func (base *ModelBaseMgo) Create() {
base.MonGo.InsertOne(base.schema)
}
func (base *ModelBaseMgo) Index(key string) {
base.MonGo.Index(key)
}
func (base *ModelBaseMgo) Update(update interface{}) {
base.MonGo.FindOneAndUpdate(base.pri, update)
}
func (base *ModelBaseMgo) Save() {
base.MonGo.FindOneAndUpdate(base.pri, base.schema)
}