TStaningBookServiceImpl.java

package com.zehong.system.service.impl;

import cn.hutool.core.date.DateTime;
import com.zehong.common.core.domain.entity.SysUser;
import com.zehong.common.core.domain.model.LoginUser;
import com.zehong.common.utils.DateUtils;
import com.zehong.common.utils.SecurityUtils;
import com.zehong.system.domain.TNews;
import com.zehong.system.domain.TStaningBook;
import com.zehong.system.domain.vo.HiddenStatVO;
import com.zehong.system.mapper.SysDeptMapper;
import com.zehong.system.mapper.THiddenTroubleWorkMapper;
import com.zehong.system.mapper.TStaningBookMapper;
import com.zehong.system.mapper.TTrainCourseMapper;
import com.zehong.system.service.ITStaningBookService;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.text.SimpleDateFormat;
import java.util.*;

/**
 * 隐患台账Service业务层处理
 *
 * @author zehong
 * @date 2022-07-09
 */
@Service
public class TStaningBookServiceImpl implements ITStaningBookService
{
    private static final Logger logger = LoggerFactory.getLogger(TStaningBookServiceImpl.class);
    @Autowired
    private TStaningBookMapper tStaningBookMapper;
    @Autowired
    private THiddenTroubleWorkMapper tHiddenTroubleWorkMapper;
    @Autowired
    private SysDeptMapper deptMapper;
    @Autowired
    private TTrainCourseMapper tTrainCourseMapper;
    /**
     * 查询隐患台账
     *
     * @param bookId 隐患台账ID
     * @return 隐患台账
     */
    @Override
    public TStaningBook selectTStaningBookById(Long bookId)
    {
        return tStaningBookMapper.selectTStaningBookById(bookId);
    }

    /**
     * 查询隐患台账列表
     *
     * @param tStaningBook 隐患台账
     * @return 隐患台账
     */
    @Override
    public List<TStaningBook> selectTStaningBookList(TStaningBook tStaningBook,Long roleId)
    {
        if(roleId!=3){//当角色是管理员时候查询全部
            tStaningBook.setUserId(0l);
        }
        List<TStaningBook> list = tStaningBookMapper.selectTStaningBookList(tStaningBook);
        for (TStaningBook t : list){
            if(t.getRectificationTerm()!=null&&t.getRectificationTerm().getTime()<new Date().getTime()&&t.getState()==0){
                t.setFinishDay(1l);
            }else {
                t.setFinishDay(0l);
            }
        }
        return  list;
    }
    public List<Long> selectDeptIds(List<Long> list,List<Long> deptIds){
        List<Long> resultlist = deptMapper.selectDeptByParentId(deptIds);
        if(resultlist.size()==0){
            return list;
        }
        list.addAll(resultlist);
        return selectDeptIds(list,resultlist);
    }
    /**
     * 新增隐患台账
     *
     * @param tStaningBook 隐患台账
     * @return 结果
     */
    @Override
    public int insertTStaningBook(TStaningBook tStaningBook)
    {
        tStaningBook.setCreateTime(DateUtils.getNowDate());
            /** 手机端消息新封装类 */
            TNews tNews=new TNews();
            /**设置所属模块*/
            tNews.setModule(2);
            //查询接收人姓名
//            String nickName = tTrainCourseMapper.selectRecipient(tStaningBook.getRectification());
        SimpleDateFormat simpleDateFormat = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
        String myDate = simpleDateFormat.format(tStaningBook.getRectificationTerm());
            /**设置内容*/
            tNews.setMessageContent("隐患排查："+tStaningBook.getTroubleName()+"，请于"+myDate+"前整改完毕。");
            LoginUser loginUser = SecurityUtils.getLoginUser();
            SysUser user = loginUser.getUser();
            Long userId = user.getUserId();
            /** 发布人ID*/
            tNews.setPublisherId(userId);
            /** 接收人id*/
            tNews.setRecipientId(Math.toIntExact(tStaningBook.getRectification()));
            int i = tStaningBookMapper.insertTStaningBook(tStaningBook);
             /** 所属模块id*/
            tNews.setModuleId(Long.valueOf(tStaningBook.getBookId()));
            tNews.setReleaseTime(new DateTime());
            /**
             * 消息添加
             */
            tTrainCourseMapper.insertNews(tNews);


        return i;
    }

