小黄条

yujian

2019-12-10 c8041ec0544bf122e6819e6bf698997ccbf30aaf

 fanli/src/main/java/com/yeshi/fanli/service/impl/user/UserInfoExtraServiceImpl.java

@@ -1,5 +1,6 @@
package com.yeshi.fanli.service.impl.user;



import java.io.InputStream;

import java.text.ParseException;

import java.text.SimpleDateFormat;

import java.util.ArrayList;

@@ -12,12 +13,15 @@
import javax.annotation.Resource;



import org.springframework.stereotype.Service;

import org.springframework.transaction.annotation.Transactional;

import org.yeshi.utils.DateUtil;

import org.yeshi.utils.HttpUtil;

import org.yeshi.utils.entity.FileUploadResult;

import org.yeshi.utils.tencentcloud.COSManager;



import com.yeshi.fanli.dao.mybatis.user.UserInfoExtraMapper;

import com.yeshi.fanli.dao.mybatis.user.UserRankRecordMapper;

import com.yeshi.fanli.dto.wx.WXAccountInfoDTO;

import com.yeshi.fanli.entity.bus.user.ThreeSale;

import com.yeshi.fanli.entity.bus.user.UserInfo;

import com.yeshi.fanli.entity.bus.user.UserInfoExtra;

import com.yeshi.fanli.entity.bus.user.UserRank;

@@ -36,6 +40,7 @@
import com.yeshi.fanli.service.inter.user.UserRankService;

import com.yeshi.fanli.service.inter.user.invite.ThreeSaleSerivce;

import com.yeshi.fanli.util.Constant;

import com.yeshi.fanli.util.InviteCodeFilterUtil;

import com.yeshi.fanli.util.StringUtil;

import com.yeshi.fanli.util.account.UserUtil;

import com.yeshi.fanli.util.wx.WXLoginUtil;

@@ -155,6 +160,7 @@
      }



      boolean isupdateRank = true;

      UserInfoExtra extra = new UserInfoExtra();



      UserInfoExtra userInfoExtra = userInfoExtraMapper.getInfoExtraByUid(uid);

      if (userInfoExtra != null && userInfoExtra.getId() != null) {

@@ -166,14 +172,91 @@
               isupdateRank = false;

            }

         }

         extra.setId(userInfoExtra.getId());

      } else {

         userInfoExtra = new UserInfoExtra();

         userInfoExtra.setUserInfo(new UserInfo(uid));

         extra.setUserInfo(new UserInfo(uid));

      }



		
      // 更新等级

      if (isupdateRank) {

         updateRank(userInfoExtra);

         List<UserRank> listRank = userRankService.getAllRank();

         if (listRank == null || listRank.size() == 0) {

            throw new UserInfoExtraException(1, "系统等级不存在");

         }



         int selfOrderNum = 0;

         int sharedOrderNum = 0;

         int inviteOrderNum = 0;

         Map<String, Object> map = commonOrderCountService.lastMonthSettleOrderNumber(uid);

         if (map != null) {

            // 返利订单

            if (map.get("totalSelf") != null) {

               selfOrderNum = Integer.parseInt(map.get("totalSelf").toString());

            }



            // 分享订单

            if (map.get("totalShared") != null) {

               sharedOrderNum = Integer.parseInt(map.get("totalShared").toString());

            }



            // 邀请订单

            if (map.get("totalInvite") != null) {

               inviteOrderNum = Integer.parseInt(map.get("totalInvite").toString());

            }

         }



         UserRank rank = null;

         for (UserRank userRank : listRank) {

            // 邀请满足

            Integer inviteNum = userRank.getInviteNum();

            if (inviteOrderNum >= inviteNum) {

               rank = userRank;

               extra.setRankOrderNum(inviteOrderNum);

               extra.setRankSource(Constant.TYPE_INVITE);

            }



            // 分享满足

            int shareNum = userRank.getShareNum();

            if (sharedOrderNum >= shareNum) {

               rank = userRank;

               extra.setRankOrderNum(sharedOrderNum);

               extra.setRankSource(Constant.TYPE_SHAER);

            }



            // 返利满足

            int directNum = userRank.getRebateNum();

            if (selfOrderNum >= directNum) {

               rank = userRank;

               extra.setRankOrderNum(selfOrderNum);

               extra.setRankSource(Constant.TYPE_REBATE);

            }



            if (rank != null) {

               extra.setUserRank(rank);

            } else {

               rank = userRank;

               // 默认最低等级:青铜

               extra.setUserRank(rank);

               extra.setRankOrderNum(selfOrderNum);

               extra.setRankSource(Constant.TYPE_REBATE);

               break;

            }

         }



         // 等级更新时间 当月1号

         Calendar calendar1 = Calendar.getInstance();

         calendar1.set(Calendar.DAY_OF_MONTH, 1);

         extra.setRankUpdateTime(calendar1.getTime());



         // 保存等级信息

         saveUserInfoExtra(extra);



         // 保存等级变化记录

         UserRankRecord userRankRecord = new UserRankRecord();

         userRankRecord.setUid(uid);

         userRankRecord.setRankId(rank.getId());

         userRankRecord.setCreateTime(new Date());

         userRankRecordMapper.insertSelective(userRankRecord);

      }

   }



