package com.whyc.service;
|
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
import com.github.pagehelper.PageHelper;
|
import com.github.pagehelper.PageInfo;
|
import com.whyc.dto.Response;
|
import com.whyc.mapper.FaultUploadMapper;
|
import com.whyc.pojo.FaultUpload;
|
import com.whyc.util.ActionUtil;
|
import org.springframework.stereotype.Service;
|
|
import javax.annotation.Resource;
|
import java.util.*;
|
import java.util.stream.Collectors;
|
|
@Service
|
public class FaultUploadService {
|
|
@Resource
|
private FaultUploadMapper mapper;
|
|
public Response add(FaultUpload fault) {
|
int userId = ActionUtil.getUser().getUId().intValue();
|
Date now = new Date();
|
fault.setUploadUserId(userId);
|
fault.setCreateTime(now);
|
mapper.insert(fault);
|
return new Response().setII(1,"上传完成");
|
}
|
|
public Response updateConfirm(FaultUpload fault) {
|
int userId = ActionUtil.getUser().getUId().intValue();
|
Date now = new Date();
|
fault.setConfirmUserId(userId);
|
fault.setConfirmTime(now);
|
mapper.updateById(fault);
|
return new Response().setII(1,"确认完成");
|
}
|
|
public Response listPage(int pageNum, int pageSize) {
|
PageHelper.startPage(pageNum,pageSize);
|
int userId = ActionUtil.getUser().getUId().intValue();
|
/*QueryWrapper<FaultUpload> query = Wrappers.query();
|
query.eq("upload_user_id",userId);
|
List<FaultUpload> faultUploads = mapper.selectList(query);*/
|
List<FaultUpload> faultUploads = mapper.getList(userId);
|
PageInfo<FaultUpload> pageInfo = new PageInfo<>(faultUploads);
|
return new Response().set(1,pageInfo);
|
}
|
|
public Response getTypeNum() {
|
int userId = ActionUtil.getUser().getUId().intValue();
|
List<FaultUpload> faultUploads = mapper.getList(userId);
|
Map<Integer, List<FaultUpload>> typeListMap = faultUploads.stream().collect(Collectors.groupingBy(FaultUpload::getType));
|
Map<Integer,Integer> typeMap = new HashMap<>();
|
Set<Integer> typeSet = typeListMap.keySet();
|
for (Integer type : typeSet) {
|
typeMap.put(type,typeListMap.get(type).size());
|
}
|
return new Response().set(1,typeMap);
|
|
}
|
}
|