yujian
2019-12-19 626d711cb15896055c13fe344eb7fcc824589715
fanli/src/main/java/com/yeshi/fanli/controller/client/v2/InviteControllerV2.java
@@ -5,14 +5,18 @@
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.Map;
import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.yeshi.utils.DateUtil;
import org.yeshi.utils.JsonUtil;
import com.yeshi.fanli.dto.wx.WXAccountInfoDTO;
import com.yeshi.fanli.entity.accept.AcceptData;
import com.yeshi.fanli.entity.bus.msg.UserSystemMsg;
import com.yeshi.fanli.entity.bus.msg.UserSystemMsgTypeEnum;
@@ -20,16 +24,31 @@
import com.yeshi.fanli.entity.bus.user.ThreeSaleExtraInfo;
import com.yeshi.fanli.entity.bus.user.UserInfo;
import com.yeshi.fanli.entity.bus.user.UserInfoExtra;
import com.yeshi.fanli.entity.bus.user.UserInviteSeparate;
import com.yeshi.fanli.entity.bus.user.WeiXinUser;
import com.yeshi.fanli.entity.system.BusinessSystem;
import com.yeshi.fanli.entity.system.SystemClientParams;
import com.yeshi.fanli.exception.user.UserAccountException;
import com.yeshi.fanli.exception.user.UserInfoExtraException;
import com.yeshi.fanli.log.LogHelper;
import com.yeshi.fanli.service.inter.config.ConfigService;
import com.yeshi.fanli.service.inter.config.SpreadImgService;
import com.yeshi.fanli.service.inter.config.SystemClientParamsService;
import com.yeshi.fanli.service.inter.msg.UserSystemMsgService;
import com.yeshi.fanli.service.inter.push.PushService;
import com.yeshi.fanli.service.inter.user.SpreadUserImgService;
import com.yeshi.fanli.service.inter.user.UserInfoExtraService;
import com.yeshi.fanli.service.inter.user.UserInfoService;
import com.yeshi.fanli.service.inter.user.UserInviteSeparateService;
import com.yeshi.fanli.service.inter.user.invite.ThreeSaleExtraInfoSerivce;
import com.yeshi.fanli.service.inter.user.invite.ThreeSaleSerivce;
import com.yeshi.fanli.service.inter.user.vip.UserVIPInfoService;
import com.yeshi.fanli.util.Constant;
import com.yeshi.fanli.util.RedisManager;
import com.yeshi.fanli.util.StringUtil;
import com.yeshi.fanli.util.VersionUtil;
import com.yeshi.fanli.util.annotation.RequestSerializableByKey;
import com.yeshi.fanli.util.wx.WXLoginUtil;
import net.sf.json.JSONArray;
import net.sf.json.JSONObject;
@@ -37,6 +56,9 @@
@Controller
@RequestMapping("api/v2/invite")
public class InviteControllerV2 {
   @Resource
   private RedisManager redisManager;
   @Resource
   private ConfigService configService;
@@ -59,6 +81,23 @@
   @Resource
   private ThreeSaleExtraInfoSerivce threeSaleExtraInfoSerivce;
   @Resource
   private UserVIPInfoService userVIPInfoService;
   @Resource
   private UserInviteSeparateService userInviteSeparateService;
   @Resource
   private SpreadImgService spreadImgService;
   @Resource
   private SpreadUserImgService spreadUserImgService;
   @Resource
   private SystemClientParamsService systemClientParamsService;
   /**
    * 邀请码验证
    * 
@@ -79,22 +118,71 @@
   }
   /**
    * 邀请码微信
    * 邀请码微信验证
    * 
    * @param out
    */
   @RequestMapping(value = "verifyWX")
   public void verifyWX(AcceptData acceptData, Long uid, String code, PrintWriter out) {
      try {
         UserInfo userInfo = userInfoExtraService.getInviterInfo(
               Constant.getWXAccount(acceptData.getPlatform(), acceptData.getVersion()), uid, code);
         WXAccountInfoDTO wxAccount = Constant.getWXAccount(acceptData.getPlatform(), acceptData.getVersion());
         // 获取微信信息
         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 activeWX = StringUtil.Md5("activeWX:" + weiXinUser.getUnionid());
         redisManager.cacheCommonString(activeWX, JsonUtil.getSimpleGson().toJson(weiXinUser), 60 * 20);
         // 邀请关系用户信息
         UserInfo userInfo = userInfoExtraService.getInviterInfo(uid, wxUnionId);
         JSONObject data = new JSONObject();
         data.put("nickName", userInfo.getNickName());
         data.put("portrait", userInfo.getPortrait());
         data.put("tips", "确认要成为TA的一级队员吗?邀请关系一旦建立不可更改。");
         data.put("key", activeWX);
         out.print(JsonUtil.loadTrueResult(data));
      } catch (UserInfoExtraException e) {
         out.print(JsonUtil.loadFalseResult(e.getMsg()));
      }
   }
   /**
    * 微信激活
    *
    * @param acceptData
    * @param uid
    * @param key
    *            验证返回的key
    * @param out
    */
   @RequestSerializableByKey(key = "'activeInviteWX-'+#uid")
   @RequestMapping(value = "activeInviteWX")
   public void activeInviteWX(AcceptData acceptData, Long uid, String key, PrintWriter out) {
      try {
         String value = redisManager.getCommonString(key);
         if (StringUtil.isNullOrEmpty(value))
            throw new UserAccountException(1, "微信授权失效");
         WeiXinUser weiXinUser = JsonUtil.getSimpleGson().fromJson(value, WeiXinUser.class);
         if (weiXinUser == null)
            throw new UserAccountException(1, "微信授权失效");
         userInfoExtraService.activeInviteWX(uid, weiXinUser);
         // 删除缓存
         redisManager.removeCommonString(key);
         out.print(JsonUtil.loadTrueResult("邀请码激活成功"));
      } catch (UserInfoExtraException e) {
         out.print(JsonUtil.loadFalseResult(e.getMsg()));
      } catch (Exception e) {
         out.print(JsonUtil.loadFalseResult("邀请码激活失败"));
         e.printStackTrace();
      }
   }
@@ -122,8 +210,12 @@
         count = threeSaleSerivce.countSecondTeam(uid, 1);
      }
      boolean needVIPInfo = false;
      if (VersionUtil.greaterThan_2_1(acceptData.getPlatform(), acceptData.getVersion()))
         needVIPInfo = true;
      // 组织数据
      JSONObject resultData = organizeTeam(count, listTeam, uid, type);
      JSONObject resultData = organizeTeam(count, listTeam, uid, type,needVIPInfo);
      // 第一页判断是否激活 是否有队员
      if (type == 1 && page == 1) {
@@ -148,7 +240,7 @@
    * @param type
    * @return
    */
   private JSONObject organizeTeam(long count, List<ThreeSale> list, Long uid, int type) {
   private JSONObject organizeTeam(long count, List<ThreeSale> list, Long uid, int type, boolean needVipInfo) {
      JSONObject result = new JSONObject();
      JSONArray resultArray = new JSONArray();
@@ -157,6 +249,14 @@
      SimpleDateFormat format = new SimpleDateFormat("yyyy.MM.dd HH:mm");
      SimpleDateFormat formatYears = new SimpleDateFormat("yyyy");
      List<Long> workerIdList = new ArrayList<>();
      Map<Long, Boolean> vipMap = null;
      if (needVipInfo) {
         for (ThreeSale threeSale : list)
            workerIdList.add(threeSale.getWorker().getId());
         if (workerIdList.size() > 0)
            vipMap = userVIPInfoService.listByUids(workerIdList);
      }
      for (ThreeSale threeSale : list) {
         UserInfo worker = threeSale.getWorker();
         if (worker == null) {
@@ -172,7 +272,39 @@
         Date inviteTime = new Date(threeSale.getCreateTime());
         object.put("inviteTime", format.format(inviteTime) + " 加入");
         // 添加是否为VIP
         if (vipMap != null && vipMap.get(worker.getId()) != null) {
            object.put("vip", vipMap.get(worker.getId()));
            if (vipMap.get(worker.getId())) {
               UserInviteSeparate separate = userInviteSeparateService.selectByWorkerIdAndBossId(worker.getId(), uid);
               if (separate != null && separate.getEndTime() != null) {
                  try {
                     JSONObject contentJson1 = new JSONObject();
                     contentJson1.put("color", "#888888");
                     contentJson1.put("content", "将于");
                     JSONObject contentJson2 = new JSONObject();
                     contentJson2.put("color", "#F14242");
                     contentJson2.put("content", DateUtil.dateDiff4(new Date(), separate.getEndTime()));
                     JSONObject contentJson3 = new JSONObject();
                     contentJson3.put("color", "#888888");
                     contentJson3.put("content", "后脱离邀请关系");
                     JSONArray array = new JSONArray();
                     array.add(contentJson1);
                     array.add(contentJson2);
                     array.add(contentJson3);
                     object.put("vipBreakTime",array);
                  } catch (Exception e) {
                     e.printStackTrace();
                  }
               }
            }
         }
         String fontColor1 = "#888888";
         JSONArray array = new JSONArray();
         if (threeSale.getState()) {
@@ -429,4 +561,51 @@
         out.print(JsonUtil.loadFalseResult("提醒失败"));
      }
   }
   @RequestSerializableByKey(key = "'getInviteImgs-'+#uid")
   @RequestMapping(value = "/getInviteImgs", method = RequestMethod.POST)
   public void getInviteImgs(AcceptData acceptData, Long uid, Integer page, PrintWriter out, HttpServletRequest request) {
      if (uid == null || uid <= 0) {
         out.print(JsonUtil.loadFalseResult("用户未登录"));
         return;
      }
      if (page == null || page < 1)
         page = 1;
      try {
         String inviteCode = userInfoExtraService.getInviteCodeByUid(uid);
         if (StringUtil.isNullOrEmpty(inviteCode)) {
            out.print(JsonUtil.loadFalseResult("邀请码未激活"));
            return;
         }
         int count = 3;
         if (page > 1)
            count = Integer.MAX_VALUE;
         // 邀请图
         List<String> imgList = spreadUserImgService.getUserSpreadImgToPage(uid, page, count, request);
         // 邀请规则
         BusinessSystem system = new BusinessSystem();
         system.setId(4L);
         system.setPlatform(1);
         system.setCreatetime(java.lang.System.currentTimeMillis());
         SystemClientParams values = systemClientParamsService.getSystemClientParamsBySystemAndKey(system, "inviteRules");
         JSONObject data = new JSONObject();
         data.put("count", spreadImgService.countAll());
         data.put("inviteCode", inviteCode);
         data.put("helpLink", configService.get("invite_help_link"));
         data.put("inviteRules",  values.getValue().replace("\n", "<br><br>"));
         data.put("imgUrlList", JsonUtil.getSimpleGsonWithDate().toJson(imgList));
         data.put("inviteLink", "http://a.app.qq.com/o/simple.jsp?pkgname=com.yeshi.ec.rebate");
         out.print(JsonUtil.loadTrue(0, JsonUtil.getSimpleGson().toJson(data), "成功"));
      } catch (Exception e) {
         LogHelper.errorDetailInfo(e);
         out.print(JsonUtil.loadFalseResult("邀请信息获取失败"));
      }
   }
}