eemcs/mh-common/src/main/java/com/mh/common/utils/NettyTools.java

package com.mh.common.utils;


import com.google.common.cache.Cache;
import com.google.common.cache.CacheBuilder;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;

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 NettyTools {

    /**
     * 响应消息缓存
     */
    private static final Cache<String, BlockingQueue<String>> responseMsgCache = CacheBuilder.newBuilder()
            .maximumSize(500)
            .expireAfterWrite(1000, TimeUnit.SECONDS)
            .build();


    /**
     * 等待响应消息
     * @param key 消息唯一标识
     * @return ReceiveDdcMsgVo
     */
    public static boolean waitReceiveMsg(String key) {

        try {
            //设置超时时间
            String vo = Objects.requireNonNull(responseMsgCache.getIfPresent(key))
                .poll(1000 * 10, TimeUnit.MILLISECONDS);

            //删除key
            responseMsgCache.invalidate(key);
            return StringUtils.isNotBlank(vo);
        } catch (Exception e) {
            log.error("获取数据异常,sn={},msg=null",key);
            return false;
        }

    }

    /**
     * 初始化响应消息的队列
     * @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.getIfPresent(key) != null){
            Objects.requireNonNull(responseMsgCache.getIfPresent(key)).add(msg);
            return;
        }

        log.warn("sn {}不存在",key);
    }

}