package client import ( "hudongzhuanjia/controllers" "hudongzhuanjia/libs/filter" "hudongzhuanjia/models" pay_service "hudongzhuanjia/services/pay" red_envelope_service "hudongzhuanjia/services/red_envelope" "hudongzhuanjia/utils" "hudongzhuanjia/utils/code" "strings" "time" ) type NoticeRedPackEvent struct { OutTradeNo string `json:"out_trade_no"` Prompt string `json:"prompt"` RedPackInfoId int64 `json:"red_pack_info_id"` ActivityId int64 `json:"activity_id"` Status int `json:"status"` } type LiveCtl struct { controllers.AuthorCtl //controllers.BaseCtl } // 详情 func (t *LiveCtl) Detail() { activityId := t.MustGetInt64("activity_id") areaId := t.MustGetInt64("area_id") userId := t.MustGetUID() err := new(models.LiveViewer).Record(userId, activityId) t.CheckErr(err) live := new(models.LiveConfig) exist, err := live.GetByActivityId(activityId) t.CheckErr(err) t.Assert(exist, code.MSG_ACTIVITY_NOT_EXIST, "直播活动不存在") config := new(models.LiveConfigArea) exist, err = config.GetByActivityIdAndAreaId(activityId, areaId) t.CheckErr(err) if exist { live.SharePosterImg = config.MergeSharePoster } live.AdminLiveUrl = "" t.JSON(live) } func (t *LiveCtl) Like() { activityId := t.MustGetInt64("activity_id") //userId := t.MustGetUID() _, err := new(models.LiveConfig).Like(activityId) t.CheckErr(err) live := new(models.LiveConfig) exist, err := live.GetByActivityId(activityId) t.CheckErr(err) t.Assert(exist, code.MSG_ACTIVITY_NOT_EXIST, "直播活动不存在") t.JSON(map[string]interface{}{ "like": live.LikeNum, "watch": live.WatchNum, }) } func (t *LiveCtl) LoopQuery() { activityId := t.MustGetInt64("activity_id") live := new(models.LiveConfig) exist, err := live.GetByActivityId(activityId) t.CheckErr(err) t.Assert(exist, code.MSG_ACTIVITY_NOT_EXIST, "直播活动不存在") t.JSON(map[string]interface{}{ "like": live.LikeNum, "watch": live.WatchNum, }) } // 下单发送红包 // 维护一个队列进行循环, 遍历是否付款成功 func (t *LiveCtl) SendLiveRedPack() { userId := t.MustGetUID() // 用户 uid activityId := t.MustGetInt64("activity_id") // activity_id num := t.MustGetInt("num") // 红包数量 prompt := t.MustGet("prompt") // 提示 amount := utils.Float64CusDecimal(t.MustGetDouble("amount"), 2) // 金额 areaId := t.MustGetInt64("area_id") activity := new(models.Activity) exist, err := models.GetById(activity, activityId) t.CheckErr(err) t.Assert(exist, code.MSG_ACTIVITY_NOT_EXIST, "互动不存在") area := new(models.AreaStore) exist, err = models.GetById(area, areaId) t.CheckErr(err) t.Assert(exist, code.MSG_AREASTORE_NOT_EXIST, "地区不存在") user := models.User{} exist, err = models.GetById(&user, userId) t.CheckErr(err) t.Assert(exist, code.MSG_USER_NOT_EXIST, "用户不存在") ip := strings.Split(t.Request.OriginRequest.RemoteAddr, ":") res, err := pay_service.UnifiedOrder("欧轩互动-直播红包", ip[0], user.Openid, int64(amount*100), 3, activityId, userId) t.CheckErr(err) rule := models.LiveRedEnvelopeRule{} rule.OutTradeNo = res["out_trade_no"].(string) rule.Amount = amount rule.UserId = userId rule.ActivityId = activityId rule.Prompt = filter.Replace(prompt) rule.IsDelete = false rule.UpdatedAt = time.Now() rule.CreatedAt = time.Now() _, err = rule.Add() t.CheckErr(err) records := red_envelope_service.GenRedPack(int(amount*100), num) for _, v := range records { record := new(models.ShakeRedEnvelopeRecord) record.AreaId = area.Id record.AreaName = area.Name record.LiveRedEnvelopeRuleId = rule.Id record.ActivityId = activityId record.Name = activity.Name record.UserId = 0 record.Amount = utils.Float64CusDecimal(float64(v)/float64(100), 2) record.CreatedAt = time.Now() record.UpdatedAt = time.Now() _, err = record.Add() t.CheckErr(err) } res["red_pack_info_id"] = rule.Id t.JSON(res) } // 支付之后可以遍历查询是否成功 -- 前端发送消息 // 不建议, 通过im 系统进行通知 func (t *LiveCtl) QueryLiveRedPack() { outTradeNo := t.MustGet("out_trade_no") res, err := pay_service.OrderQuery(outTradeNo) t.CheckErr(err) info := new(models.LiveRedEnvelopeRule) exist, err := info.GetByOutTradeNo(outTradeNo) t.CheckErr(err) t.Assert(exist, code.MSG_LIVE_RED_PACK_INFO_NOT_EXIST, "直播红包信息不存在") if res.TradeState == pay_service.CODE_TRADE_SUCCESS { info.Status = 1 info.UpdateStatusById(info.Id, info.Status) t.JSON(map[string]interface{}{ "red_pack_info_id": info.Id, "status": info.Status, }) } else { t.JSON(map[string]interface{}{ "red_pack_info_id": info.Id, "status": info.Status, }) } } // 领取红包 func (t *LiveCtl) GetLiveRedPack() { liveRedPackInfoId := t.MustGetInt64("live_red_pack_info_id") userId := t.MustGetUID() user := models.User{} exist, err := models.GetById(&user, userId) t.CheckErr(err) t.Assert(exist, code.MSG_USER_NOT_EXIST, "不存在用户") redPack := new(models.ShakeRedEnvelopeRecord) exist, err = redPack.GetByInfoId(liveRedPackInfoId) t.CheckErr(err) if !exist { // 通知其他的人 t.ERROR("红包被领完了", code.MSG_LIVE_RED_PACK_NOT_EXIST) return } // 乐观锁 ==> 防止并发 redPack.OpenId = user.Openid redPack.Receiver = user.Id redPack.TransferType = 1 redPack.PartnerTradeNo = utils.RandomStr(32) row, err := redPack.UpdateStatusById(redPack.Id, 1) t.CheckErr(err) if row != 1 { t.ERROR("红包被领完了", code.MSG_LIVE_RED_PACK_NOT_EXIST) return } pay_service.PutTransferDelayQueue("欧轩互动-红包活动", user.Openid, redPack.PartnerTradeNo, redPack.Amount, 5, 5*60) t.JSON(redPack) }