wisdomisite-java/src/main/java/com/zhgd/xmgl/task/WorkerTask.java

package com.zhgd.xmgl.task;

import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.date.DateUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.zhgd.xmgl.modules.basicdata.entity.SystemUser;
import com.zhgd.xmgl.modules.basicdata.mapper.SystemUserMapper;
import com.zhgd.xmgl.modules.basicdata.service.INoticeService;
import com.zhgd.xmgl.modules.project.entity.ProjectExternalSystemService;
import com.zhgd.xmgl.modules.project.mapper.ProjectExternalSystemServiceMapper;
import com.zhgd.xmgl.modules.project.mapper.ProjectMapper;
import com.zhgd.xmgl.modules.project.service.IProjectUfaceConfigService;
import com.zhgd.xmgl.modules.worker.entity.WorkerBlacklist;
import com.zhgd.xmgl.modules.worker.entity.WorkerInfo;
import com.zhgd.xmgl.modules.worker.mapper.WorkerBlacklistMapper;
import com.zhgd.xmgl.modules.worker.mapper.WorkerCertificateMapper;
import com.zhgd.xmgl.modules.worker.mapper.WorkerInfoMapper;
import com.zhgd.xmgl.modules.worker.service.IWorkerCertificateService;
import com.zhgd.xmgl.util.ElecardUtil;
import lombok.extern.slf4j.Slf4j;
import net.javacrumbs.shedlock.core.SchedulerLock;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;

import java.util.Date;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

/**
 * @program: wisdomSite
 * @description: 劳务人员定时任务
 * @author: Mr.Peng
 * @create: 2021-09-10 17:31
 **/
@Slf4j
@Component
@RestController
public class WorkerTask {
    @Autowired
    private ProjectExternalSystemServiceMapper projectExternalSystemServiceMapper;
    @Autowired
    private WorkerInfoMapper workerInfoMapper;
    @Autowired
    private ProjectMapper projectMapper;
    @Autowired
    private SystemUserMapper systemUserMapper;

    @Autowired
    private INoticeService noticeService;
    @Autowired
    private IProjectUfaceConfigService projectUfaceConfigService;
    @Autowired
    private IWorkerCertificateService workerCertificateService;
    @Autowired
    private WorkerCertificateMapper workerCertificateMapper;
    @Autowired
    private WorkerBlacklistMapper workerBlacklistMapper;

