盗墓笔记全集,豆豆小说阅读网

新聞中心

這里有您想知道的互聯(lián)網(wǎng)營銷解決方案

用這四招優(yōu)雅地實現(xiàn)SpringBoot異步線程間數(shù)據(jù)傳遞

Spring Boot 自定義線程池實現(xiàn)異步開發(fā)相信看過陳某的文章都了解，但是在實際開發(fā)中需要在父子線程之間傳遞一些數(shù)據(jù)，比如用戶信息，鏈路信息等等

網(wǎng)站建設(shè)哪家好，找創(chuàng)新互聯(lián)公司！專注于網(wǎng)頁設(shè)計、網(wǎng)站建設(shè)、微信開發(fā)、小程序制作、集團企業(yè)網(wǎng)站建設(shè)等服務(wù)項目。為回饋新老客戶創(chuàng)新互聯(lián)還提供了清徐免費建站歡迎大家使用！

比如用戶登錄信息使用ThreadLocal存放保證線程隔離，代碼如下：

/**
 * @author 公眾號：碼猿技術(shù)專欄
 * @description 用戶上下文信息
 */
public class OauthContext {
    private static  final  ThreadLocal loginValThreadLocal=new ThreadLocal<>();

    public static  LoginVal get(){
        return loginValThreadLocal.get();
    }
    public static void set(LoginVal loginVal){
        loginValThreadLocal.set(loginVal);
    }
    public static void clear(){
        loginValThreadLocal.remove();
    }
}

那么子線程想要獲取這個LoginVal如何做呢？

今天就來介紹幾種優(yōu)雅的方式實現(xiàn)Spring Boot 內(nèi)部的父子線程的數(shù)據(jù)傳遞。

1. 手動設(shè)置

每執(zhí)行一次異步線程都要分為兩步：

獲取父線程的LoginVal
將LoginVal設(shè)置到子線程，達到復(fù)用

代碼如下：

public void handlerAsync() {
        //1. 獲取父線程的loginVal
        LoginVal loginVal = OauthContext.get();
        log.info("父線程的值：{}",OauthContext.get());
        CompletableFuture.runAsync(()->{
            //2. 設(shè)置子線程的值，復(fù)用
           OauthContext.set(loginVal);
           log.info("子線程的值：{}",OauthContext.get());
        });
    }

雖然能夠?qū)崿F(xiàn)目的，但是每次開異步線程都需要手動設(shè)置，重復(fù)代碼太多，看了頭疼，你認為優(yōu)雅嗎？

2. 線程池設(shè)置TaskDecorator

TaskDecorator是什么？官方api的大致意思：這是一個執(zhí)行回調(diào)方法的裝飾器，主要應(yīng)用于傳遞上下文，或者提供任務(wù)的監(jiān)控/統(tǒng)計信息。

知道有這么一個東西，如何去使用？

TaskDecorator是一個接口，首先需要去實現(xiàn)它，代碼如下：

/**
 * @author 公眾號：碼猿技術(shù)專欄
 * @description 上下文裝飾器
 */
public class ContextTaskDecorator implements TaskDecorator {
    @Override
    public Runnable decorate(Runnable runnable) {
        //獲取父線程的loginVal
        LoginVal loginVal = OauthContext.get();
        return () -> {
            try {
                // 將主線程的請求信息，設(shè)置到子線程中
                OauthContext.set(loginVal);
                // 執(zhí)行子線程，這一步不要忘了
                runnable.run();
            } finally {
                // 線程結(jié)束，清空這些信息，否則可能造成內(nèi)存泄漏
                OauthContext.clear();
            }
        };
    }
}

這里我只是設(shè)置了LoginVal，實際開發(fā)中其他的共享數(shù)據(jù)，比如SecurityContext，RequestAttributes....

TaskDecorator需要結(jié)合線程池使用，實際開發(fā)中異步線程建議使用線程池，只需要在對應(yīng)的線程池配置一下，代碼如下：

@Bean("taskExecutor")
public ThreadPoolTaskExecutor taskExecutor() {
        ThreadPoolTaskExecutor poolTaskExecutor = new ThreadPoolTaskExecutor();
        poolTaskExecutor.setCorePoolSize(xx);
        poolTaskExecutor.setMaxPoolSize(xx);
        // 設(shè)置線程活躍時間（秒）
        poolTaskExecutor.setKeepAliveSeconds(xx);
        // 設(shè)置隊列容量
        poolTaskExecutor.setQueueCapacity(xx);
        //設(shè)置TaskDecorator，用于解決父子線程間的數(shù)據(jù)復(fù)用
        poolTaskExecutor.setTaskDecorator(new ContextTaskDecorator());
        poolTaskExecutor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
        // 等待所有任務(wù)結(jié)束后再關(guān)閉線程池
        poolTaskExecutor.setWaitForTasksToCompleteOnShutdown(true);
        return poolTaskExecutor;
    }

此時業(yè)務(wù)代碼就不需要去設(shè)置子線程的值，直接使用即可，代碼如下：

public void handlerAsync() {
        log.info("父線程的用戶信息：{}", OauthContext.get());
        //執(zhí)行異步任務(wù)，需要指定的線程池
        CompletableFuture.runAsync(()-> log.info("子線程的用戶信息：{}", OauthContext.get()),taskExecutor);
    }

來看一下結(jié)果，如下圖：

這里使用的是CompletableFuture執(zhí)行異步任務(wù)，使用@Async這個注解同樣是可行的。

注意：無論使用何種方式，都需要指定線程池

