yujian
2020-04-17 aba192810c97b03ca3ca4aef1d44f0b60aa243e8
fanli/src/main/java/com/yeshi/fanli/service/impl/user/vip/TeamUserLevelStatisticServiceImpl.java
@@ -8,10 +8,10 @@
import org.springframework.stereotype.Service;
import com.yeshi.fanli.dao.mybatis.user.vip.TeamUserLevelStatisticMapper;
import com.yeshi.fanli.entity.bus.user.ThreeSale;
import com.yeshi.fanli.entity.bus.user.UserInviteValidRecord;
import com.yeshi.fanli.entity.bus.user.vip.TeamUserLevelStatistic;
import com.yeshi.fanli.entity.bus.user.vip.UserLevelEnum;
import com.yeshi.fanli.service.inter.user.invite.ThreeSaleSerivce;
import com.yeshi.fanli.service.inter.user.invite.UserInviteValidRecordService;
import com.yeshi.fanli.service.inter.user.vip.TeamUserLevelStatisticService;
import com.yeshi.fanli.service.manger.user.UserLevelManager;
@@ -19,13 +19,14 @@
public class TeamUserLevelStatisticServiceImpl implements TeamUserLevelStatisticService {
   @Resource
   private ThreeSaleSerivce threeSaleSerivce;
   @Resource
   private UserLevelManager userLevelManager;
   @Resource
   private TeamUserLevelStatisticMapper teamUserLevelStatisticMapper;
   @Resource
   private UserInviteValidRecordService userInviteValidRecordService;
   @Override
   public TeamUserLevelStatistic selectByUid(Long uid) {
@@ -34,16 +35,22 @@
   @Override
   public void initData(Long uid) {
      int state = ThreeSale.STATE_SUCCESS;
      long count = threeSaleSerivce.countFirstTeam(uid, state);
      List<ThreeSale> list = threeSaleSerivce.listFirstTeam(0L, (int) count, uid, state);
      int pageSize = 1000;
      // 直接粉丝统计
      int daRenFirstCount = 0;
      int normalFirstCount = 0;
      int highFirstCount = 0;
      int superFirstCount = 0;
      int tearcherFirstCount = 0;
      for (ThreeSale ts : list) {
         UserLevelEnum level = userLevelManager.getUserLevel(ts.getWorker().getId());
      for (int i = 0; i < 100; i ++) {
         List<UserInviteValidRecord> list = userInviteValidRecordService.listFirstTeam(i * pageSize, pageSize, uid);
         if (list == null || list.size() == 0) {
            break;
         }
         for (UserInviteValidRecord ts : list) {
            UserLevelEnum level = userLevelManager.getUserLevel(ts.getWorkerId());
         if (level == UserLevelEnum.daRen) {
            daRenFirstCount++;
         } else if (level == UserLevelEnum.normalVIP) {
@@ -56,21 +63,23 @@
            tearcherFirstCount++;
         }
      }
      }
      // 间接粉丝统计
      int daRenSecondCount = 0;
      int normalSecondCount = 0;
      int highSecondCount = 0;
      int superSecondCount = 0;
      int tearcherSecondCount = 0;
      for (int i = 0; i < 100; i ++) {
         List<UserInviteValidRecord> list = userInviteValidRecordService.listSecondTeam(i * pageSize, pageSize, uid);
         if (list == null || list.size() == 0) {
            break;
         }
      long secondCount = threeSaleSerivce.countSecondTeam(uid, state);
      int pageSize = 1000;
      int toalPage = (int) (secondCount % pageSize == 0 ? secondCount / pageSize : secondCount / pageSize + 1);
      for (int i = 0; i < toalPage; i++) {
         list = threeSaleSerivce.listSecondTeam(i * pageSize, pageSize, uid, state);
         if (list != null)
            for (ThreeSale ts : list) {
               UserLevelEnum level = userLevelManager.getUserLevel(ts.getWorker().getId());
         for (UserInviteValidRecord ts : list) {
            UserLevelEnum level = userLevelManager.getUserLevel(ts.getWorkerId());
               if (level == UserLevelEnum.daRen) {
                  daRenSecondCount++;
               } else if (level == UserLevelEnum.normalVIP) {