    /**
     * 修改隐患台账
     *
     * @param tStaningBook 隐患台账
     * @return 结果
     */
    @Override
    public int updateTStaningBook(TStaningBook tStaningBook)
    {
        return tStaningBookMapper.updateTStaningBook(tStaningBook);
    }

    /**
     * 批量删除隐患台账
     *
     * @param bookIds 需要删除的隐患台账ID
     * @return 结果
     */
    @Override
    public int deleteTStaningBookByIds(Long[] bookIds)
    {
        return tStaningBookMapper.deleteTStaningBookByIds(bookIds);
    }

    /**
     * 删除隐患台账信息
     *
     * @param bookId 隐患台账ID
     * @return 结果
     */
    @Override
    public int deleteTStaningBookById(Long bookId)
    {
        return tStaningBookMapper.deleteTStaningBookById(bookId);
    }

    @Override
    public Map statics(String type) {
        return tStaningBookMapper.statics(type);
    }

    @Override
    public Map<String,List> hiddenStat(Integer troubleLevel, Date beginTime, Date endTime){
        Map<String,List> result = new HashMap<>(16);
        try {
            List<HiddenStatVO> barData = tStaningBookMapper.hiddenBarStat(troubleLevel,beginTime,endTime);
            //柱状图月份数据
            List<String> months = getMonthBetweenDate(beginTime,endTime);
            //柱状图隐患数量
            List<Integer> hiddens = new ArrayList<>();
            //柱状图整改数量
            List<Integer> rectifieds = new ArrayList<>();
            for(String month : months){
                boolean isHas = false;
                for(HiddenStatVO hidden : barData){
                    if(month.equals(hidden.getMonth())){
                        isHas = true;
                        hiddens.add(hidden.getHiddenCount());
                        rectifieds.add(hidden.getRectifiedCount());
                        break;
                    }
                }
                if(!isHas){
                    hiddens.add(0);
                    rectifieds.add(0);
                }
            }
            result.put("months",months);
            result.put("hiddens",hiddens);
            result.put("rectifieds",rectifieds);
            //环状统计图
            Map<String,Integer> pieData = tStaningBookMapper.hiddenPieStat(troubleLevel,beginTime,endTime);
            List<Map<String,Object>> pieResult = new ArrayList<>();
            Map<String,Object> rectifiedMap = new HashMap<>(16);
            rectifiedMap.put("name","已整改");
            rectifiedMap.put("value",pieData.get("rectifiedCount"));
            pieResult.add(rectifiedMap);
            Map<String,Object> stayRectifiedMap = new HashMap<>(16);
            stayRectifiedMap.put("name","未整改");
            stayRectifiedMap.put("value",pieData.get("stayRectifiedCount"));
            pieResult.add(stayRectifiedMap);
            result.put("pieData",pieResult);
        } catch (Exception e) {
            logger.error("隐患统计错误：" + e);
        }

        return result;
    }


    /**
     * 获取两个日期之间的所有月份 (年月)
     *
     * @param startTime
     * @param endTime
     * @return：YYYY-MM
     */
    private List<String> getMonthBetweenDate(Date startTime, Date endTime){
        SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM");
        // 声明保存日期集合
        List<String> list = new ArrayList<String>();
        //用Calendar 进行日期比较判断
        Calendar calendar = Calendar.getInstance();
        while (startTime.getTime()<=endTime.getTime()){
            // 把日期添加到集合
            list.add(sdf.format(startTime));
            // 设置日期
            calendar.setTime(startTime);
            //把日期增加一天
            calendar.add(Calendar.MONTH, 1);
            // 获取增加后的日期
            startTime=calendar.getTime();
        }
        return list;
    }

}