3. InheritableThreadLocal

這種方案不建議使用，InheritableThreadLocal雖然能夠?qū)崿F(xiàn)父子線程間的復(fù)用，但是在線程池中使用會存在復(fù)用的問題。

這種方案使用也是非常簡單，直接用InheritableThreadLocal替換ThreadLocal即可，代碼如下：

/**
 * @author 公眾號：碼猿技術(shù)專欄
 * @description 用戶上下文信息
 */
public class OauthContext {
    private static  final  InheritableThreadLocal loginValThreadLocal=new InheritableThreadLocal<>();

    public static  LoginVal get(){
        return loginValThreadLocal.get();
    }
    public static void set(LoginVal loginVal){
        loginValThreadLocal.set(loginVal);
    }
    public static void clear(){
        loginValThreadLocal.remove();
    }
}

4. TransmittableThreadLocal

TransmittableThreadLocal是阿里開源的工具，彌補了InheritableThreadLocal的缺陷，在使用線程池等會池化復(fù)用線程的執(zhí)行組件情況下，提供??ThreadLocal??值的傳遞功能，解決異步執(zhí)行時上下文傳遞的問題。

使用起來也是非常簡單，添加依賴如下：


 com.alibaba
 transmittable-thread-local
 2.14.2

OauthContext改造代碼如下：

/**
 * @author 公眾號：碼猿技術(shù)專欄
 * @description 用戶上下文信息
 */
public class OauthContext {
    private static  final TransmittableThreadLocal loginValThreadLocal=new TransmittableThreadLocal<>();

    public static  LoginVal get(){
        return loginValThreadLocal.get();
    }
    public static void set(LoginVal loginVal){
        loginValThreadLocal.set(loginVal);
    }
    public static void clear(){
        loginValThreadLocal.remove();
    }
}

TransmittableThreadLocal原理

從定義來看，TransimittableThreadLocal繼承于InheritableThreadLocal，并實現(xiàn)TtlCopier接口，它里面只有一個copy方法。所以主要是對InheritableThreadLocal的擴展。

public class TransmittableThreadLocal extends InheritableThreadLocal implements TtlCopier

在TransimittableThreadLocal中添加holder屬性。這個屬性的作用就是被標記為具備線程傳遞資格的對象都會被添加到這個對象中。

要標記一個類，比較容易想到的方式，就是給這個類新增一個Type字段，還有一個方法就是將具備這種類型的的對象都添加到一個靜態(tài)全局集合中。之后使用時，這個集合里的所有值都具備這個標記。

// 1. holder本身是一個InheritableThreadLocal對象
// 2. 這個holder對象的value是WeakHashMap, ?>
//   2.1 WeekHashMap的value總是null,且不可能被使用。
//    2.2 WeekHasshMap支持value=null
private static InheritableThreadLocal, ?>> holder = new InheritableThreadLocal, ?>>() {
  @Override
  protected WeakHashMap, ?> initialValue() {
    return new WeakHashMap, Object>();
  }
 
  /**
   * 重寫了childValue方法，實現(xiàn)上直接將父線程的屬性作為子線程的本地變量對象。
   */
  @Override
  protected WeakHashMap, ?> childValue(WeakHashMap, ?> parentValue) {
    return new WeakHashMap, Object>(parentValue);
  }
};

應(yīng)用代碼是通過TtlExecutors工具類對線程池對象進行包裝。工具類只是簡單的判斷，輸入的線程池是否已經(jīng)被包裝過、非空校驗等，然后返回包裝類ExecutorServiceTtlWrapper。根據(jù)不同的線程池類型，有不同和的包裝類。

@Nullable
public static ExecutorService getTtlExecutorService(@Nullable ExecutorService executorService) {
  if (TtlAgent.isTtlAgentLoaded() || executorService == null || executorService instanceof TtlEnhanced) {
    return executorService;
  }
  return new ExecutorServiceTtlWrapper(executorService);
}

進入包裝類ExecutorServiceTtlWrapper?？梢宰⒁獾讲徽撌峭ㄟ^ExecutorServiceTtlWrapper#submit方法或者是ExecutorTtlWrapper#execute方法，都會將線程對象包裝成TtlCallable或者TtlRunnable，用于在真正執(zhí)行run方法前做一些業(yè)務(wù)邏輯。

/**
 * 在ExecutorServiceTtlWrapper實現(xiàn)submit方法
 */
@NonNull
@Override
public  Future submit(@NonNull Callable task) {
  return executorService.submit(TtlCallable.get(task));
}

/**
 * 在ExecutorTtlWrapper實現(xiàn)execute方法
 */
@Override
public void execute(@NonNull Runnable command) {
  executor.execute(TtlRunnable.get(command));
}

所以，重點的核心邏輯應(yīng)該是在TtlCallable#call()或者TtlRunnable#run()中。以下以TtlCallable為例，TtlRunnable同理類似。在分析call()方法之前，先看一個類Transmitter。

public static class Transmitter {
  /**
    * 捕獲當前線程中的是所有TransimittableThreadLocal和注冊ThreadLocal的值。
    */
  @NonNull
  public static Object capture() {
    return new Snapshot(captureTtlValues(), captureThreadLocalValues());
  }
 
    /**
    * 捕獲TransimittableThreadLocal的值,將holder中的所有值都添加到HashMap后返回。
    */
  private static HashMap, Object> captureTtlValues() {
    HashMap, Object> ttl2Value = 
      new HashMap, Object>();
    for (TransmittableThreadLocal