@@ -0,0 +1,96 @@
|
||||
package verifyreportmod
|
||||
|
||||
import (
|
||||
"time"
|
||||
|
||||
"91porn-server/common/db"
|
||||
|
||||
"go.mongodb.org/mongo-driver/bson/primitive"
|
||||
)
|
||||
|
||||
var mdb *db.MongoDB
|
||||
|
||||
const (
|
||||
ProcessingStatusDefault = 0 //等待处理
|
||||
ProcessingStatusComplete = 1 //审核通过
|
||||
ProcessingStatusIgnore = 2 //审核不通过
|
||||
|
||||
ObjectTypeLoufeng = "loufeng"
|
||||
ObjectTypeNenMo = "nenmo"
|
||||
ObjectTypeYyLottery = "yylottery"
|
||||
)
|
||||
|
||||
type Video struct {
|
||||
}
|
||||
|
||||
type VerifyReport struct {
|
||||
ID primitive.ObjectID `json:"id" bson:"_id,omitempty"`
|
||||
UID uint64 `json:"uid" bson:"uid"` //用户id
|
||||
ObjectType string `json:"objectType" bson:"objectType"` //对象类型
|
||||
ProductID primitive.ObjectID `json:"productID" bson:"productID"` //对象id
|
||||
ServiceDetails string `json:"serviceDetails" bson:"serviceDetails"` //内容
|
||||
Imgs []string `json:"imgs" bson:"imgs"` //图片
|
||||
//Videos []string `json:"videos" bson:"videos"` //录像
|
||||
ProcessingStatus int `json:"processingStatus" bson:"processingStatus"` //处理状态
|
||||
ProcessedReceipt string `json:"processedReceipt" bson:"processedReceipt"` //处理回执
|
||||
BrokerUID *uint64 `json:"brokerUID,omitempty" bson:"brokerUID,omitempty"` //经纪人ID
|
||||
CreatedAt time.Time `json:"createdAt" bson:"createdAt,omitempty"` //创建时间
|
||||
UpdatedAt time.Time `json:"updatedAt" bson:"updatedAt,omitempty"` //修改时间
|
||||
}
|
||||
|
||||
type VerifyReportUpsert struct {
|
||||
UID uint64 `json:"uid" bson:"uid"` //用户id
|
||||
ObjectType string `json:"objectType" bson:"objectType"` //对象类型
|
||||
ProductID primitive.ObjectID `json:"productID" bson:"productID"` //对象id
|
||||
ServiceDetails string `json:"serviceDetails" bson:"serviceDetails"` //内容
|
||||
Imgs []string `json:"imgs" bson:"imgs"` //图片
|
||||
//Videos []string `json:"videos" bson:"videos"` //录像
|
||||
ProcessingStatus int `json:"processingStatus" bson:"processingStatus"` //处理状态
|
||||
ProcessedReceipt string `json:"processedReceipt" bson:"processedReceipt"` //处理回执
|
||||
BrokerUID *uint64 `json:"brokerUID,omitempty" bson:"brokerUID,omitempty"` //经纪人ID
|
||||
CreatedAt time.Time `json:"createdAt" bson:"createdAt,omitempty"` //创建时间
|
||||
}
|
||||
|
||||
func Init() {
|
||||
mdb = db.Init(table)
|
||||
initIndex()
|
||||
}
|
||||
|
||||
type QuerySelector struct {
|
||||
ID *primitive.ObjectID `json:"id,omitempty" bson:"_id,omitempty"`
|
||||
UID *uint64 `json:"uid,omitempty" bson:"uid,omitempty"`
|
||||
ObjectType *int `json:"objectType,omitempty" bson:"objectType,omitempty"` //举报对象类型
|
||||
ProductID *primitive.ObjectID `json:"productID,omitempty" bson:"productID,omitempty"` //楼凤id
|
||||
FeedbackType *int `json:"feedbackType,omitempty" bson:"feedbackType,omitempty"` //举报类型
|
||||
ProcessingStatus *int `json:"processingStatus,omitempty" bson:"processingStatus,omitempty"` //处理状态
|
||||
StartTime *time.Time ` json:"startTime,omitempty" bson:"-"` // 开始时间
|
||||
EndTime *time.Time ` json:"endTime,omitempty" bson:"-"` // 结束时间
|
||||
BrokerUID *uint64 `json:"brokerUID,omitempty" bson:"brokerUID,omitempty"` //经纪人ID
|
||||
TimeString *string `json:"timeString,omitempty" bson:"-"` //时间
|
||||
|
||||
}
|
||||
|
||||
type EditSelector struct {
|
||||
ID primitive.ObjectID `json:"id,omitempty" bson:"_id,omitempty"`
|
||||
ProcessingStatus *int `json:"processingStatus,omitempty" bson:"processingStatus,omitempty"` //处理状态
|
||||
ProcessedReceipt *string `json:"processedReceipt,omitempty" bson:"processedReceipt,omitempty"` //处理回执
|
||||
ServiceDetails *string `json:"serviceDetails,omitempty" bson:"serviceDetails,omitempty"` //内容
|
||||
BrokerUID *uint64 `json:"brokerUID,omitempty" bson:"brokerUID,omitempty"` //经纪人ID
|
||||
Imgs *[]string `json:"imgs,omitempty" bson:"imgs,omitempty"` //图片
|
||||
UpdatedAt time.Time `json:"updatedAt" bson:"updatedAt,omitempty"` //修改时间
|
||||
}
|
||||
|
||||
type VerifyReportAppRes struct {
|
||||
Avatar string `json:"avatar"`
|
||||
Name string `json:"name"`
|
||||
IsVIP bool `json:"is_vip"`
|
||||
VipLevel int `json:"vipLevel"`
|
||||
Age int `json:"age"`
|
||||
Gender string `json:"gender"`
|
||||
SuperUser bool `json:"superUser"`
|
||||
VerifyReport `json:",inline" bson:",inline"`
|
||||
Title string `json:"title,omitempty"`
|
||||
City string `json:"city,omitempty"`
|
||||
Price string `json:"price,omitempty"`
|
||||
Cover string `json:"cover,omitempty"`
|
||||
}
|
||||
@@ -0,0 +1,265 @@
|
||||
package verifyreportmod
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"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.VerifyReport
|
||||
|
||||
// InitIndex 设置index
|
||||
func initIndex() {
|
||||
coll := coll(nil)
|
||||
many := []mongo.IndexModel{
|
||||
{
|
||||
Keys: bson.D{{Key: "processingStatus", Value: 1}},
|
||||
},
|
||||
{
|
||||
Keys: bson.D{{Key: "productID", Value: 1}},
|
||||
},
|
||||
{
|
||||
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 VerifyReport) error {
|
||||
v, err := validInsert(f.UID, f.ProductID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if !v.ID.IsZero() {
|
||||
log.Warn(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error: Repeat upload:", "Insert", table, "validInsert"),
|
||||
log.Any("LouFengFeedback", f))
|
||||
return errors.New("Repeat upload")
|
||||
}
|
||||
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("LouFengFeedback", f))
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// validInsert
|
||||
func validInsert(uid uint64, id primitive.ObjectID) (vf VerifyReport, err error) {
|
||||
if err = coll(nil).FindOne(&vf, bson.M{"uid": uid, "productID": id, "processingStatus": ProcessingStatusDefault}); err != nil {
|
||||
log.Warn(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "validInsert", table, "FindOne", err),
|
||||
log.Any("uid", uid),
|
||||
log.Any("id", id))
|
||||
|
||||
return
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// GetByPID 根据id获取验证报告
|
||||
func GetByPID(pid, objectType string, page commod.Page) (v []VerifyReport, hasNext bool, err error) {
|
||||
skip := int64(page.Skip())
|
||||
limit := int64(page.Limit()) + 1
|
||||
opts := options.FindOptions{
|
||||
Skip: &skip,
|
||||
Limit: &limit,
|
||||
Sort: bson.D{{Key: "createdAt", Value: -1}},
|
||||
}
|
||||
productID, err := primitive.ObjectIDFromHex(pid)
|
||||
if err != nil {
|
||||
log.Warn(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "GetByPID", table, "ObjectIDFromHex", err),
|
||||
log.Any("objectType", objectType),
|
||||
log.Any("productID", pid))
|
||||
return
|
||||
}
|
||||
if err = coll(nil).Find(&v, bson.M{"objectType": objectType, "productID": productID, "processingStatus": ProcessingStatusComplete}, &opts); err != nil {
|
||||
log.Warn(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "GetByPID", table, "FindOne", err),
|
||||
log.Any("objectType", objectType),
|
||||
log.Any("productID", pid))
|
||||
return
|
||||
}
|
||||
if uint64(len(v)) > page.PageSize {
|
||||
v = v[:page.PageSize]
|
||||
hasNext = true
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// GetVeRport 获取验证报告
|
||||
func GetVeRport(objectType string, page commod.Page) (v []VerifyReport, hasNext bool, err error) {
|
||||
skip := int64(page.Skip())
|
||||
limit := int64(page.Limit()) + 1
|
||||
opts := options.FindOptions{
|
||||
Skip: &skip,
|
||||
Limit: &limit,
|
||||
Sort: bson.D{{Key: "createdAt", Value: -1}},
|
||||
}
|
||||
if err = coll(nil).Find(&v, bson.M{"objectType": objectType, "processingStatus": ProcessingStatusComplete}, &opts); err != nil {
|
||||
log.Warn(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "GetByPID", table, "FindOne", err),
|
||||
log.Any("objectType", objectType))
|
||||
return
|
||||
}
|
||||
if uint64(len(v)) > page.PageSize {
|
||||
v = v[:page.PageSize]
|
||||
hasNext = true
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// GetByBrokerUID 根据经纪人id获取
|
||||
func GetByBrokerUID(brokerUID uint64, objectType string, page commod.Page) (v []VerifyReport, hasNext bool, err error) {
|
||||
skip := int64(page.Skip())
|
||||
limit := int64(page.Limit()) + 1
|
||||
opts := options.FindOptions{
|
||||
Skip: &skip,
|
||||
Limit: &limit,
|
||||
Sort: bson.D{{Key: "createdAt", Value: -1}},
|
||||
}
|
||||
if err = coll(nil).Find(&v, bson.M{"objectType": objectType, "brokerUID": brokerUID, "processingStatus": ProcessingStatusComplete}, &opts); err != nil {
|
||||
log.Warn(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "GetByBrokerUID", table, "Find", err),
|
||||
log.Any("objectType", objectType),
|
||||
log.Any("brokerUID", brokerUID))
|
||||
return
|
||||
}
|
||||
if uint64(len(v)) > page.PageSize {
|
||||
v = v[:page.PageSize]
|
||||
hasNext = true
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// GetByBrokerUID 根据经纪人id获取总量
|
||||
func CountByBrokerUID(brokerUID uint64, objectType string) (int64, error) {
|
||||
return coll(nil).Count(bson.M{"objectType": objectType, "brokerUID": brokerUID, "processingStatus": ProcessingStatusComplete})
|
||||
}
|
||||
|
||||
// todo web
|
||||
// StdFind 通用查询
|
||||
func StdFind(q QuerySelector, page commod.Page) (data []VerifyReport, total int64, err error) {
|
||||
data = make([]VerifyReport, 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) 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 err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// Update 修改LouFeng类型
|
||||
func FindUpdate(set *EditSelector) (vf VerifyReport, err error) {
|
||||
set.UpdatedAt = time.Now()
|
||||
if err = coll(nil).FindOneAndUpdate(&vf, 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
|
||||
}
|
||||
|
||||
// InsertBulket 插入一条数据
|
||||
func InsertBulket(vr []VerifyReportUpsert) (err error) {
|
||||
wm := make([]mongo.WriteModel, len(vr))
|
||||
for i := range vr {
|
||||
filter := bson.M{"productID": vr[i].ProductID}
|
||||
insert, _ := common.ToBsonM(vr[i])
|
||||
update := bson.M{"$set": bson.M{"updatedAt": vr[i].CreatedAt}, "$setOnInsert": insert}
|
||||
wm[i] = mongo.NewUpdateOneModel().
|
||||
SetFilter(filter).
|
||||
SetUpdate(update).
|
||||
SetUpsert(true)
|
||||
}
|
||||
ordered := false
|
||||
opts := options.BulkWriteOptions{
|
||||
Ordered: &ordered,
|
||||
}
|
||||
res, err := coll(nil).Bulk(wm, &opts)
|
||||
if err != nil {
|
||||
log.Warn(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "InsertMany", table, "InsertMany", err))
|
||||
return
|
||||
}
|
||||
fmt.Println("=======res", res.InsertedCount)
|
||||
return
|
||||
}
|
||||
Reference in New Issue
Block a user