青羊经侦大队-数据平台
baizonghao
2023-03-21 e0670c59089d8461b13b7f910fd538e9a5a6edef
src/main/java/com/example/jz/service/impl/CauseServiceImpl.java
@@ -3,11 +3,11 @@
import com.alibaba.excel.EasyExcel;
import com.alibaba.excel.context.AnalysisContext;
import com.alibaba.excel.event.AnalysisEventListener;
import com.baomidou.mybatisplus.core.conditions.Wrapper;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.core.toolkit.support.SFunction;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.example.jz.dao.*;
import com.example.jz.enums.CauseEnums;
@@ -20,6 +20,7 @@
import com.example.jz.modle.entity.*;
import com.example.jz.modle.vo.CauseReportVo;
import com.example.jz.modle.vo.CauseVo;
import com.example.jz.modle.vo.NewCauseVo;
import com.example.jz.modle.vo.UserVo;
import com.example.jz.service.CauseService;
import com.example.jz.service.ReportService;
@@ -34,6 +35,8 @@
import javax.annotation.Resource;
import java.util.*;
import java.util.function.Function;
import java.util.function.Predicate;
import java.util.stream.Collectors;
/**
@@ -155,6 +158,41 @@
    }
    @Override
    public List<NewCauseVo> getVxCauseList(String phone) {
        User user = userDao.selectOne(new LambdaQueryWrapper<User>().eq(User::getUserMobile, phone));
        List<Report> reports = reportDao.selectList(new LambdaQueryWrapper<Report>().eq(Report::getUserId, user.getId()));
        List<Integer> list = reports.stream()
                .map(item -> item.getCauseId())
                .filter(integer -> integer != null)
                .distinct()
                .collect(Collectors.toList());
        List<NewCauseVo> res;
        if (!Objects.isNull(list) && list.size()!= 0){
            List<Cause> causes = causeDao.selectList(new LambdaQueryWrapper<Cause>().notIn(Cause::getId, list));
            res = causes.stream()
                    .map(item -> {
                        NewCauseVo newCauseVo = new NewCauseVo();
                        newCauseVo.setName(item.getName());
                        newCauseVo.setId(item.getId());
                        return newCauseVo;
                    }).collect(Collectors.toList());
        }else {
            List<Cause> causes = causeDao.selectList(null);
            res = causes.stream()
                    .map(item -> {
                        NewCauseVo newCauseVo = new NewCauseVo();
                        newCauseVo.setName(item.getName());
                        newCauseVo.setId(item.getId());
                        return newCauseVo;
                    }).collect(Collectors.toList());
        }
        return res;
    }
    @Override
    public List<UserVo> getManagerList() {
        return userDao.selectList(new QueryWrapper<User>().eq("role", 1)).stream().map(a -> {
            UserVo userVo = new UserVo();