92 lines
3.2 KiB
Java
92 lines
3.2 KiB
Java
|
package com.ruoyi.cache;
|
|||
|
|
|
|||
|
|
import cn.hutool.core.util.StrUtil;
|
|||
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
|||
|
|
import com.github.benmanes.caffeine.cache.Cache;
|
|||
|
|
import com.github.benmanes.caffeine.cache.Caffeine;
|
|||
|
|
import com.ruoyi.database.domain.SysDictDataMp;
|
|||
|
|
import com.ruoyi.database.service.SysDictDataMpService;
|
|||
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|||
|
|
import org.springframework.scheduling.annotation.Scheduled;
|
|||
|
|
import org.springframework.stereotype.Component;
|
|||
|
|
|
|||
|
|
import java.util.List;
|
|||
|
|
import java.util.Map;
|
|||
|
|
import java.util.stream.Collectors;
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 字典缓存
|
|||
|
|
*
|
|||
|
|
* @author 戴鹏涛
|
|||
|
|
* @since 2023-09-13 15:40:44
|
|||
|
|
*/
|
|||
|
|
@Component
|
|||
|
|
public class DictCache {
|
|||
|
|
/**
|
|||
|
|
* key: dictType-dictValue，value: dictLabel; 例：place_category-1，居民小区
|
|||
|
|
* key: dictType-dictLabel，value: dictValue; 例：place_category-居民小区，1
|
|||
|
|
*/
|
|||
|
|
private static final Cache<String, String> cache = Caffeine.newBuilder().build();
|
|||
|
|
|
|||
|
|
private static SysDictDataMpService sysDictDataMpService;
|
|||
|
|
|
|||
|
|
@Autowired
|
|||
|
|
DictCache(SysDictDataMpService sysDictDataMpService) {
|
|||
|
|
DictCache.sysDictDataMpService = sysDictDataMpService;
|
|||
|
|
//初始化
|
|||
|
|
DictLabelLoad();
|
|||
|
|
DictValueLoad();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public static String getDictLabel(String key) {
|
|||
|
|
return cache.get(key, s -> {
|
|||
|
|
try {
|
|||
|
|
List<String> split = StrUtil.split(key, "-");
|
|||
|
|
String dictType = split.get(0);
|
|||
|
|
String dictValue = split.get(1);
|
|||
|
|
return sysDictDataMpService.getOne(new LambdaQueryWrapper<SysDictDataMp>()
|
|||
|
|
.eq(SysDictDataMp::getDictType, dictType)
|
|||
|
|
.eq(SysDictDataMp::getDictValue, dictValue)).getDictLabel();
|
|||
|
|
} catch (Exception e) {
|
|||
|
|
return "";
|
|||
|
|
}
|
|||
|
|
});
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public static String getDictValue(String key) {
|
|||
|
|
return cache.get(key, s -> {
|
|||
|
|
try {
|
|||
|
|
List<String> split = StrUtil.split(key, "-");
|
|||
|
|
String dictType = split.get(0);
|
|||
|
|
String dictLabel = split.get(1);
|
|||
|
|
return sysDictDataMpService.getOne(new LambdaQueryWrapper<SysDictDataMp>()
|
|||
|
|
.eq(SysDictDataMp::getDictType, dictType)
|
|||
|
|
.eq(SysDictDataMp::getDictLabel, dictLabel)).getDictValue();
|
|||
|
|
} catch (Exception e) {
|
|||
|
|
return "";
|
|||
|
|
}
|
|||
|
|
});
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 缓存刷新
|
|||
|
|
*/
|
|||
|
|
@Scheduled(cron = "0 0 3 * * ?")
|
|||
|
|
public static void refresh() {
|
|||
|
|
DictLabelLoad();
|
|||
|
|
DictValueLoad();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private static void DictLabelLoad() {
|
|||
|
|
List<SysDictDataMp> list = sysDictDataMpService.list();
|
|||
|
|
Map<String, String> dictMap = list.stream().collect(Collectors.toMap(e -> StrUtil.format("{}-{}", e.getDictType(), e.getDictValue()), SysDictDataMp::getDictLabel, (v1, v2) -> v2));
|
|||
|
|
cache.putAll(dictMap);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private static void DictValueLoad() {
|
|||
|
|
List<SysDictDataMp> list = sysDictDataMpService.list();
|
|||
|
|
Map<String, String> dictMap = list.stream().collect(Collectors.toMap(e -> StrUtil.format("{}-{}", e.getDictType(), e.getDictLabel()), SysDictDataMp::getDictValue, (v1, v2) -> v2));
|
|||
|
|
cache.putAll(dictMap);
|
|||
|
|
}
|
|||
|
|
}
|