java利用CompletableFuture實現多任務並行.大大節省調用時長

 
 1 直接上代碼吧。java版本的。
背景是我有一個需求是用戶注銷。注銷時我需要判斷是否有訂單，結算單等業務。這樣我就需要調用其他業務線的接口。
接口需要調用N個（這里是5個）分別判斷是否允許注銷。
悲觀假設每個接口調用要1s那么5個接口就需要5s以上，並且任何一個不可行則整個判斷將不通過。
這時候就需要用到異步任務並行。 這里采用

CompletableFuture的thenCombine 來實現多個異步任務並行。代碼簡單。將兩個test跑一下就可以看到效果，
異常case是將userId寫成 >100的值。第4個任務將會拋錯。這樣就可以看到整體運行結果。
當然拋錯也可以轉換為自己認識的錯誤方式，由調用方來決定是否繼續往下走。
本文為原創，雖然例子很簡單但是可以解決我的需求。分享出來希望能幫到您，
當然如果您有更好的方式歡迎留言共同探討，我的郵箱 [e-mail] 有更好的歡迎共同探討。

import java.util.concurrent.CompletableFuture;
import lombok.extern.slf4j.Slf4j;
import org.junit.Test;

@Slf4j
public class SupplyAsyncText {

  @Test
  public void testSyncLogOff() {
    int userId = 45;
    // 同步
    long begin = System.currentTimeMillis();
    StringBuilder sb = new StringBuilder();
    sb.append(canDo1(userId));
    sb.append(canDo2(userId));
    sb.append(canDo3(userId));
    sb.append(canDo4(userId));
    sb.append(canDo5(userId));
    long end = System.currentTimeMillis();
    // 每一個執行1s 同步：結果是執行應該是 > 5s 的
    log.info("noUseAsync |{}|{} s", sb.toString(), (end - begin) / 1000.0);
  }

  @Test
  public void testCanLogOff() {
    // case 設定：如果 userId > 100 則在某一個環節拋錯
    int userId = 10;
    // 異步 每一個執行1s 異步並行：結果是執行應該 < 5s
    long begin2 = System.currentTimeMillis();
    String result =
        CompletableFuture.supplyAsync(() -> canDo1(userId))
            .thenCombine(CompletableFuture.supplyAsync(() -> canDo2(userId)), (s1, s2) -> s1 + s2)
            .thenCombine(CompletableFuture.supplyAsync(() -> canDo3(userId)), (s1, s2) -> s1 + s2)
            .thenCombine(CompletableFuture.supplyAsync(() -> canDo4(userId)), (s1, s2) -> s1 + s2)
            .thenCombine(CompletableFuture.supplyAsync(() -> canDo5(userId)), (s1, s2) -> s1 + s2)
            .exceptionally(
                e -> {
                  log.error("exception: ", e);
                  return e.getMessage();
                })
            .join();
    long end2 = System.currentTimeMillis();
    log.info("supplyAsync|{}|{} s", result, (end2 - begin2) / 1000.0);
  }

  private String canDo1(int userId) {
    try {
      Thread.sleep(1000);
    } catch (InterruptedException e) {
      e.printStackTrace();
    }
    return "ok";
  }

  private String canDo2(int userId) {
    try {
      Thread.sleep(1000);
    } catch (InterruptedException e) {
      e.printStackTrace();
    }
    return "ok";
  }

  private String canDo3(int userId) {
    try {
      Thread.sleep(1000);
    } catch (InterruptedException e) {
      e.printStackTrace();
    }
    return "ok";
  }

  private String canDo4(int userId) {
    try {
      Thread.sleep(1000);
    } catch (InterruptedException e) {
      e.printStackTrace();
    }
    if (userId > 100) {
      throw new RuntimeException(String.format("用戶[%s]還有訂單，不可注銷",userId));
    }
    return "ok";
  }

  private String canDo5(int userId) {
    try {
      Thread.sleep(1000);
    } catch (InterruptedException e) {
      e.printStackTrace();
    }
    return "ok";
  }
}