stat_center/service/activitySer/lotterycode.go

package activitySer

import (
	"context"
	"strings"
	"time"

	"stat/api/proto"
	"stat/common/gencode"
	"stat/common/stdLog"
	"stat/models/activitymod"

	"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"
)

// 生成活动抽奖码
func GenLotteryCodes(ctx context.Context, param proto.GenLotteryCodeReq) (stdLog.CodeType, error) {
	now := time.Now()
	filter := bson.M{activitymod.FN.ActivityInfo.Id: param.ActId}
	info, err := activitymod.FindOneActivityInfo(ctx, filter)
	if err != nil {
		return stdLog.QueryDataError, err
	}
	doc := &activitymod.LottertyCodeInfo{
		ActId:     param.ActId.Hex(),
		ActName:   info.Name,
		IsUsed:    false,
		CreatedAt: now,
	}
	for i := 0; i < int(param.CodeNum); i++ {
		code, err := genLotteryCode(ctx, int(param.CodeLen))
		if err != nil {
			return stdLog.Failure, err
		}
		doc.LotteryCode = code
		doc.Id = primitive.NewObjectID()
		if err := activitymod.InsertOneLottertyCodeInfo(ctx, doc); err != nil {
			return stdLog.InsertDataError, err
		}

	}
	return stdLog.Success, nil
}

func genLotteryCode(ctx context.Context, len int) (string, error) {
	// 生成码
	for {
		code := strings.ToUpper(gencode.CodeGenerate(len))
		filter := bson.M{activitymod.FN.LottertyCodeInfo.LotteryCode: code}
		_, err := activitymod.FindOneLottertyCodeInfo(ctx, filter)
		if err != nil {
			if err != mongo.ErrNoDocuments {
				return "", err
			}
			return code, nil
		}
	}
}

// 修改抽奖码信息
func UpdateLotteryCode(ctx context.Context, param proto.UpdateLotteryCodeReq) (stdLog.CodeType, error) {
	now := time.Now()
	filter := bson.M{activitymod.FN.LottertyCodeInfo.Id: param.LotteryCodeId}
	updateDoc := bson.M{"$set": bson.M{
		activitymod.FN.LottertyCodeInfo.Remark:    param.Mark,
		activitymod.FN.LottertyCodeInfo.UpdatedAt: now,
	}}
	err := activitymod.UpdateOneLottertyCodeInfo(ctx, filter, updateDoc)
	if err != nil {
		return stdLog.UpdateDataError, err
	}
	return stdLog.Success, nil
}

// 删除一个活动抽奖码
func DeleteOneLotteryCode(ctx context.Context, param proto.DelLotteryCodeReq) (stdLog.CodeType, error) {
	filter := bson.M{activitymod.FN.ActivityInfo.Id: param.LotteryCodeId}
	count, err := activitymod.DelOneLottertyCodeInfo(ctx, filter)
	if err != nil {
		return stdLog.UpdateDataError, err
	}
	if count == 0 {
		return stdLog.DataNotExist, nil
	}

	return stdLog.Success, nil
}

// 活动抽奖码列表
func GetLotteryCodeList(ctx context.Context, param proto.LotteryCodeListReq) (stdLog.CodeType, *proto.LotteryCodeListResp, error) {
	if param.PageSize <= 0 {
		param.PageSize = 10
	}
	if param.PageNum <= 0 {
		param.PageNum = 1
	}
	data := &proto.LotteryCodeListResp{List: make([]*activitymod.LottertyCodeInfo, 0)}
	filter := bson.M{}
	if param.LotteryCode != nil {
		filter[activitymod.FN.LottertyCodeInfo.LotteryCode] = *param.LotteryCode
	}
	if param.ActId != nil {
		filter[activitymod.FN.LottertyCodeInfo.ActId] = *param.ActId
	}
	if param.ActName != nil {
		filter[activitymod.FN.LottertyCodeInfo.ActName] = bson.M{"$regex": *param.ActName}
	}
	if param.IsUsed != nil {
		filter[activitymod.FN.LottertyCodeInfo.IsUsed] = *param.IsUsed
	}
	if param.GiftCode != nil {
		filter[activitymod.FN.LottertyCodeInfo.GiftCode] = *param.GiftCode
	}
	count, err := activitymod.CountLottertyCodeInfos(ctx, filter)
	if err != nil {
		return stdLog.QueryDataError, data, err
	}
	data.Total = count

	opt := &options.FindOptions{}
	skip := (param.PageNum - 1) * param.PageSize
	if param.PageSize <= 0 {
		param.PageSize = 10
	}
	opt.SetSkip(skip)
	opt.SetLimit(param.PageSize)
	opt.SetSort(bson.M{activitymod.FN.LottertyCodeInfo.CreatedAt: -1})
	list, err := activitymod.FindManyLottertyCodeInfo(ctx, filter, opt)
	if err != nil {
		return stdLog.QueryDataError, data, err
	}
	data.List = list
	return stdLog.Success, data, nil
}