zyos_go_mq_consume/consume/zhios_acquisition_condition.go

package consume

import (
	"applet/app/db"
	"applet/app/db/model"
	"applet/app/utils"
	"applet/app/utils/logx"
	"applet/consume/md"
	"code.fnuoos.com/go_rely_warehouse/zyos_go_mq.git/rabbit"
	"encoding/json"
	"errors"
	"fmt"
	"github.com/streadway/amqp"
	"time"
	"xorm.io/xorm"
)

func ZhiosAcquisitionCondition(queue md.MqQueue) {
	fmt.Println(">>>>>>>>>>>>>>>>>>>>>>>>")
	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(300)
	delivery := ch.Consume(queue.Name, false)

	var res amqp.Delivery
	var ok bool
	for {
		res, ok = <-delivery
		if ok == true {
			//fmt.Println(string(res.Body))
			fmt.Println(">>>>>>>>>>>>>>>>ZhiosAcquisitionCondition<<<<<<<<<<<<<<<<<<<<<<<<<")
			err = handleZhiosAcquisition(res.Body)
			//_ = res.Reject(false)
			fmt.Println(err)
			if err == nil {
				_ = res.Ack(true)
			} else {
				var canalMsg *md.ZhiosAcquisition
				var tmpString string
				err := json.Unmarshal(res.Body, &tmpString)
				if err == nil {
					fmt.Println(tmpString)
					err = json.Unmarshal([]byte(tmpString), &canalMsg)
					if err == nil {
						ch.Publish(queue.ExchangeName, utils.SerializeStr(canalMsg), queue.RoutKey)
					}
				}

			}
		} else {
			panic(errors.New("error getting message"))
		}
	}
	fmt.Println("get msg done")
}
func handleZhiosAcquisition(msg []byte) error {
	//1、解析canal采集至mq中queue的数据结构体
	var canalMsg *md.ZhiosAcquisition
	fmt.Println(string(msg))
	var tmpString string
	err := json.Unmarshal(msg, &tmpString)
	if err != nil {
		fmt.Println(err.Error())
		return err
	}
	fmt.Println(tmpString)
	err = json.Unmarshal([]byte(tmpString), &canalMsg)
	if err != nil {
		return err
	}
	mid := canalMsg.Mid
	eg := db.DBs[mid]
	if eg == nil {
		return nil
	}
	if canalMsg.Uid == "" {
		return nil
	}
	userInfo, _ := db.UserFindByID(eg, canalMsg.Uid)
	if userInfo == nil {
		return nil
	}
	userProfile, _ := db.UserProfileFindByID(eg, canalMsg.Uid)
	cfg := db.GetAcquisitionCfg(eg, canalMsg.Id, userInfo.CreateAt)
	if cfg == nil {
		return nil
	}
	nextUserProfile, _ := db.UserProfileFindByID(eg, userProfile.ParentUid)
	var user = &md.User{Info: userInfo, Profile: userProfile}

	bools, str := checkAllCompleteTmp(eg, user, cfg)
	isFull := 0
	fullTime := 0
	toRewardTime := 0
	if bools {
		isFull = 1
		fullTime = int(time.Now().Unix())
		toRewardTime = int(time.Now().Unix()) + utils.StrToInt(cfg.RewardAccountDay)*86400
	}
	//写入奖励记录

	//新的设置 读第一个新注册的奖励
	lv := 0
	if len(cfg.RewardRule.LvRewardList) > 0 {
		isEnd := 0
		for k, v := range cfg.RewardRule.LvRewardList {
			if isEnd == 1 {
				continue
			}
			if utils.StrToInt(v.Id) == user.Info.Level {
				isEnd = 1
			}
			lv = utils.StrToInt(v.Id)
			InvitedReward := v.InvitedReward
			if cfg.RewardRule.RewardType == "1" {
				InvitedReward = Rands(v.InvitedReward, v.InvitedRewardMax)
			}
			//直推
			DirectSuccess := v.DirectSuccess
			if cfg.RewardRule.RewardType == "1" {
				DirectSuccess = Rands(v.DirectSuccess, v.DirectSuccessMax)
			}
			//间推
			IndirectSuccess := v.IndirectSuccess
			if cfg.RewardRule.RewardType == "1" {
				IndirectSuccess = Rands(v.IndirectSuccess, v.IndirectSuccessMax)
			}
			InvitedSource := 0
			DirectSource := 1
			IndirectSource := 2
			InvitedSourceStr := "注册奖励"
			DirectSourceStr := "直推好友"
			IndirectSourceStr := "间推好友"
			if k > 0 {
				InvitedSource = 3
				DirectSource = 4
				IndirectSource = 5
				InvitedSourceStr = "升级" + v.Name + "奖励"
				DirectSourceStr = "直推好友升级" + v.Name + "奖励"
				IndirectSourceStr = "间推好友升级" + v.Name + "奖励"
			}
			if utils.StrToFloat64(InvitedReward) > 0 {
				ownRewardLog, ownhas, _ := db.GetNewAcquisitionRewardLogWhere(eg, userProfile.Uid, user.Profile.Uid, lv)
				if !ownhas {
					ownRewardLog = &model.NewAcquisitionRewardLog{
						Uid:        user.Profile.Uid,
						ToUid:      user.Profile.Uid,
						Title:      user.Info.Nickname,
						Source:     InvitedSource,
						SourceText: InvitedSourceStr,
						Money:      InvitedReward,
						CreatedAt:  int(time.Now().Unix()),
						State:      0,
						CoinId:     utils.StrToInt(cfg.RewardRule.RewardCoinId),
						RewardType: utils.StrToInt(cfg.RewardRule.RewardType),
						InviteTime: int(userInfo.CreateAt.Unix()),
						Lv:         lv,
					}

					db.InsertNewRewardLog(eg, ownRewardLog)
				}
				if ownRewardLog.GivenAt == 0 {

					ownRewardLog.CompleteCon = str
					ownRewardLog.IsFull = isFull
					if ownRewardLog.FullTime == 0 {
						ownRewardLog.FullTime = fullTime
					}
					if ownRewardLog.ToRewardTime == 0 {
						ownRewardLog.ToRewardTime = toRewardTime
					}
					eg.Where("id=?", ownRewardLog.Id).Update(ownRewardLog)
				}
			}
			if utils.StrToFloat64(DirectSuccess) > 0 {
				if userProfile.ParentUid > 0 {
					//写入奖励记录
					extendRewardLog, extendHas, _ := db.GetNewAcquisitionRewardLogWhere(eg, userProfile.ParentUid, userProfile.Uid, lv)
					if !extendHas {
						extendRewardLog = &model.NewAcquisitionRewardLog{
							Uid:        user.Profile.ParentUid,
							ToUid:      user.Profile.Uid,
							Title:      user.Info.Nickname,
							Source:     DirectSource,
							SourceText: DirectSourceStr,
							Money:      DirectSuccess,
							CreatedAt:  int(time.Now().Unix()),
							State:      0,
							CoinId:     utils.StrToInt(cfg.RewardRule.RewardCoinId),
							RewardType: utils.StrToInt(cfg.RewardRule.RewardType),
							InviteTime: int(userInfo.CreateAt.Unix()),
							Lv:         lv,
						}
						db.InsertNewRewardLog(eg, extendRewardLog)
					}
					if extendRewardLog.GivenAt == 0 {

						extendRewardLog.CompleteCon = str
						extendRewardLog.IsFull = isFull
						if extendRewardLog.FullTime == 0 {
							extendRewardLog.FullTime = fullTime
						}
						if extendRewardLog.ToRewardTime == 0 {
							extendRewardLog.ToRewardTime = toRewardTime
						}
						eg.Where("id=?", extendRewardLog.Id).Update(extendRewardLog)
					}
				}
			}
			if utils.StrToFloat64(IndirectSuccess) > 0 {
				if nextUserProfile != nil && nextUserProfile.ParentUid > 0 {
					IndirectRewardLog, IndirectHas, _ := db.GetNewAcquisitionRewardLogWhere(eg, nextUserProfile.ParentUid, userProfile.Uid, lv)
					if !IndirectHas {
						IndirectRewardLog = &model.NewAcquisitionRewardLog{
							Uid:        nextUserProfile.ParentUid,
							ToUid:      user.Profile.Uid,
							Title:      user.Info.Nickname,
							Source:     IndirectSource,
							SourceText: IndirectSourceStr,
							Money:      IndirectSuccess,
							CreatedAt:  int(time.Now().Unix()),
							State:      0,
							CoinId:     utils.StrToInt(cfg.RewardRule.RewardCoinId),
							RewardType: utils.StrToInt(cfg.RewardRule.RewardType),
							InviteTime: int(userInfo.CreateAt.Unix()),
							Lv:         lv,
						}
						db.InsertNewRewardLog(eg, IndirectRewardLog)
					}
					if IndirectRewardLog.GivenAt == 0 {
						IndirectRewardLog.CompleteCon = str
						IndirectRewardLog.IsFull = isFull
						if IndirectRewardLog.FullTime == 0 {
							IndirectRewardLog.FullTime = fullTime
						}
						if IndirectRewardLog.ToRewardTime == 0 {
							IndirectRewardLog.ToRewardTime = toRewardTime
						}
						eg.Where("id=?", IndirectRewardLog.Id).Update(IndirectRewardLog)
					}
				}
			}
		}
	} else {
		InvitedReward := cfg.RewardRule.InvitedReward
		if cfg.RewardRule.RewardType == "1" {
			InvitedReward = Rands(cfg.RewardRule.InvitedReward, cfg.RewardRule.InvitedRewardMax)
		}
		//直推
		DirectSuccess := cfg.RewardRule.DirectSuccess
		if cfg.RewardRule.RewardType == "1" {
			DirectSuccess = Rands(cfg.RewardRule.DirectSuccess, cfg.RewardRule.DirectSuccessMax)
		}
		//间推
		IndirectSuccess := cfg.RewardRule.IndirectSuccess
		if cfg.RewardRule.RewardType == "1" {
			IndirectSuccess = Rands(cfg.RewardRule.IndirectSuccess, cfg.RewardRule.IndirectSuccessMax)
		}
		if utils.StrToFloat64(InvitedReward) > 0 {
			ownRewardLog, ownhas, _ := db.GetNewAcquisitionRewardLog(eg, &model.NewAcquisitionRewardLog{
				Uid:   user.Profile.Uid,
				ToUid: user.Profile.Uid,
			})
			if !ownhas {
				ownRewardLog = &model.NewAcquisitionRewardLog{
					Uid:        user.Profile.Uid,
					ToUid:      user.Profile.Uid,
					Title:      user.Info.Nickname,
					Source:     0,
					SourceText: "注册奖励",
					Money:      InvitedReward,
					CreatedAt:  int(time.Now().Unix()),
					State:      0,
					CoinId:     utils.StrToInt(cfg.RewardRule.RewardCoinId),
					RewardType: utils.StrToInt(cfg.RewardRule.RewardType),
					InviteTime: int(userInfo.CreateAt.Unix()),
					Lv:         lv,
				}

				db.InsertNewRewardLog(eg, ownRewardLog)
			}
			if ownRewardLog.GivenAt == 0 {
				ownRewardLog.CompleteCon = str
				ownRewardLog.IsFull = isFull
				if ownRewardLog.FullTime == 0 {
					ownRewardLog.FullTime = fullTime
				}
				if ownRewardLog.ToRewardTime == 0 {
					ownRewardLog.ToRewardTime = toRewardTime
				}
				eg.Where("id=?", ownRewardLog.Id).Update(ownRewardLog)
			}
		}
		if utils.StrToFloat64(DirectSuccess) > 0 {
			if userProfile.ParentUid > 0 {
				//写入奖励记录
				extendRewardLog, extendHas, _ := db.GetNewAcquisitionRewardLog(eg, &model.NewAcquisitionRewardLog{
					Uid:   userProfile.ParentUid,
					ToUid: userProfile.Uid,
				})
				if !extendHas {
					extendRewardLog = &model.NewAcquisitionRewardLog{
						Uid:        user.Profile.ParentUid,
						ToUid:      user.Profile.Uid,
						Title:      user.Info.Nickname,
						Source:     1,
						SourceText: "直推好友",
						Money:      DirectSuccess,
						CreatedAt:  int(time.Now().Unix()),
						State:      0,
						CoinId:     utils.StrToInt(cfg.RewardRule.RewardCoinId),
						RewardType: utils.StrToInt(cfg.RewardRule.RewardType),
						InviteTime: int(userInfo.CreateAt.Unix()),
						Lv:         lv,
					}
					db.InsertNewRewardLog(eg, extendRewardLog)
				}
				if extendRewardLog.GivenAt == 0 {
					extendRewardLog.CompleteCon = str
					extendRewardLog.IsFull = isFull
					if extendRewardLog.FullTime == 0 {
						extendRewardLog.FullTime = fullTime
					}
					if extendRewardLog.ToRewardTime == 0 {
						extendRewardLog.ToRewardTime = toRewardTime
					}
					eg.Where("id=?", extendRewardLog.Id).Update(extendRewardLog)
				}
			}
		}
		if utils.StrToFloat64(IndirectSuccess) > 0 {
			if nextUserProfile != nil && nextUserProfile.ParentUid > 0 {
				IndirectRewardLog, IndirectHas, _ := db.GetNewAcquisitionRewardLog(eg, &model.NewAcquisitionRewardLog{
					Uid:   nextUserProfile.ParentUid,
					ToUid: userProfile.Uid,
				})
				if !IndirectHas {
					IndirectRewardLog = &model.NewAcquisitionRewardLog{
						Uid:        nextUserProfile.ParentUid,
						ToUid:      user.Profile.Uid,
						Title:      user.Info.Nickname,
						Source:     2,
						SourceText: "间推好友",
						Money:      IndirectSuccess,
						CreatedAt:  int(time.Now().Unix()),
						State:      0,
						CoinId:     utils.StrToInt(cfg.RewardRule.RewardCoinId),
						RewardType: utils.StrToInt(cfg.RewardRule.RewardType),
						InviteTime: int(userInfo.CreateAt.Unix()),
						Lv:         lv,
					}
					db.InsertNewRewardLog(eg, IndirectRewardLog)
				}
				if IndirectRewardLog.GivenAt == 0 {
					IndirectRewardLog.CompleteCon = str
					IndirectRewardLog.IsFull = isFull
					if IndirectRewardLog.FullTime == 0 {
						IndirectRewardLog.FullTime = fullTime
					}
					if IndirectRewardLog.ToRewardTime == 0 {
						IndirectRewardLog.ToRewardTime = toRewardTime
					}
					eg.Where("id=?", IndirectRewardLog.Id).Update(IndirectRewardLog)
				}
			}
		}
	}

	return nil
}
func Rands(minVal, maxVal string) string {
	min := int(utils.StrToFloat64(minVal) * 100)
	max := int(utils.StrToFloat64(maxVal) * 100)
	return utils.Float64ToStrByPrec(float64(utils.RandIntRand(min, max))/100, 3)
}

