提交

zhangback

8 天以前 b054362aaf616bfe0be0b50ae5dc2137091dbd7d

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
package com.ruoyi.tms.task;
 
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.ruoyi.tms.domain.TmsContract;
import com.ruoyi.tms.domain.TmsMessageNotify;
import com.ruoyi.tms.service.ITmsContractService;
import com.ruoyi.tms.service.ITmsMessageNotifyService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
 
import java.time.LocalDate;
import java.util.List;
import java.util.stream.Collectors;
@Slf4j
@Component("contractExpireNotifyTask")
public class ContractExpireNotifyTask {
 
    @Autowired
    ITmsContractService tmsContractService;
 
    @Autowired
    ITmsMessageNotifyService tmsMessageNotifyService;
 
    /**
     * 检查合同即将到期
     */
    public void checkContractExpire() {
        LocalDate today = LocalDate.now();
        LocalDate targetDate = today.plusDays(90);
 
        List<TmsContract> tmsContracts = tmsContractService.getBaseMapper().selectList(new LambdaQueryWrapper<TmsContract>()
                        .eq(TmsContract::getContractStatus, 0)
                .between(TmsContract::getContractEndDate, today, targetDate)
        );
        if (tmsContracts.isEmpty()){
            return;
        }
        log.info("即将到期的合同有{}个", tmsContracts.size());
        // 提取合同ID
        List<Integer> ids = tmsContracts.stream()
                .map(TmsContract::getId)
                .collect(Collectors.toList());
 
        // 批量更新为“临期”
        tmsContractService.getBaseMapper().update(
                new LambdaUpdateWrapper<TmsContract>()
                        .in(TmsContract::getId, ids)
                        .set(TmsContract::getContractStatus, 1)  // 1 = 临期
        );
 
        List<TmsMessageNotify> messageNotifies = tmsContracts.stream().map(item -> {
            TmsMessageNotify notify = new TmsMessageNotify();
            notify.setTitle("合同到期提醒");
            notify.setContent("【合同编号" + item.getContractCode() + "】即将到期");
            notify.setType(0);
            notify.setStatus(0);
            notify.setReadStatus(0);
            notify.setExtraData("{\"contactStatus\":1,\"contractId\":\"" + item.getId() + "\"}");
            return notify;
        }).collect(Collectors.toList());
        tmsMessageNotifyService.insertTmsMessageNotifyBatch(messageNotifies);
    }
    /**
     * 检查合同已到期
     */
    public void handleExpiredContracts() {
        LocalDate today = LocalDate.now();
 
        List<TmsContract> tmsContracts = tmsContractService.getBaseMapper().selectList(new LambdaQueryWrapper<TmsContract>()
                .ne(TmsContract::getContractStatus, 2)
                .le(TmsContract::getContractEndDate, today)
        );
        if (tmsContracts.isEmpty()){
            return;
        }
        log.info("已到期的合同有{}个", tmsContracts.size());
        // 提取合同ID
        List<Integer> ids = tmsContracts.stream()
                .map(TmsContract::getId)
                .collect(Collectors.toList());
 
        // 批量更新为“临期”
        tmsContractService.getBaseMapper().update(
                new LambdaUpdateWrapper<TmsContract>()
                        .in(TmsContract::getId, ids)
                        .set(TmsContract::getContractStatus, 2)  // 1 = 临期
        );
 
        List<TmsMessageNotify> messageNotifies = tmsContracts.stream().map(item -> {
            TmsMessageNotify notify = new TmsMessageNotify();
            notify.setTitle("合同到期提醒");
            notify.setContent("【合同编号" + item.getContractCode() + "】已到期");
            notify.setType(0);
            notify.setStatus(0);
            notify.setReadStatus(0);
            notify.setExtraData("{\"contactStatus\":2, \"contractId\":\"" + item.getId() + "\"}");
            return notify;
        }).collect(Collectors.toList());
        tmsMessageNotifyService.insertTmsMessageNotifyBatch(messageNotifies);
    }
 
 
}