关于处理异步后台调用接口问题及异步调用远程接口无法获取token问题处理

忘是亡心i 2023-10-10 17:32 10阅读 0赞

**需求:一个界面的报告状态需要随时更改，，，可以在修改后异步后台执行状态查询更新操作**

具体方法为

try {
                //异步执行数据采集
                CompletableFuture<Void> future = CompletableFuture.runAsync(() -> {
                    logger.info("开始异步报告id为{}！！！",reportId);
                    reportByChangeStatus(reportId, now, userid);
                    logger.info("报告id为{}异步更新状态成功！！！",reportId);
                }).exceptionally((e) -> {
                    //异常处理
                    e.printStackTrace();
                    return null;
                });
            } catch (Exception e) {
                e.printStackTrace();
            }

中间的

reportByChangeStatus(reportId, now, userid);

便是需要异步执行的具体方法

这是简单的不涉及异步方法在调用远程接口的实现

**若异步方法中有需要调用远程接口的方法需要将token传过去。**

具体方法为：

try {
                LoginUser loginUser = manage.getLoginUser();
                //保存token，供feignClient拦截器使用
                RequestTokenHolder.saveToken(loginUser.getToken());
                //异步执行数据采集
                CompletableFuture<Void> future = CompletableFuture.runAsync(() -> {
                    logger.info("开始异步报告id为{}！！！",reportId);
                    reportByChangeStatus(reportId, now, userid);
                    logger.info("报告id为{}异步更新状态成功！！！",reportId);
                }).exceptionally((e) -> {
                    //异常处理
                    e.printStackTrace();
                    return null;
                });
            } catch (Exception e) {
                e.printStackTrace();
            } finally {
                RequestTokenHolder.removeToken();
            }

其中用到了一个异步线程共享token的工具类，也就是在调用异步方法之前，将获取到的token先塞进当前的线程

import java.util.concurrent.ConcurrentHashMap;
    
    /**
     * 线程共享token工具类
     */
    public class RequestTokenHolder {
    
        private static final ConcurrentHashMap<Thread,ThreadLocal<String>> threadLocalMap = new ConcurrentHashMap<>();
    
        public static void saveToken(String token){
            ThreadLocal<String> threadLocal = new ThreadLocal<>();
            threadLocal.set(token);
            threadLocalMap.put(Thread.currentThread(),threadLocal);
        }
    
        public static String getToken(){
            Thread thread = Thread.currentThread();
            ThreadLocal<String> threadLocal = threadLocalMap.get(thread);
            return threadLocal!=null?threadLocal.get():null;
        }
    
        public static void removeToken(){
            Thread thread = Thread.currentThread();
            ThreadLocal<String> threadLocal = threadLocalMap.get(thread);
            if (threadLocal == null){
                return;
            }
            threadLocal.remove();
            threadLocalMap.remove(thread);
        }
    
    }

同时需要添加一个接口的token拦截器，获取token

import feign.RequestInterceptor;
    import feign.RequestTemplate;
    import org.slf4j.Logger;
    import org.slf4j.LoggerFactory;
    import org.springframework.stereotype.Component;
    
    /**
     * 异步线程透传请求 token 拦截器
     */
    @Component
    public class AsyncTokenConfig implements RequestInterceptor {
    
        private final Logger logger = LoggerFactory.getLogger(getClass());
    
        @Override
        public void apply(RequestTemplate requestTemplate) {
            String token = RequestTokenHolder.getToken();
            if (StringUtils.isNotBlank(token) && !requestTemplate.headers().containsKey(CacheConstants.AUTHORIZATION_HEADER)){
                logger.debug("透传token:{}", token);
                requestTemplate.header(CacheConstants.AUTHORIZATION_HEADER, token);
            }
        }
    }

这样就算完成了