//判断是否符合条件
func checkAllCompleteTmp(eg *xorm.Engine, user *md.User, acqCfg *md.AcquisitionCfg) (bool, string) {
	res := true
	str := ""
	if acqCfg.SuccessConditions.Register.Open == "1" {
		res = res && AcqRegisterTmp(user, acqCfg)
		if res {
			str += ",Register"
		}
	}
	if acqCfg.SuccessConditions.TaobaoAuthorization.Open == "1" {
		res = res && AcqTaoBaoAuthTmp(user, acqCfg)
		if res {
			str += ",TaobaoAuthorization"
		}
	}
	if acqCfg.SuccessConditions.FirstOrder.Open == "1" {
		res = res && AcqFirstOrder(eg, user, acqCfg)
		if res {
			str += ",FirstOrder"
		}
	}
	if acqCfg.SuccessConditions.SelfOrder.Open == "1" {
		res = res && AcqSelfOrder(eg, user, acqCfg)
		if res {
			str += ",SelfOrder"
		}
	}
	if acqCfg.SuccessConditions.OrderPay.Open == "1" {
		res = res && AcqOrderPay(eg, user, acqCfg)
		if res {
			str += ",OrderPay"
		}
	}
	if len(str) > 0 {
		str = str[1:]
	}
	return res, str
}
func AcqRegisterTmp(user *md.User, acqCfg *md.AcquisitionCfg) bool {
	var startTime = utils.TimeStdParseUnix(acqCfg.StartTime)
	var endTime = utils.TimeStdParseUnix(acqCfg.EndTime)
	if startTime == 0 || endTime == 0 {
		return false
	}
	//时间不在活动范围之内返回false
	if user.Info.CreateAt.Unix() < startTime {
		return false
	}
	if user.Info.CreateAt.Unix() > endTime {
		return false
	}
	return true
}