    /**
     * 定时修改用户码状态
     */
    @SchedulerLock(name = "updateWorkerCode", lockAtMostFor = 1000 * 60 * 60, lockAtLeastFor = 1000 * 60 * 5)
    @Scheduled(cron = "0 0 1 * * ?")
    public void updateWorkerCode() {
        log.info("------------------------------健康码同步开始-----------------------------------------");
        try {
            List<ProjectExternalSystemService> list = projectExternalSystemServiceMapper.getChangtongCodeSystemList();
            if (list != null && list.size() > 0) {
                for (ProjectExternalSystemService projectExternalSystemService : list) {
                    getWorkerCodeState(projectExternalSystemService);
                }
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
        log.info("------------------------------健康码同步结束-----------------------------------------");
    }

    public void getWorkerCodeState(ProjectExternalSystemService projectExternalSystemService) {
        try {
            List<WorkerInfo> list = workerInfoMapper.selectAllWorkerInfoList(projectExternalSystemService.getProjectSn());
            if (list != null && list.size() > 0) {
                String token = ElecardUtil.getToken(projectExternalSystemService.getSystemUrl(), projectExternalSystemService.getAppId(), projectExternalSystemService.getAppSecert(),
                        projectExternalSystemService.getPublicKey(), projectExternalSystemService.getPrivateKey());
                if (StringUtils.isNotEmpty(token)) {
                    int redNum = 0;
                    int orangeNum = 0;
                    StringBuilder redWorkerName = new StringBuilder();
                    StringBuilder orangeWorkerName = new StringBuilder();
                    for (WorkerInfo workerInfo : list) {
                        log.info("---------------" + workerInfo.getWorkerName());
                        String enterStatus = ElecardUtil.getCardInfo(projectExternalSystemService.getSystemUrl(), projectExternalSystemService.getAppId(), projectExternalSystemService.getAppSecert(),
                                projectExternalSystemService.getPublicKey(), projectExternalSystemService.getPrivateKey(), token, workerInfo.getIdCard());
                        //0 :绿色, 1:黄色 ,2:红色
                        //红黄码需设备取消授权不允许通行
                        if (StringUtils.isNotEmpty(enterStatus)) {
                            if ("2".equals(enterStatus)) {
                                redNum++;
                                if (redWorkerName.length() > 0) {
                                    redWorkerName.append("、");
                                }
                                redWorkerName.append(workerInfo.getWorkerName());
                                //码状态，0无码，1红，2黄，3绿
                                workerInfo.setCodeState(1);
                                workerInfo.setUfaceDevId("0");
                                projectUfaceConfigService.deleteWorkerInfo(workerInfo);
                            } else if ("1".equals(enterStatus)) {
                                orangeNum++;
                                if (orangeWorkerName.length() > 0) {
                                    orangeWorkerName.append("、");
                                }
                                orangeWorkerName.append(workerInfo.getWorkerName());
                                workerInfo.setCodeState(2);
                                workerInfo.setUfaceDevId("0");
                                projectUfaceConfigService.deleteWorkerInfo(workerInfo);
                            } else {
                                workerInfo.setCodeState(3);
                            }
                            workerInfoMapper.updateById(workerInfo);

                        }
                    }
                    if (redNum > 0 || orangeNum > 0) {
                        //String title="今日昌通码核验黄码"+orangeNum+"人，红码"+redNum+"人";
                        StringBuilder title = new StringBuilder();
                        title.append("今日昌通码核验黄码").append(orangeNum).append("人，");
                        if (orangeWorkerName.length() > 0) {
                            title.append("分别为：").append(orangeWorkerName.toString()).append("，");
                        }
                        title.append("红码").append(redNum).append("人");
                        if (redWorkerName.length() > 0) {
                            title.append("分别为：").append(redWorkerName.toString()).append("。");
                        }
                        StringBuilder title2 = new StringBuilder();
                        if (orangeWorkerName.length() > 0) {
                            title2.append("黄码人员：").append(orangeWorkerName.toString()).append("，");
                        }
                        if (redWorkerName.length() > 0) {
                            title2.append("红码人员：").append(redWorkerName.toString()).append("。");
                        }
                        title2.append("已销权");
                        noticeService.sendProjectNoicte(projectExternalSystemService.getProjectSn(), "红黄码核验", title.toString(), "17");
                        noticeService.sendProjectNoicte(projectExternalSystemService.getProjectSn(), "红黄码销权", title2.toString(), "17");
                    }
                }
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    /**
     * 劳务人员，截止日期到期的话，就自动填充原因，就是证件到期到黑名单中
     */
    @SchedulerLock(name = "addWorkerBlacklist", lockAtMostFor = 1000 * 60 * 60, lockAtLeastFor = 1000 * 60 * 5)
    @Scheduled(cron = "0 0 1 * * ?")
    @GetMapping("xmgl/task/addWorkerBlacklist")
    public void addWorkerBlacklist() {
        List<WorkerInfo> list = workerCertificateMapper.selectExpiredNotInBlackWorkerList();
        for (WorkerInfo workerInfo : list) {
            WorkerBlacklist black = new WorkerBlacklist();
            black.setProjectSn(workerInfo.getProjectSn());
            black.setWorkerId(workerInfo.getId());
            black.setWorkerName(workerInfo.getWorkerName());
            black.setIdCard(workerInfo.getIdCard());
            black.setAddReason("证件已到期");
            black.setReason("证件已到期");
            black.setAddTime(DateUtil.now());
            black.setCreateTime(new Date());
            black.setType(1);
            workerBlacklistMapper.insert(black);
        }
    }

    /**
     * 系统整改提前一个星期对人员的资质证书到期进行预警，预警后发送消息提示项目管理员账号
     */
    @SchedulerLock(name = "alarmWorkerBlack", lockAtMostFor = 1000 * 60 * 60, lockAtLeastFor = 1000 * 60 * 5)
    @Scheduled(cron = "0 0 1 * * ?")
    @GetMapping("xmgl/task/alarmWorkerBlack")
    public void alarmWorkerBlack() {
        List<WorkerInfo> list = workerCertificateMapper.alarmWorkerBlack();
        Map<String, List<WorkerInfo>> projectMap = list.stream().collect(Collectors.groupingBy(WorkerInfo::getProjectSn));
        if (CollUtil.isNotEmpty(projectMap)) {
            for (Map.Entry<String, List<WorkerInfo>> entry : projectMap.entrySet()) {
                List<SystemUser> systemUserList = systemUserMapper.selectList(new LambdaQueryWrapper<SystemUser>()
                        .eq(SystemUser::getSn, entry.getKey()));
                List<WorkerInfo> workerInfoList = entry.getValue();
                if (CollUtil.isNotEmpty(workerInfoList)) {
                    for (WorkerInfo workerInfo : workerInfoList) {
                        if (CollUtil.isNotEmpty(systemUserList)) {
                            for (SystemUser systemUser : systemUserList) {
                                noticeService.addUserNotice(systemUser.getUserId(), workerInfo.getWorkerName() + "的" + workerInfo.getCertificateTypeName() + "即将到期，请尽快更新！", "人员的资质证书即将到期", "20");
                            }
                        }
                    }
                }
            }
        }
    }

}