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 }