func AcqTaoBaoAuthTmp(user *md.User, acqCfg *md.AcquisitionCfg) bool {
	if user.Profile.AccTaobaoAuthTime > 0 {
		return true
	}
	return false
}

func AcqFirstOrder(eg *xorm.Engine, user *md.User, acqCfg *md.AcquisitionCfg) bool {
	endTime := int(user.Info.CreateAt.Unix()) + utils.StrToInt(acqCfg.SuccessConditions.FirstOrder.Day)*86400
	return commAmount(eg, utils.IntToStr(user.Info.Uid), endTime, 0, acqCfg)
}
func AcqSelfOrder(eg *xorm.Engine, user *md.User, acqCfg *md.AcquisitionCfg) bool {
	return commAmount(eg, utils.IntToStr(user.Info.Uid), 0, 2, acqCfg)
}
func AcqOrderPay(eg *xorm.Engine, user *md.User, acqCfg *md.AcquisitionCfg) bool {
	return commAmount(eg, utils.IntToStr(user.Info.Uid), 0, 1, acqCfg)
}
func sqlSelect(eg *xorm.Engine, uid string, endTime, types int, acqCfg *md.AcquisitionCfg, arr []string) int {
	sql := `SELECT COUNT(*) as count FROM %s ol
	LEFT JOIN %s olr on olr.oid=%s and olr.uid=%s
	WHERE ol.uid=? %s %s
`
	str := ""
	if endTime > 0 {
		str += " AND olr.create_at<=" + utils.IntToStr(endTime)
	}
	if types == 2 {
		str += " AND olr.amount>=" + acqCfg.SuccessConditions.SelfOrder.Money
	}
	if types == 1 {
		str += " AND %s>=" + acqCfg.SuccessConditions.OrderPay.Money
		str = fmt.Sprintf(str, arr[0])
	}
	sqlOrd := fmt.Sprintf(sql, arr[1], arr[2], arr[3], arr[4], str, arr[5])
	ordResult, err := db.QueryNativeString(eg, sqlOrd, uid)
	fmt.Println(sqlOrd)
	fmt.Println(err)
	count := 0
	for _, v := range ordResult {
		count = utils.StrToInt(v["count"])
	}
	return count
}
func commAmount(eg *xorm.Engine, uid string, endTime, types int, acqCfg *md.AcquisitionCfg) bool {
	state := "0,1,2,3,5"
	psoState := "'订单付款','订单结算'"
	mallState := "1,2,3"
	o2oState := "1,2,3,4"
	b2cState := "1,2,3,4"
	if acqCfg.SuccessConditions.FirstOrder.Status == "1" {
		state = "1,2,3,5"
		mallState = "2,3"
		o2oState = "2,3,4"
		b2cState = "2,3,4"
	}
	if acqCfg.SuccessConditions.FirstOrder.Status == "2" {
		state = "2,3,5"
		mallState = "2,3"
		o2oState = "2,3,4"
		b2cState = "2,3,4"
	}
	if acqCfg.SuccessConditions.FirstOrder.Status == "3" {
		state = "3,5"
		mallState = "3"
		o2oState = "3,4"
		b2cState = "3,4"
		psoState = "'订单结算'"
	}

	arr := []string{"ol.paid_price", "ord_list", "ord_list_relate", "ol.ord_id", "ol.uid", " and ol.state in(" + state + ")"}
	count := sqlSelect(eg, uid, endTime, types, acqCfg, arr)
	arr = []string{"ol.paid_price", "privilege_card_ord", "ord_list_relate", "ol.ord_id", "ol.uid", " and ol.state=1"}
	count += sqlSelect(eg, uid, endTime, types, acqCfg, arr)
	arr = []string{"ol.commission", "duoyou_ord_list", "ord_list_relate", "ol.oid", "ol.uid", " and ol.id>0"}
	count += sqlSelect(eg, uid, endTime, types, acqCfg, arr)
	arr = []string{"ol.amount", "recharge_order", "ord_list_relate", "ol.oid", "ol.uid", " and ol.status<>'已退款'"}
	count += sqlSelect(eg, uid, endTime, types, acqCfg, arr)
	arr = []string{"ol.amount", "playlet_sale_order", "ord_list_relate", "ol.custom_oid", "ol.uid", " and ol.status in(" + psoState + ")"}
	count += sqlSelect(eg, uid, endTime, types, acqCfg, arr)

	arr = []string{"ol.cost_price", "mall_ord", "mall_ord_list_relate", "ol.ord_id", "ol.uid", " and ol.state in(" + mallState + ")"}
	count += sqlSelect(eg, uid, endTime, types, acqCfg, arr)

	arr = []string{"ol.cost_price", "o2o_ord", "o2o_ord_list_relate", "ol.ord_id", "ol.uid", " and ol.state in(" + o2oState + ")"}
	count += sqlSelect(eg, uid, endTime, types, acqCfg, arr)
	arr = []string{"ol.actual_pay_amount", "o2o_pay_to_merchant", "o2o_ord_list_relate", "ol.pay_id", "ol.uid", " and ol.state >=1"}
	count += sqlSelect(eg, uid, endTime, types, acqCfg, arr)
	arr = []string{"ol.cost_price", "b2c_ord", "b2c_ord_list_relate", "ol.ord_id", "ol.uid", " and ol.state in(" + b2cState + ")"}
	count += sqlSelect(eg, uid, endTime, types, acqCfg, arr)
	if count > 0 {
		return true
	}
	return false
}