diff --git a/controller/game_card.go b/controller/game_card.go index 137aa0d..480b230 100644 --- a/controller/game_card.go +++ b/controller/game_card.go @@ -1297,6 +1297,7 @@ func HmPushWXPayNotice(c *gin.Context) { if !user.OpenMemberTime.IsZero() { spendType = 3 } + isShopAssistantCode := false expireTime := newTime.AddDate(1, 0, 0) if user.MemberExpire.After(newTime) { expireTime = user.MemberExpire.AddDate(1, 0, 0) @@ -1346,6 +1347,7 @@ func HmPushWXPayNotice(c *gin.Context) { logger.Error("user coupon err:", err) } else { if userCoupon.Approach == 1 { + isShopAssistantCode = true err = model.StorePromotion(userCoupon.PromotionalSales, userCoupon.Uid, nil, &user) if err != nil { logger.Error("store promotion err:", err) @@ -1392,6 +1394,9 @@ func HmPushWXPayNotice(c *gin.Context) { //if user.MemberLevel != 2 { // qs = qs.SetMemberOpenTime(newTime).SetMemberLevel(record.MemberLevel) //} + if isShopAssistantCode { + qs = qs.SetInviteForm(1) + } qs = qs.SetMemberOpenTime(newTime).SetMemberLevel(record.MemberLevel) _, err = qs.SetMemberType(2).SetMemberStatus(2). SetAction(2).SetSpendType(spendType).UpdateNum() @@ -1440,11 +1445,13 @@ func HmPushWXPayNotice(c *gin.Context) { UserInviteRecordId: inviteRecordNew.ID, FirstInvite: 0, } + if isShopAssistantCode { + userInviteNew.InviteForm = 1 + } err = model.DB.Create(userInviteNew).Error if err != nil { logger.Error("create invite record err:", err) } - } if invite.FromUid != 0 {