egg_mq_consume/consume/egg_energy_auto_exchange_green_energy.go

package consume

import (
	"applet/app/cfg"
	"applet/app/db"
	md4 "applet/app/md"
	utils2 "applet/app/utils"
	"applet/app/utils/logx"
	"applet/consume/md"
	"code.fnuoos.com/EggPlanet/egg_system_rules.git"
	"code.fnuoos.com/EggPlanet/egg_system_rules.git/rule/egg_energy"
	md2 "code.fnuoos.com/EggPlanet/egg_system_rules.git/rule/egg_energy/md"
	"code.fnuoos.com/go_rely_warehouse/zyos_go_mq.git/rabbit"
	"encoding/json"
	"errors"
	"fmt"
	"github.com/streadway/amqp"
)

func EggEnergyAutoExchangeGreenEnergyConsume(queue md.MqQueue) {
	fmt.Println(">>>>>>>>>>>>EggEnergyAutoExchangeGreenEnergyConsume>>>>>>>>>>>>")
	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(1)
	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 = handleEggEnergyAutoExchangeGreenEnergyConsume(res.Body, ch)
			if err != nil {
				fmt.Println("EggEnergyAutoExchangeGreenEnergyConsume_ERR:::::", err.Error())
				utils2.FilePutContents("EggEnergyAutoExchangeGreenEnergyConsume_ERR", utils2.SerializeStr(map[string]interface{}{
					"body": res.Body,
					"err":  err.Error(),
				}))
				_ = res.Reject(true) //TODO::拒绝
			} else {
				err = res.Ack(true)
				fmt.Println("err ::: ", err)
			}
		} else {
			panic(errors.New("error getting message"))
		}
	}
	fmt.Println("get msg done")
}

func handleEggEnergyAutoExchangeGreenEnergyConsume(msgData []byte, ch *rabbit.Channel) error {
	//1、解析mq中queue的数据结构体
	var msg *md2.EggEnergyStructForAutoExchangeGreenEnergy
	err := json.Unmarshal(msgData, &msg)
	if err != nil {
		return err
	}
	engine := db.Db

	//2、调用制度方法
	err = egg_energy.ActivityCoinAutoExchangeEggPersonEnergy(engine, *msg)
	if err != nil {
		return err
	}
	//3、给到直推奖励
	ch.Publish(md2.EggEnergyExchange, md4.EggEnergyRecruitNewUserRewardMq{
		Uid: msg.Uid,
	}, "egg_energy_recruit_new_user_reward")
	return nil
}