@@ -189,101 +272,10 @@
      userInfoExtra.setFirstLoginTime(new Date());

      userInfoExtra.setCreateTime(new Date());

      userInfoExtra.setUpdateTime(new Date());

      userInfoExtra.setActiveTime(new Date());

      userInfoExtraMapper.insertSelective(userInfoExtra);

   }



   @Override

   public UserInfoExtra updateRank(UserInfoExtra userInfoExtra) throws UserInfoExtraException {

      UserInfo userInfo = userInfoExtra.getUserInfo();

      if (userInfo == null) {

         throw new UserInfoExtraException(1, "用户不存在");

      }



      Long uid = userInfo.getId();

      if (uid == null) {

         throw new UserInfoExtraException(1, "用户ID不存在");

      }



      List<UserRank> listRank = userRankService.getAllRank();

      if (listRank == null || listRank.size() == 0) {

         throw new UserInfoExtraException(1, "系统等级不存在");

      }



      int selfOrderNum = 0;

      int sharedOrderNum = 0;

      int inviteOrderNum = 0;

      Map<String, Object> map = commonOrderCountService.lastMonthSettleOrderNumber(uid);

      if (map != null) {

         // 返利订单

         if (map.get("totalSelf") != null) {

            selfOrderNum = Integer.parseInt(map.get("totalSelf").toString());

         }



         // 分享订单

         if (map.get("totalShared") != null) {

            sharedOrderNum = Integer.parseInt(map.get("totalShared").toString());

         }



         // 邀请订单

         if (map.get("totalInvite") != null) {

            inviteOrderNum = Integer.parseInt(map.get("totalInvite").toString());

         }

      }



      UserRank rank = null;

      for (UserRank userRank : listRank) {

         // 邀请满足

         Integer inviteNum = userRank.getInviteNum();

         if (inviteOrderNum >= inviteNum) {

            rank = userRank;

            userInfoExtra.setRankOrderNum(inviteOrderNum);

            userInfoExtra.setRankSource(Constant.TYPE_INVITE);

         }



         // 分享满足

         int shareNum = userRank.getShareNum();

         if (sharedOrderNum >= shareNum) {

            rank = userRank;

            userInfoExtra.setRankOrderNum(sharedOrderNum);

            userInfoExtra.setRankSource(Constant.TYPE_SHAER);

         }



         // 返利满足

         int directNum = userRank.getRebateNum();

         if (selfOrderNum >= directNum) {

            rank = userRank;

            userInfoExtra.setRankOrderNum(selfOrderNum);

            userInfoExtra.setRankSource(Constant.TYPE_REBATE);

         }



         if (rank != null) {

            userInfoExtra.setUserRank(rank);

         } else {

            rank = userRank;

            // 默认最低等级:青铜

            userInfoExtra.setUserRank(rank);

            userInfoExtra.setRankOrderNum(selfOrderNum);

            userInfoExtra.setRankSource(Constant.TYPE_REBATE);

            break;

         }

      }



      // 等级更新时间 当月1号

      Calendar calendar1 = Calendar.getInstance();

      calendar1.set(Calendar.DAY_OF_MONTH, 1);

      userInfoExtra.setRankUpdateTime(calendar1.getTime());



      // 保存信息并返回

      saveUserInfoExtra(userInfoExtra);



      UserRankRecord userRankRecord = new UserRankRecord();

      userRankRecord.setUid(uid);

      userRankRecord.setRankId(rank.getId());

      userRankRecord.setCreateTime(new Date());

      userRankRecordMapper.insertSelective(userRankRecord);



      return userInfoExtra;

   }



   @Override

   public UserInfoExtra saveUserInfoExtra(UserInfoExtra userInfoExtra) throws UserInfoExtraException {

@@ -319,31 +311,27 @@
   }



   @Override

   @Transactional(rollbackFor = Exception.class)

   public String activateInviteCode(Long uid, String inviteCode) throws UserInfoExtraException {

      if (uid == null || inviteCode == null) {

         throw new UserInfoExtraException(1, "用户id、邀请码不能为空");

      }



      if (uid == null || inviteCode == null) 
         throw new UserInfoExtraException(1, "激活信息不能为空");

		
      // 被邀请人信息

      UserInfo invitee = userInfoService.selectByPKey(uid);

      if (invitee == null) {

      if (invitee == null)

         throw new UserInfoExtraException(1, "用户不存在");

      }



      // 额外信息

      UserInfoExtra extra = userInfoExtraMapper.getInfoExtraByUid(uid);

      if (extra != null) {

         String inviteCodeHas = extra.getInviteCode();

         if (inviteCodeHas != null && inviteCodeHas.trim().length() > 0) {

            throw new UserInfoExtraException(1, "已存在邀请码,请刷新页面");

         }

      }

      if (extra != null && !StringUtil.isNullOrEmpty(extra.getInviteCode()))

         throw new UserInfoExtraException(1, "已经激活, 无需再次激活");



      // 邀请人信息

      UserInfo inviter = userInfoService.getInfoByPhoneOrInviteCode(inviteCode, inviteCode);

      if (inviter == null) {

      inviteCode = inviteCode.toUpperCase();

      // 邀请码对应用户信息

      UserInfo inviter = userInfoService.getUserInfoByInviteCode(inviteCode);

      if (inviter == null)

         throw new UserInfoExtraException(1, "上级邀请码不存在");

      }



		
      // 绑定关系

      try {

         threeSaleSerivce.bindRelationshipByInviteCode(invitee, inviter);

@@ -356,51 +344,42 @@
         throw new UserInfoExtraException(1, "激活失败");

      }



      // 邀请码有效、生成邀请码

      String code = UserUtil.getInviteCode(uid);

      if (code == null || code.trim().length() == 0) {

      // 生成邀请码

      String mycode = createInviteCode(uid);

      if (StringUtil.isNullOrEmpty(mycode)) 
         throw new UserInfoExtraException(1, "激活码生成失败");

      }



      UserInfoExtra userInfoExtra = new UserInfoExtra();

      userInfoExtra.setUserInfo(invitee);

      userInfoExtra.setInviteCode(code);



      // 保存额外信息

      saveUserInfoExtra(userInfoExtra);



      Long id = userInfoExtra.getId();

      if (id == null) {

         throw new UserInfoExtraException(1, "激活码生成失败");

      userInfoExtra.setInviteCode(mycode);

      userInfoExtra.setUpdateTime(new Date());

      if (extra != null) {

         userInfoExtra.setId(extra.getId());

         userInfoExtraMapper.updateByPrimaryKeySelective(userInfoExtra);

      } else {

         userInfoExtra.setCreateTime(new Date());

         userInfoExtraMapper.insertSelective(userInfoExtra);

      }



      return code;

      return mycode;

   }



   @Override

   @Transactional(rollbackFor = Exception.class)

   public void activationInviteWX(WXAccountInfoDTO wxAccount, Long uid, String code) throws UserInfoExtraException {

      if (uid == null || code == null) {

      if (uid == null || code == null)

         throw new UserInfoExtraException(1, "激活信息不完整");

      }



      // 用户信息

		 
      UserInfo invitee = userInfoService.selectByPKey(uid);

      if (invitee == null) {

      if (invitee == null)

         throw new UserInfoExtraException(1, "用户不存在");

      }



      // 用户额外信息

      // 额外信息

      UserInfoExtra extra = userInfoExtraMapper.getInfoExtraByUid(uid);

      if (extra != null) {

         String inviteCodeHas = extra.getInviteCode();

         if (inviteCodeHas != null && inviteCodeHas.trim().length() > 0) {

            throw new UserInfoExtraException(1, "已经激活, 无需再次激活");

         }

      }

      if (extra != null && !StringUtil.isNullOrEmpty(extra.getInviteCode()))

         throw new UserInfoExtraException(1, "已经激活, 无需再次激活");



      // 获取微信信息

      WeiXinUser weiXinUser = WXLoginUtil.getWeiXinUserWithSavePortrait(code, wxAccount.getAppId(),

            wxAccount.getAppSecret());

      WeiXinUser weiXinUser = WXLoginUtil.getWeiXinUser(code, wxAccount.getAppId(), wxAccount.getAppSecret());

      if (weiXinUser == null) {

         throw new UserInfoExtraException(1, "微信授权失败");

      }

@@ -438,7 +417,20 @@
         String defaultPortrait = Constant.systemCommonConfig.getDefaultPortrait();

         if (!StringUtil.isNullOrEmpty(weiXinUser.getHeadimgurl()) && !StringUtil.isNullOrEmpty(defaultPortrait)

               && defaultPortrait.equals(invitee.getPortrait())) {

            updateUserInfo.setPortrait(weiXinUser.getHeadimgurl());

            String headimgurl = weiXinUser.getHeadimgurl();

            InputStream asInputStream = HttpUtil.getAsInputStream(headimgurl);

            if (asInputStream == null) {

               LogHelper.test("微信头像下载失败: " + weiXinUser.getUnionid() + " " + headimgurl);

            } else {

               FileUploadResult result = COSManager.getInstance().uploadFile(asInputStream, String.format(

                     "/portrait/wx/%s_%s.jpg", weiXinUser.getUnionid(), System.currentTimeMillis() + ""));

               if (result != null && !StringUtil.isNullOrEmpty(result.getUrl())) {

                  headimgurl = result.getUrl();

               } else {

                  LogHelper.test("微信头像上传失败: " + weiXinUser.getUnionid() + " " + headimgurl);

               }

            }

            updateUserInfo.setPortrait(headimgurl);

         }



         userInfoService.updateByPrimaryKeySelective(updateUserInfo);

@@ -447,31 +439,14 @@
         throw new UserInfoExtraException(1, "绑定微信与激活微信不一致");

      }



      // 邀请人ID -1.5.3新版

      // 是否存在邀请人

      Long inviterId = userInviteRecordService.getNewestInviterId(wxUnionId);



      // 兼容1.5.3 之前版本

      ThreeSale threeSale = threeSaleSerivce.getByWorkerId(uid);

      if (inviterId == null && threeSale != null) {

         Long bossId = threeSale.getBoss().getId();

         UserInfoExtra inviterExtra = userInfoExtraMapper.getInfoExtraByUid(bossId);

         if (inviterExtra != null && inviterExtra.getInviteCode() == null) {

            // 更新邀请码

            UserInfoExtra inviterInfoExtra = new UserInfoExtra();

            inviterInfoExtra.setId(inviterExtra.getId());

            inviterInfoExtra.setInviteCode(UserUtil.getInviteCode(bossId));

            userInfoExtraMapper.updateByPrimaryKeySelective(inviterInfoExtra);

         }

      }



      // 没有被邀请过

      if (inviterId == null && threeSale == null) {

      if (inviterId == null)

         throw new UserInfoExtraException(1, "没有对应的邀请关系");

      }



      // 绑定关系

      try {

         threeSaleSerivce.bindRelationshipByWX(invitee, inviterId, threeSale);

         threeSaleSerivce.bindRelationshipByWX(invitee, inviterId);

      } catch (ThreeSaleException e) {

         try {

            LogHelper.errorDetailInfo(e);

@@ -481,12 +456,11 @@
         throw new UserInfoExtraException(1, "关系绑定时失败");

      }



      // 邀请码有效、生成邀请码

      String inviteCode = UserUtil.getInviteCode(uid);

      if (inviteCode == null || inviteCode.trim().length() == 0) {

      // 生成邀请码

      String inviteCode = createInviteCode(uid);

      if (StringUtil.isNullOrEmpty(inviteCode)) 
         throw new UserInfoExtraException(1, "激活码生成失败");

      }

      // 保存邀请码

		
      UserInfoExtra userInfoExtra = new UserInfoExtra();

      userInfoExtra.setUserInfo(invitee);

      userInfoExtra.setInviteCode(inviteCode);

@@ -501,8 +475,157 @@
   }



   @Override

   @Transactional(rollbackFor = Exception.class)

   public void activeInviteWX(Long uid, WeiXinUser weiXinUser) throws UserInfoExtraException {

      if (uid == null || weiXinUser == null) {

         throw new UserInfoExtraException(1, "激活信息不完整");

      }



      // 用户信息

      UserInfo invitee = userInfoService.selectByPKey(uid);

      if (invitee == null) {

         throw new UserInfoExtraException(1, "用户不存在");

      }



      // 用户额外信息

      UserInfoExtra extra = userInfoExtraMapper.getInfoExtraByUid(uid);

      if (extra != null) {

         String inviteCodeHas = extra.getInviteCode();

         if (inviteCodeHas != null && inviteCodeHas.trim().length() > 0) {

            throw new UserInfoExtraException(1, "已经激活, 无需再次激活");

         }

      }



      String wxUnionId = weiXinUser.getUnionid();

      if (wxUnionId == null || wxUnionId.trim().length() == 0) {

         throw new UserInfoExtraException(1, "微信授权失败");

      }



      // 验证数据

      String wxUnionIdExist = invitee.getWxUnionId();

      if (StringUtil.isNullOrEmpty(wxUnionIdExist)) {

         UserInfo newUser = userInfoService.getEffectiveUserInfoByWXUnionId(weiXinUser.getUnionid());

         if (newUser != null) {

            throw new UserInfoExtraException(1, "此微信已被其他帐号绑定");

         }



         // 自动绑定微信

         UserInfo updateUserInfo = new UserInfo(uid);

         updateUserInfo.setWxOpenId(weiXinUser.getOpenid());

         updateUserInfo.setWxUnionId(weiXinUser.getUnionid());

         updateUserInfo.setWxName(weiXinUser.getNickname());

         updateUserInfo.setWxPic(weiXinUser.getHeadimgurl());



         // 将默认昵称替换成微信昵称

         String defaultNickName = Constant.systemCommonConfig.getDefaultNickName();

         if (!StringUtil.isNullOrEmpty(defaultNickName) && !StringUtil.isNullOrEmpty(weiXinUser.getNickname())) {

            defaultNickName = defaultNickName + uid;

            if ((defaultNickName.equals(invitee.getNickName()) || "返利券".equals(invitee.getNickName())

                  || invitee.getNickName().startsWith(Constant.systemCommonConfig.getDefaultNickName()))) {

               updateUserInfo.setNickName(weiXinUser.getNickname());

            }

         }



         // 将默认头像替换成微信头像

         String defaultPortrait = Constant.systemCommonConfig.getDefaultPortrait();

         if (!StringUtil.isNullOrEmpty(weiXinUser.getHeadimgurl()) && !StringUtil.isNullOrEmpty(defaultPortrait)

               && defaultPortrait.equals(invitee.getPortrait())) {

            String headimgurl = weiXinUser.getHeadimgurl();

            InputStream asInputStream = HttpUtil.getAsInputStream(headimgurl);

            if (asInputStream == null) {

               LogHelper.test("微信头像下载失败: " + weiXinUser.getUnionid() + " " + headimgurl);

            } else {

               FileUploadResult result = COSManager.getInstance().uploadFile(asInputStream, String.format(

                     "/portrait/wx/%s_%s.jpg", weiXinUser.getUnionid(), System.currentTimeMillis() + ""));

               if (result != null && !StringUtil.isNullOrEmpty(result.getUrl())) {

                  headimgurl = result.getUrl();

               } else {

                  LogHelper.test("微信头像上传失败: " + weiXinUser.getUnionid() + " " + headimgurl);

               }

            }

            updateUserInfo.setPortrait(headimgurl);

         }



         userInfoService.updateByPrimaryKeySelective(updateUserInfo);



      } else if (!wxUnionId.equals(wxUnionIdExist)) {

         throw new UserInfoExtraException(1, "绑定微信与激活微信不一致");

      }



      // 邀请人ID -1.5.3新版

      Long inviterId = userInviteRecordService.getNewestInviterId(wxUnionId);

      if (inviterId == null) {

         throw new UserInfoExtraException(1, "没有对应的邀请关系");

      }



      // 绑定关系

      try {

         threeSaleSerivce.bindRelationshipByWX(invitee, inviterId);

      } catch (ThreeSaleException e) {

         try {

            LogHelper.errorDetailInfo(e);

         } catch (Exception e1) {

            e1.printStackTrace();

         }

         throw new UserInfoExtraException(1, "关系绑定时失败");

      }



      // 生成邀请码

      String inviteCode = createInviteCode(uid);;

      if (StringUtil.isNullOrEmpty(inviteCode)) 
         throw new UserInfoExtraException(1, "激活码生成失败");

		
      UserInfoExtra userInfoExtra = new UserInfoExtra();

      userInfoExtra.setUserInfo(invitee);

      userInfoExtra.setInviteCode(inviteCode);

      userInfoExtra.setUpdateTime(new Date());

      if (extra != null) {

         userInfoExtra.setId(extra.getId());

         userInfoExtraMapper.updateByPrimaryKeySelective(userInfoExtra);

      } else {

         userInfoExtra.setCreateTime(new Date());

         userInfoExtraMapper.insertSelective(userInfoExtra);

      }

   }



   @Override

   public void updateActiveTime(Long uid, Date date) {

      UserInfoExtra userInfoExtra = new UserInfoExtra();

      userInfoExtra.setUserInfo(new UserInfo(uid));

      userInfoExtra.setActiveTime(date);

      userInfoExtraMapper.updateInfoExtraByUid(userInfoExtra);

   }



   @Override

   public UserInfoExtra getUserInfoExtra(Long uid) {

      return userInfoExtraMapper.getInfoExtraByUid(uid);

   }



   @Override

   public UserInfoExtra getUserALLInfo(Long uid) {

      return userInfoExtraMapper.gerUserRank(uid);

   }



   @Override

   @Transactional

   public UserInfoExtra getByUidForUpdate(Long uid) {

      return userInfoExtraMapper.getByUidForUpdate(uid);

   }



   @Override

   @Transactional

   public void updateGoldCoin(Long id, Integer goldCoin) {

      userInfoExtraMapper.updateGoldCoin(id, goldCoin);

   }

	
   @Override

   public void addGoldCoinByUid(Long uid, Integer goldCoin) {

      userInfoExtraMapper.addGoldCoinByUid(uid, goldCoin);

   }

	
   @Override

   public void updateByPrimaryKeySelective(UserInfoExtra record) {

      userInfoExtraMapper.updateByPrimaryKeySelective(record);

   }



   @Override

@@ -511,38 +634,53 @@
   }



   @Override

   public String getUserInviteCode(Long uid) throws UserInfoExtraException {



   public String getInviteCodeByUid(Long uid) {

      UserInfoExtra userInfoExtra = userInfoExtraMapper.getInfoExtraByUid(uid);

      if (userInfoExtra == null) {

         userInfoExtra = new UserInfoExtra();

      }



      // 邀请码

      String inviteCode = userInfoExtra.getInviteCode();



      if (inviteCode == null || inviteCode.trim().length() == 0) {

         // 判断用户

      if (userInfoExtra != null) {

         // vip邀请码优先

         if (!StringUtil.isNullOrEmpty(userInfoExtra.getInviteCodeVip()))

            return userInfoExtra.getInviteCodeVip();

			
         if (!StringUtil.isNullOrEmpty(userInfoExtra.getInviteCode()))

            return userInfoExtra.getInviteCode();

			
         // 是否满足生产邀请码

         int relationshipNum = threeSaleSerivce.getSuccessRelationshipNum(uid);



         if (relationshipNum > 0) {

            // 邀请码有效、生成邀请码

            inviteCode = UserUtil.getInviteCode(uid);

            if (inviteCode == null || inviteCode.trim().length() == 0) {

               throw new UserInfoExtraException(1, "激活码生成失败");

            try {

               String inviteCode = createInviteCode(uid);

               if (!StringUtil.isNullOrEmpty(inviteCode)) {

                  UserInfoExtra extra= new UserInfoExtra();

                  extra.setId(userInfoExtra.getId());

                  extra.setInviteCode(inviteCode);

                  userInfoExtraMapper.updateByPrimaryKeySelective(extra);

                  return inviteCode;

               }

            } catch (Exception e) {

               LogHelper.errorDetailInfo(e);

            }



            // 保存邀请码

            userInfoExtra.setUserInfo(new UserInfo(uid));

            userInfoExtra.setInviteCode(inviteCode);

            // 保存附加信息

            saveUserInfoExtra(userInfoExtra);

         }

      }



      return inviteCode;

      return "";

   }



   /**

    * 生产邀请码

    * @param uid

    * @return

    */

   private String createInviteCode(Long uid) {

      String inviteCode = null;

      for (int i = 1; i <= 5; i ++) {

         inviteCode = UserUtil.getInviteCode(i, uid);

         long countCode = userInfoExtraMapper.countByInviteCode(inviteCode);

         if (countCode <= 0)

            break;

      }

      return inviteCode;

   }

	
	
   @Override

   public long countByRankId(Long rankId) {

      return userInfoExtraMapper.countByRankId(rankId);

@@ -584,21 +722,20 @@


   @Override

   public UserInfo getUserByInviteCode(String inviteCode) throws UserInfoExtraException {

      if (StringUtil.isNullOrEmpty(inviteCode)) {

      if (StringUtil.isNullOrEmpty(inviteCode))

         throw new UserInfoExtraException(1, "邀请码不能为空");

      }

      // 邀请人信息

      UserInfo inviter = userInfoService.getInfoByPhoneOrInviteCode(inviteCode, inviteCode);

      if (inviter == null) {

		 
      UserInfo inviter = userInfoService.getUserInfoByInviteCode(inviteCode);

      if (inviter == null)

         throw new UserInfoExtraException(1, "上级邀请码不存在");

      }

		
      return inviter;

   }



   @Override

   public UserInfo getInviterInfo(WXAccountInfoDTO wxAccount, Long uid, String code) throws UserInfoExtraException {

      if (code == null) {

         throw new UserInfoExtraException(1, "code信息不完整");

   public UserInfo getInviterInfo(Long uid, String wxUnionId) throws UserInfoExtraException {

      if (wxUnionId == null) {

         throw new UserInfoExtraException(1, "unionId信息不完整");

      }



      // 用户信息

@@ -616,20 +753,10 @@
         }

      }



      // 获取微信信息

      WeiXinUser weiXinUser = WXLoginUtil.getWeiXinUser(code, wxAccount.getAppId(), wxAccount.getAppSecret());

      if (weiXinUser == null) {

         throw new UserInfoExtraException(1, "微信授权失败");

      }

      String wxUnionId = weiXinUser.getUnionid();

      if (wxUnionId == null || wxUnionId.trim().length() == 0) {

         throw new UserInfoExtraException(1, "微信授权失败");

      }



      // 验证数据

      String wxUnionIdExist = invitee.getWxUnionId();

      if (StringUtil.isNullOrEmpty(wxUnionIdExist)) {

         UserInfo newUser = userInfoService.getEffectiveUserInfoByWXUnionId(weiXinUser.getUnionid());

         UserInfo newUser = userInfoService.getEffectiveUserInfoByWXUnionId(wxUnionId);

         if (newUser != null) {

            throw new UserInfoExtraException(1, "此微信已被其他帐号绑定");

         }

@@ -649,4 +776,46 @@
      }

      return inviter;

   }



   @Override

   public void updateAutoExtract(Integer days) {

      userInfoExtraMapper.updateAutoExtract(days);

   }



   @Transactional

   @Override

   public void updateInviteCodeVip(String inviteCodeVip, Long uid) throws UserInfoExtraException {

      if (inviteCodeVip == null || inviteCodeVip.length() < 4 || inviteCodeVip.length() > 12)

         throw new UserInfoExtraException(1, "邀请码必须为4到12位");

		
      UserInfoExtra extra = getUserInfoExtra(uid);

      if (extra == null)

         throw new UserInfoExtraException(2, "用户信息不存在");



      if (StringUtil.isNullOrEmpty(extra.getInviteCode()))

         throw new UserInfoExtraException(3, "邀请码尚未激活");

		
      if (!StringUtil.isNullOrEmpty(extra.getInviteCodeVip()))

         throw new UserInfoExtraException(4, "只能修改一次");

		
      // 转换成大写

      inviteCodeVip = inviteCodeVip.toUpperCase();

		
      // 验证特殊码

      if(InviteCodeFilterUtil.isSpecialCode(inviteCodeVip))

         throw new UserInfoExtraException(5, "邀请码已存在");

		
      if (inviteCodeVip.equalsIgnoreCase(extra.getInviteCode()))

         throw new UserInfoExtraException(5, "不能为原邀请码");

		
      long count = userInfoExtraMapper.countByInviteCode(inviteCodeVip);

      if (count > 0)

         throw new UserInfoExtraException(6, "邀请码已存在");

		
      UserInfoExtra update = new UserInfoExtra();

      update.setId(extra.getId());

      update.setInviteCodeVip(inviteCodeVip);

      update.setUpdateTime(new Date());

      userInfoExtraMapper.updateByPrimaryKeySelective(update);

   }

}