go_dreamfactory/sys/db/model.go

85 lines
2.2 KiB
Go

package db
import (
"errors"
"go_dreamfactory/lego/core"
"go_dreamfactory/lego/sys/log"
"go.mongodb.org/mongo-driver/bson"
)
type Autogenerated struct {
ID string `json:"_id"`
UID string `json:"uid"`
Act string `json:"act"` // insert update delete
D []interface{}
}
// type data struct {
// Table string
// Wheremap [map[string]]interface{} // 如果是insert 条件就是nil del 只有条件
// Modifymap map[string]map[string]interface{}
// insertData []interface{}
// }
type QueryStruct struct {
Selector bson.M
Query bson.M
}
const (
DB_ModelTable core.SqlTable = "model"
)
type IModel interface {
Model_UpdateDBBylog() (err error) // 读取日志并更新对应的表
Model_InsertDBBylog(data *Autogenerated) (err error) // 插入日志
Model_UpdateUserDataByUid(uid string) (err error) // 读取指定玩家信息到db中
}
func (this *DB) Mail_UpdateDBBylog() (err error) {
data := &Autogenerated{}
this.mgo.FindOneAndDelete(DB_ModelTable, bson.M{}).Decode(data) // 查找最新的一条并且删除
log.Debugf("======= insert log : %+v =======", data)
if data.Act == "insert" {
query := data.D[1].([]interface{})
_, err := this.mgo.InsertMany(data.D[0].(core.SqlTable), query)
if err != nil {
log.Errorf("insert %s db err:%v", data.D[0].(core.SqlTable), err)
}
} else if data.Act == "delete" {
_, err := this.mgo.DeleteMany(data.D[0].(core.SqlTable), data.D[1].(map[string]interface{}))
if err != nil {
log.Errorf("delete %s db err:%v", data.D[0].(core.SqlTable), err)
}
} else { // update
if len(data.D) < 2 {
log.Errorf("parameter len _id : %s,uid : %s d.len:%v", data.ID, data.UID, len(data.D))
return errors.New("parameter len err")
}
where := data.D[1].(map[string]interface{})
_obj := &QueryStruct{}
for k, v := range where {
_obj.Selector[k] = v
}
query := data.D[2].(map[string]interface{})
for k, v := range query {
_obj.Query[k] = v
}
this.mgo.FindOneAndUpdate(data.D[0].(core.SqlTable), _obj.Selector, _obj.Query)
}
return
}
func (this *DB) Model_InsertDBBylog(data *Autogenerated) (err error) {
_, err = this.mgo.InsertOne(DB_ModelTable, data)
if err != nil {
log.Errorf("insert model db err %v", err)
}
return err
}