fix

Helius

2021-07-15 73204530b7c20b4b90f7277d34c7eb17c53021aa

 gc-user/src/main/java/com/xzx/gc/user/service/DistribService.java

@@ -1,35 +1,53 @@
package com.xzx.gc.user.service;

import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.convert.Convert;
import cn.hutool.core.date.DateUnit;
import cn.hutool.core.date.DateUtil;
import com.github.pagehelper.PageHelper;
import com.github.pagehelper.PageInfo;
import com.xzx.gc.common.exception.RestException;
import com.xzx.gc.common.utils.StringUtils;
import com.xzx.gc.entity.AccountInfo;
import com.xzx.gc.entity.SysMessage;
import com.xzx.gc.entity.UserHeadRelate;
import com.xzx.gc.entity.UserShareInfo;
import com.xzx.gc.user.dto.HeadProfitLitDto;
import com.xzx.gc.user.dto.HeadTeamDto;
import com.xzx.gc.user.mapper.*;
import com.xzx.gc.user.vo.HeadInfoVo;
import com.xzx.gc.user.vo.HeadProfitListVo;
import com.xzx.gc.user.vo.HeadTeamVo;
import com.xzx.gc.user.dto.AdddistribDto;
import com.xzx.gc.user.dto.DistribInfoListDto;
import com.xzx.gc.user.dto.DistribListDto;
import com.xzx.gc.user.mapper.AccountMapper;
import com.xzx.gc.user.mapper.SysMessageMapper;
import com.xzx.gc.user.mapper.UserHeadRelateMapper;
import com.xzx.gc.user.mapper.UserShareInfoMapper;
import com.xzx.gc.user.vo.HeadInfoVo;
import com.xzx.gc.user.vo.*;
import cn.hutool.core.util.ObjectUtil;
import cn.hutool.core.util.StrUtil;
import com.xzx.gc.entity.RedPaperRule;
import com.xzx.gc.model.JsonResult;
import com.xzx.gc.user.dto.UpdateSettingDto;
import com.xzx.gc.user.mapper.RedPaperRuleMapper;
import com.xzx.gc.user.vo.ViewSettingVo;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import javax.annotation.Resource;
import java.math.BigDecimal;
import java.util.HashMap;
import java.util.List;

import java.math.BigDecimal;
import java.util.Date;
import java.util.List;
import java.util.Map;

@Slf4j
@Service
@Transactional
public class DistribService {
@@ -38,7 +56,8 @@
    private RedPaperRuleMapper redPaperRuleMapper;
    @Autowired
    private UserHeadRelateMapper userHeadRelateMapper;

    @Autowired
    private UserHeadDetailsMapper userHeadDetailsMapper;
    @Autowired
    private AccountMapper accountMapper;
    @Autowired
@@ -87,6 +106,7 @@


    public void join(String headId, String userId) {
        log.info("#加入团队:{}, {}#", headId, userId);
        AccountInfo headInfo = accountMapper.selectOneByUserId(headId);
        if (AccountInfo.IS_HEAD_N.equals(headInfo.getIsHead())) {
            throw new RestException(-3, "不是团长");
@@ -111,14 +131,17 @@
        userHeadRelate.setHeadUserId(headId);
        userHeadRelate.setUserId(userId);
        userHeadRelate.setScore(BigDecimal.ZERO);
        userHeadRelate.setCreatedTime(new Date());
        userHeadRelate.setAmount(BigDecimal.ZERO);
        userHeadRelateMapper.insert(userHeadRelate);

        SysMessage sysMessage = new SysMessage();
        sysMessage.setCreateTime(DateUtil.format(new Date(), "yyyy-MM-dd HH:mm:ss"));
        sysMessage.setCreateUserId(userId);
        sysMessage.setCreateUserId(headId);
        sysMessage.setUserId(headId);
        sysMessage.setMobilePhone(accountInfo.getAccountName());
        sysMessage.setMessageType("4");
        sysMessage.setMessageSubTypeName("团队消息");
        sysMessage.setMessage("手机号:" + accountInfo.getAccountName() + "的用户, 成功加入您的团队");
        sysMessage.setFlag("2");
        sysMessageMapper.insert(sysMessage);
@@ -134,6 +157,16 @@

        headInfoVo.setIsHead(accountInfo.getIsHead());
        return headInfoVo;
    }

    public List<HeadProfitListVo> findHeadProfitListInPage(HeadProfitLitDto profitLitDto) {
        PageHelper.startPage(profitLitDto.getPage(), profitLitDto.getLimit());
        return userHeadDetailsMapper.selectHeadProfitListInPage(profitLitDto.getUserId());
    }

    public List<HeadTeamVo> findHeadTeamListInPage(HeadTeamDto teamDto) {
        PageHelper.startPage(teamDto.getPage(), teamDto.getLimit());
        return userHeadRelateMapper.selectHeadTeamListInPage(teamDto.getUserId());
    }

    public void addSetting(UpdateSettingDto model) {
@@ -180,4 +213,59 @@
            redPaperRuleMapper.updateByPrimaryKey(redPaperRule);
        }
    }

    public Map<String, Object> distribList(DistribListDto distribListDto) {
        PageHelper.startPage(distribListDto.getPage(), distribListDto.getLimit());
        if(StrUtil.isNotEmpty(distribListDto.getQueryCol())){
            String decode = StringUtils.encode(distribListDto.getQueryCol());
            distribListDto.setName(decode);
            distribListDto.setPhone(distribListDto.getQueryCol());
        }
        List<DistribListVo> distribListVos = accountMapper.selectDsitribListByQueryCol(distribListDto);
        if(CollUtil.isNotEmpty(distribListVos)){
            for(DistribListVo distribListVo : distribListVos){
                String nickname = distribListVo.getNickname();
                String decode = StringUtils.decode(nickname);
                distribListVo.setNickname(decode);
            }
        }
        PageInfo<DistribListVo> pageInfo = new PageInfo<>(distribListVos);
        Map<String, Object> data = new HashMap<>();
        int count = Convert.toInt(pageInfo.getTotal());
        data.put("data", distribListVos);
        data.put("count", count);
        data.put("code", 0);
        return data;
    }

    public void adddistrib(AdddistribDto model) {
        String id = model.getId();
        AccountInfo accountInfo = accountMapper.selectByPrimaryKey(id);
        accountInfo.setIsHead(1);
        accountMapper.updateByPrimaryKey(accountInfo);
    }

    public Map<String, Object> distribInfoList(DistribInfoListDto distribInfoListDto) {
        PageHelper.startPage(distribInfoListDto.getPage(), distribInfoListDto.getLimit());
        if(StrUtil.isNotEmpty(distribInfoListDto.getName())){
            String decode = StringUtils.encode(distribInfoListDto.getName());
            distribInfoListDto.setName(decode);
        }
        List<DistribInfoListVo> distribInfoListVos = accountMapper.selectDsitribInfoListByQueryCol(distribInfoListDto);
        if(CollUtil.isNotEmpty(distribInfoListVos)){
            for(DistribInfoListVo distribInfoListVo : distribInfoListVos){
                String nickname = distribInfoListVo.getNickname();
                String decode = StringUtils.decode(nickname);
                distribInfoListVo.setNickname(decode);
            }
        }
        PageInfo<DistribInfoListVo> pageInfo = new PageInfo<>(distribInfoListVos);
        Map<String, Object> data = new HashMap<>();
        int count = Convert.toInt(pageInfo.getTotal());
        data.put("data", distribInfoListVos);
        data.put("count", count);
        data.put("code", 0);
        return data;

    }
}