egg_mq_consume/consume/video_reward_consume.go

package consume

import (
	"applet/app/cfg"
	"applet/app/db"
	"applet/app/e"
	utils2 "applet/app/utils"
	"applet/app/utils/cache"
	"applet/app/utils/logx"
	"applet/consume/md"
	"code.fnuoos.com/EggPlanet/egg_models.git/src/dao"
	"code.fnuoos.com/EggPlanet/egg_models.git/src/implement"
	"code.fnuoos.com/EggPlanet/egg_models.git/src/model"
	"code.fnuoos.com/EggPlanet/egg_system_rules.git"
	"code.fnuoos.com/EggPlanet/egg_system_rules.git/enum"
	md2 "code.fnuoos.com/EggPlanet/egg_system_rules.git/rule/egg_energy/md"
	"code.fnuoos.com/EggPlanet/egg_system_rules.git/svc"
	"code.fnuoos.com/go_rely_warehouse/zyos_go_mq.git/rabbit"
	"encoding/json"
	"errors"
	"fmt"
	"github.com/streadway/amqp"
	"github.com/tidwall/gjson"
	"time"
)

func VideoRewardConsume(queue md.MqQueue) {
	fmt.Println(">>>>>>>>>>>>VideoRewardConsume>>>>>>>>>>>>")
	ch, err := rabbit.Cfg.Pool.GetChannel()
	if err != nil {
		logx.Error(err)
		return
	}
	defer ch.Release()
	//1、将自己绑定到交换机上
	ch.Bind(queue.Name, queue.ExchangeName, queue.RoutKey)
	//2、取出数据进行消费
	ch.Qos(100)
	delivery := ch.Consume(queue.Name, false)

	egg_system_rules.Init(cfg.RedisAddr)
	var res amqp.Delivery
	var ok bool
	for {
		res, ok = <-delivery
		if ok == true {
			err = handleVideoRewardConsume(ch, res.Body)
			if err != nil {
				fmt.Println("VideoRewardConsume_ERR:::::", err.Error())
				utils2.FilePutContents("VideoRewardConsume_ERR", utils2.SerializeStr(map[string]interface{}{
					"body": res.Body,
					"err":  err.Error(),
				}))
				_ = res.Reject(false)
				//TODO::重新推回队列末尾，避免造成队列堵塞
				var msg *md.VideoReward
				json.Unmarshal(res.Body, &msg)
				ch.Publish(queue.ExchangeName, msg, queue.RoutKey)
			} else {
				//_ = res.Reject(false)
				err = res.Ack(true)
			}

			fmt.Println("err ::: ", err)
		} else {
			panic(errors.New("error getting message"))
		}
	}
	fmt.Println("get msg done")
}

func handleVideoRewardConsume(ch *rabbit.Channel, msgData []byte) error {
	time.Sleep(time.Duration(100) * time.Millisecond) //休眠100毫秒
	// 1.解析mq中queue的数据结构体
	var msg *md.VideoReward
	err := json.Unmarshal(msgData, &msg)
	if err != nil {
		return err
	}
	//1、分布式锁阻拦
	requestIdPrefix := fmt.Sprintf("video.reward.lock:%s", msg.Uid)
	cb, err := svc.HandleDistributedLockForComm(msg.Uid, "video.reward.lock.update:%s", requestIdPrefix)
	if err != nil {
		return err
	}
	if cb != nil {
		defer cb() // 释放锁
	}
	eg := db.Db
	redisConn := cache.GetPool().Get()
	sysCfgDb := implement.NewSysCfgDb(eg, redisConn)
	defer func(sysCfgDb dao.SysCfgDao) {
		err := sysCfgDb.Close()
		if err != nil {
			logx.Error("redis close err:" + err.Error()) // 记录错误信息
		}
	}(sysCfgDb)
	videoBase := sysCfgDb.SysCfgGetWithDb("video_base")
	sess := eg.NewSession()
	defer sess.Close()
	sess.Begin()
	NewVideoTotalDb := implement.NewVideoTotalDb(db.Db)
	total, _ := NewVideoTotalDb.GetVideoTotalSess(sess, msg.Uid, time.Now().Format("20060102"))
	if total == nil {
		total = &model.VideoTotal{
			Uid:  utils2.StrToInt(msg.Uid),
			Date: utils2.StrToInt(time.Now().Format("20060102")),
			Time: time.Now(),
		}
		insert, err := sess.Insert(total)
		if insert == 0 || err != nil {
			sess.Rollback()
			return e.NewErr(400, "获取奖励失败")
		}
	}
	Leave := utils2.StrToInt(gjson.Get(videoBase, "total").String()) - total.Total
	if Leave-1 < 0 {
		return nil
	}
	total.Total++
	update, err2 := sess.Where("id=?", total.Id).Cols("total").Update(total)
	if update == 0 || err2 != nil {
		sess.Rollback()
		return e.NewErr(400, "获取奖励失败")
	}
	sess.Commit()
	numKey := "video.num:" + time.Now().Format("20060102") + "." + msg.Uid
	todayRange := utils2.GetTimeRange("today")
	cache.SetEx(numKey, utils2.IntToStr(total.Total), int(todayRange["end"]-time.Now().Unix()))
	NewEggEnergyBasicSettingDb := implement.NewEggEnergyBasicSettingDb(db.Db)
	eggData, _ := NewEggEnergyBasicSettingDb.EggEnergyBasicSettingGetOne()
	err = ch.PublishV2(md2.EggEnergyExchange, md2.EggEnergyStructForDealUserVirtualCoinData{
		Kind:         "add",
		Title:        enum.EggEnergyVideoRewardPersonalActiveCoin.String(),
		TransferType: int(enum.EggEnergyVideoRewardPersonalActiveCoin),
		CoinId:       eggData.PersonEggPointsCoinId,
		Uid:          utils2.StrToInt64(msg.Uid),
		Amount:       utils2.StrToFloat64(gjson.Get(videoBase, "reward").String()),
	}, md2.EggEnergyRoutKeyForDealUserVirtualCoinData)
	if err != nil {
		ch.PublishV2(md2.EggEnergyExchange, md2.EggEnergyStructForDealUserVirtualCoinData{
			Kind:         "add",
			Title:        enum.EggEnergyVideoRewardPersonalActiveCoin.String(),
			TransferType: int(enum.EggEnergyVideoRewardPersonalActiveCoin),
			CoinId:       eggData.PersonEggPointsCoinId,
			Uid:          utils2.StrToInt64(msg.Uid),
			Amount:       utils2.StrToFloat64(gjson.Get(videoBase, "reward").String()),
		}, md2.EggEnergyRoutKeyForDealUserVirtualCoinData)
	}
	return nil
}