mh
9 months ago
7 changed files with 647 additions and 423 deletions
@ -0,0 +1,79 @@ |
|||||||
|
package com.mh.user.utils; |
||||||
|
|
||||||
|
|
||||||
|
import com.google.common.cache.Cache; |
||||||
|
import com.google.common.cache.CacheBuilder; |
||||||
|
import io.micrometer.core.instrument.util.StringUtils; |
||||||
|
import lombok.extern.slf4j.Slf4j; |
||||||
|
|
||||||
|
import java.util.Objects; |
||||||
|
import java.util.concurrent.BlockingQueue; |
||||||
|
import java.util.concurrent.LinkedBlockingQueue; |
||||||
|
import java.util.concurrent.TimeUnit; |
||||||
|
|
||||||
|
/** |
||||||
|
* @author LJF |
||||||
|
* @version 1.0 |
||||||
|
* @project TAD_Server |
||||||
|
* @description 缓存等待数据 |
||||||
|
* @date 2023/7/4 08:45:16 |
||||||
|
*/ |
||||||
|
@Slf4j |
||||||
|
public class CacheTools { |
||||||
|
|
||||||
|
/** |
||||||
|
* 响应消息缓存 |
||||||
|
*/ |
||||||
|
private static final Cache<String, BlockingQueue<String>> responseMsgCache = CacheBuilder.newBuilder() |
||||||
|
.maximumSize(500) |
||||||
|
.expireAfterWrite(1000, TimeUnit.SECONDS) |
||||||
|
.build(); |
||||||
|
|
||||||
|
|
||||||
|
/** |
||||||
|
* 等待响应消息 |
||||||
|
* |
||||||
|
* @param key 消息唯一标识 |
||||||
|
* @return ReceiveDdcMsgVo |
||||||
|
*/ |
||||||
|
public static String waitReceiveMsg(String key) { |
||||||
|
try { |
||||||
|
//设置超时时间
|
||||||
|
String vo = Objects.requireNonNull(responseMsgCache.getIfPresent(key)) |
||||||
|
.poll(1000 * 5, TimeUnit.MILLISECONDS); |
||||||
|
|
||||||
|
//删除key
|
||||||
|
responseMsgCache.invalidate(key); |
||||||
|
return vo; |
||||||
|
} catch (Exception e) { |
||||||
|
log.error("获取数据异常,sn={},msg=null", key); |
||||||
|
return ""; |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 初始化响应消息的队列 |
||||||
|
* |
||||||
|
* @param key 消息唯一标识 |
||||||
|
*/ |
||||||
|
public static void initReceiveMsg(String key) { |
||||||
|
responseMsgCache.put(key, new LinkedBlockingQueue<String>(1)); |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 设置响应消息 |
||||||
|
* |
||||||
|
* @param key 消息唯一标识 |
||||||
|
*/ |
||||||
|
public static void setReceiveMsg(String key, String msg) { |
||||||
|
if (!responseMsgCache.asMap().containsKey(key)) { |
||||||
|
initReceiveMsg(key); |
||||||
|
} |
||||||
|
if (responseMsgCache.getIfPresent(key) != null) { |
||||||
|
responseMsgCache.getIfPresent(key).add(msg); |
||||||
|
return; |
||||||
|
} |
||||||
|
log.warn("sn {}不存在", key); |
||||||
|
} |
||||||
|
} |
Loading…
Reference in new issue