初始化代码

peng

2026-03-18 e59a0201057ba67cad425fed804c82ff4ba0c6f1

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
package com.tievd.jyz.cache;
 
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.tievd.jyz.constants.SystemConstant;
import com.tievd.jyz.entity.UpgradeRecord;
import com.tievd.jyz.mapper.UpgradeRecordMapper;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
 
import javax.annotation.PostConstruct;
import java.util.*;
import java.util.concurrent.ConcurrentHashMap;
 
/**
 * 资源缓存
 *
 * @author yang'zhi'shui
 */
@Component
public class ResourceCache {
 
    @Autowired
    private UpgradeRecordMapper upgradeRecordMapper;
 
    @PostConstruct
    public void init() {
        QueryWrapper<UpgradeRecord> queryWrapper = new QueryWrapper<>();
        queryWrapper.eq("status", SystemConstant.ISSUE_UNDERWAY);
        List<UpgradeRecord> list = upgradeRecordMapper.selectList(queryWrapper);
        for (UpgradeRecord upgradeRecord : list) {
            addUpgradeRecord(upgradeRecord);
        }
    }
 
    /**
     * 升级下发记录
     */
    private static final Map<String, UpgradeRecord> UPGRADE_RECORD_MAP = new ConcurrentHashMap<>();
 
    public static List<UpgradeRecord> getUpgradeRecord(String deviceSn) {
        List<UpgradeRecord> list = new ArrayList<>();
        for (Map.Entry<String, UpgradeRecord> entry:UPGRADE_RECORD_MAP.entrySet()) {
            String key = entry.getKey();
            if (key.startsWith(deviceSn+"%")){
                list.add(entry.getValue());
            }
        }
        return list;
    }
 
    public static Collection<UpgradeRecord> getUpgradeRecord() {
        return UPGRADE_RECORD_MAP.values();
    }
 
    public static void addUpgradeRecord(UpgradeRecord upgradeRecord) {
        UPGRADE_RECORD_MAP.put(upgradeRecord.getDeviceSn() + "%" + upgradeRecord.getUpgradeId(), upgradeRecord);
    }
 
    public static void removeUpgradeRecord(UpgradeRecord upgradeRecord) {
        UPGRADE_RECORD_MAP.remove(upgradeRecord.getDeviceSn() + "%" + upgradeRecord.getUpgradeId());
    }
}