@@ -0,0 +1,147 @@
|
||||
package errfedbkmod
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"time"
|
||||
|
||||
"91porn-server/common"
|
||||
"91porn-server/common/db"
|
||||
"91porn-server/common/log"
|
||||
"91porn-server/models"
|
||||
"91porn-server/models/commod"
|
||||
|
||||
"go.mongodb.org/mongo-driver/bson"
|
||||
"go.mongodb.org/mongo-driver/bson/primitive"
|
||||
"go.mongodb.org/mongo-driver/mongo"
|
||||
"go.mongodb.org/mongo-driver/mongo/options"
|
||||
)
|
||||
|
||||
const table = models.ErrFeedback
|
||||
|
||||
// InitIndex 设置index
|
||||
func initIndex() {
|
||||
coll := coll(nil)
|
||||
many := []mongo.IndexModel{
|
||||
{
|
||||
Keys: bson.D{{Key: "uid", Value: 1}},
|
||||
},
|
||||
}
|
||||
if _, err := coll.CreateIndex(many); err != nil {
|
||||
panic(fmt.Sprintf("%s model set index err ==>[%+v]", table, err))
|
||||
}
|
||||
}
|
||||
|
||||
func coll(t *db.MongoTool) *db.MongoTool {
|
||||
if t == nil {
|
||||
return mdb.Coll(table)
|
||||
}
|
||||
return t.Coll(table)
|
||||
}
|
||||
|
||||
// todo app
|
||||
func Insert(f ErrFeedback) (err error) {
|
||||
now := time.Now()
|
||||
f.UpdatedAt = now
|
||||
f.CreatedAt = now
|
||||
if _, err = coll(nil).InsertOne(f); err != nil {
|
||||
log.Warn(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "Insert", table, "InsertOne", err), log.Any("LouFengErrFeedback", f))
|
||||
return
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func FindOneByProductId(pid primitive.ObjectID) (*ErrFeedback, error) {
|
||||
filter := bson.M{"productID": pid, "processingStatus": ProcessingStatusDefault}
|
||||
var f ErrFeedback
|
||||
if err := coll(nil).FindOne(&f, filter); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if f.ID.IsZero() {
|
||||
return nil, nil
|
||||
}
|
||||
return &f, nil
|
||||
}
|
||||
|
||||
// todo web
|
||||
// StdFind 通用查询
|
||||
func StdFind(q QuerySelector, page commod.Page) (data []ErrFeedback, total int64, err error) {
|
||||
data = make([]ErrFeedback, 0)
|
||||
skip := int64(page.Skip())
|
||||
limit := int64(page.Limit())
|
||||
opts := options.FindOptions{
|
||||
Skip: &skip,
|
||||
Limit: &limit,
|
||||
Sort: bson.D{{Key: "createdAt", Value: -1}},
|
||||
}
|
||||
f, err := common.ToBsonM(q)
|
||||
if err != nil {
|
||||
log.Warn(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "StdFind", table, "ToBsonM", err), log.Any("cond", f))
|
||||
return
|
||||
}
|
||||
createdTime := bson.M{}
|
||||
if q.StartTime != nil {
|
||||
createdTime["$gte"] = q.StartTime
|
||||
}
|
||||
if q.EndTime != nil {
|
||||
createdTime["$lt"] = q.EndTime
|
||||
}
|
||||
if q.TimeString != nil {
|
||||
f[*q.TimeString] = createdTime
|
||||
} else {
|
||||
if len(createdTime) > 0 {
|
||||
f["createdAt"] = createdTime
|
||||
}
|
||||
}
|
||||
if err = coll(nil).Find(&data, f, &opts); err != nil {
|
||||
log.Warn(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "StdFind", table, "Find", err), log.Any("cond", f))
|
||||
return
|
||||
}
|
||||
total, err = coll(nil).Count(f)
|
||||
if err != nil {
|
||||
log.Warn(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "StdFind", table, "Count", err), log.Any("cond", f))
|
||||
return
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// Update 修改LouFeng类型
|
||||
func Update(set *EditSelector) (err error) {
|
||||
set.UpdatedAt = time.Now()
|
||||
if _, err = coll(nil).UpdateOne(bson.M{"_id": set.ID}, bson.M{"$set": set}); err != nil {
|
||||
log.Warn(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "Update", table, "UpdateOne", err),
|
||||
log.Any("set", set),
|
||||
)
|
||||
return
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// Update 修改LouFeng类型
|
||||
func FindUpdate(set *EditSelector) (fb ErrFeedback, err error) {
|
||||
set.UpdatedAt = time.Now()
|
||||
if err = coll(nil).FindOneAndUpdate(&fb, bson.M{"_id": set.ID}, bson.M{"$set": set}); err != nil {
|
||||
log.Warn(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "Update", table, "UpdateOne", err),
|
||||
log.Any("set", set),
|
||||
)
|
||||
return
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// Remove 删除LouFeng类型
|
||||
func Remove(id string) (err error) {
|
||||
OID, err := primitive.ObjectIDFromHex(id)
|
||||
if err != nil {
|
||||
log.Warn(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "Remove", table, "ObjectIDFromHex", err),
|
||||
log.Any("id", id),
|
||||
)
|
||||
return
|
||||
}
|
||||
if _, err = coll(nil).DeleteOne(bson.M{"_id": OID}); err != nil {
|
||||
log.Warn(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "Remove", table, "DeleteOne", err),
|
||||
log.Any("OID", OID),
|
||||
)
|
||||
return
|
||||
}
|
||||
return
|
||||
}
|
||||
Reference in New Issue
Block a user