package com.yeshi.fanli.service.impl.user.vip;
|
|
import java.util.ArrayList;
|
import java.util.Date;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
|
import javax.annotation.Resource;
|
|
import org.springframework.stereotype.Service;
|
|
import com.yeshi.fanli.dao.mybatis.user.vip.TeamUserLevelStatisticMapper;
|
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.UserInviteValidRecordService;
|
import com.yeshi.fanli.service.inter.user.vip.TeamUserLevelStatisticService;
|
import com.yeshi.fanli.service.manger.user.UserLevelManager;
|
|
@Service
|
public class TeamUserLevelStatisticServiceImpl implements TeamUserLevelStatisticService {
|
|
@Resource
|
private UserLevelManager userLevelManager;
|
|
@Resource
|
private TeamUserLevelStatisticMapper teamUserLevelStatisticMapper;
|
|
@Resource
|
private UserInviteValidRecordService userInviteValidRecordService;
|
|
@Override
|
public TeamUserLevelStatistic selectByUid(Long uid) {
|
return teamUserLevelStatisticMapper.selectByPrimaryKey(uid);
|
}
|
|
@Override
|
public List<TeamUserLevelStatistic> listByUids(List<Long> uids) {
|
|
List<TeamUserLevelStatistic> resultList = new ArrayList<>();
|
|
List<TeamUserLevelStatistic> list = teamUserLevelStatisticMapper.listByUids(uids);
|
// 放入Map中
|
Map<Long, TeamUserLevelStatistic> map = new HashMap<>();
|
if (list != null)
|
for (TeamUserLevelStatistic s : list) {
|
map.put(s.getId(), s);
|
}
|
|
for (Long uid : uids) {
|
if (map.get(uid) == null) {
|
initData(uid);
|
TeamUserLevelStatistic statistic = selectByUid(uid);
|
resultList.add(statistic);
|
} else {
|
resultList.add(map.get(uid));
|
}
|
}
|
|
return resultList;
|
}
|
|
@Override
|
public void initData(Long uid) {
|
int pageSize = 1000;
|
|
// 直接粉丝统计
|
int daRenFirstCount = 0;
|
int highFirstCount = 0;
|
int superFirstCount = 0;
|
int tearcherFirstCount = 0;
|
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.highVIP) {
|
highFirstCount++;
|
} else if (level == UserLevelEnum.superVIP) {
|
superFirstCount++;
|
} else if (level == UserLevelEnum.tearcher) {
|
tearcherFirstCount++;
|
}
|
}
|
}
|
|
// 间接粉丝统计
|
int daRenSecondCount = 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;
|
}
|
|
for (UserInviteValidRecord ts : list) {
|
UserLevelEnum level = userLevelManager.getUserLevel(ts.getWorkerId());
|
if (level == UserLevelEnum.daRen) {
|
daRenSecondCount++;
|
} else if (level == UserLevelEnum.highVIP) {
|
highSecondCount++;
|
} else if (level == UserLevelEnum.superVIP) {
|
superSecondCount++;
|
} else if (level == UserLevelEnum.tearcher) {
|
tearcherSecondCount++;
|
}
|
}
|
}
|
|
UserLevelEnum level = userLevelManager.getUserLevel(uid);
|
TeamUserLevelStatistic statistic = new TeamUserLevelStatistic();
|
statistic.setDaRenFirstCount(daRenFirstCount);
|
statistic.setDaRenSecondCount(daRenSecondCount);
|
statistic.setHighFirstCount(highFirstCount);
|
statistic.setHighSecondCount(highSecondCount);
|
statistic.setSuperFirstCount(superFirstCount);
|
statistic.setSuperSecondCount(superSecondCount);
|
statistic.setTearcherFirstCount(tearcherFirstCount);
|
statistic.setTearcherSecondCount(tearcherSecondCount);
|
statistic.setId(uid);
|
statistic.setLevel(level);
|
add(statistic);
|
}
|
|
@Override
|
public void add(TeamUserLevelStatistic statistic) {
|
TeamUserLevelStatistic old = selectByUid(statistic.getId());
|
if (old == null) {
|
if (statistic.getCreateTime() == null)
|
statistic.setCreateTime(new Date());
|
teamUserLevelStatisticMapper.insertSelective(statistic);
|
} else {
|
statistic.setUpdateTime(new Date());
|
teamUserLevelStatisticMapper.updateByPrimaryKeySelective(statistic);
|
}
|
}
|
|
@Override
|
public void updateUserLevel(Long uid) {
|
UserLevelEnum level = userLevelManager.getUserLevel(uid);
|
TeamUserLevelStatistic statistic = new TeamUserLevelStatistic();
|
statistic.setId(uid);
|
statistic.setLevel(level);
|
statistic.setUpdateTime(new Date());
|
teamUserLevelStatisticMapper.updateByPrimaryKeySelective(statistic);
|
}
|
|
}
|