Bootstrap

JAVA语言异步非阻塞设计模式(应用篇)

1.概述

本系列文章共2篇。在上一篇《原理篇》中,我们看到了异步非阻塞模型,它能够有效降低线程IO状态的耗时,提升资源利用率和系统吞吐量。异步API可以表现为listener或Promise形式;其中Promise API提供了更强的灵活性,支持同步返回和异步回调,也允许注册任意数目的回调。

在本文《应用篇》中,我们将进一步探索异步模式和Promise的应用:

第2章:Promise与线程池。 在异步执行耗时请求时,ExecutorService+Future是一个备选方案;但是相比于Future,Promise支持纯异步获取响应数据,能够消除更多阻塞。

第3章:异常处理。 Java程序并不总能成功执行请求,有时会遇到网络问题等不可抗力。对于无法避免的异常情况,异步API必须提供异常处理机制,以提升程序的容错性。

第4章:请求调度。 Java程序有时需要提交多条请求,这些请求之间可能存在一定的关联关系,包括顺序执行、并行执行、批量执行。异步API需要对这些约束提供支持。

本文不限定Promise的具体实现,读者在生产环境可以选择一个Promise工具类(如netty DefaultPromise[A]、jdk CompletableFuture[B]等);此外,由于Promise的原理并不复杂,读者也可以自行实现所需功能。

2.Promise与线程池

Java程序有时需要执行耗时的IO操作,如数据库访问;在此期间,相比于纯内存计算,IO操作的持续时间明显更长。为了减少IO阻塞、提高资源利用率,我们应该使用异步模型,将请求提交到其他线程中执行,从而连续提交多条请求,而不必等待之前的请求返回。

本章对几种IO模型进行对比(见2.1节),考察调用者线程的阻塞情况。其中,Promise支持纯异步的请求提交及响应数据处理,能够最大程度地消除不必要的阻塞。在实际项目中,如果底层API不支持纯异步,那么我们也可以进行适当重构,使其和Promise兼容(见2.2节)。

2.1 对比:同步、Future、Promise

本节对几种 IO 模型进行对比,包括同步 IO、基于线程池(ExecutorService)的异步 IO、基于 Promise 的异步IO,考察调用者线程的阻塞情况。假设我们要执行数据库访问请求。由于需要跨越网络,单条请求需要进行耗时的 IO操作,才能最终收到响应数据;但是请求之间没有约束,允许随时提交新的请求,而不需要收到之前的响应数据。

首先我们来看看几种模型的样例代码

1.同步IO。db.writeSync()方法是同步阻塞的。函数阻塞,直至收到响应数据。因此,调用者一次只能提交一个请求,必须等待该请求返回,才能再提交下一个请求。

/* 提交请求并阻塞,直至收到响应数据*/
String result = db.writeSync("data");
process(result);

2.基于线程池(ExecutorService)的异步IO。db.writeSync()方法不变;但是将其提交到线程池中来执行,使得调用者线程不会阻塞,从而可以连续提交多条请求data1-3。

提交请求后,线程池返回 Future 对象,调用者调用 Future.get() 以获取响应数据。Future.get() 方法却是阻塞的,因此调用者在获得响应数据之前无法再提交后续请求。

/* 提交请求*/
// executor: ExecutorService
Future resultFuture1 = executor.submit(() -> db.writeSync("data1"));
Future resultFuture2 = executor.submit(() -> db.writeSync("data2"));
Future resultFuture3 = executor.submit(() -> db.writeSync("data3"));

/* 获取响应:同步*/
String result1 = resultFuture1.get();
String result2 = resultFuture2.get();
String result3 = resultFuture3.get();
process(result1);
process(result2);
process(result3);

3.基于Promise的异步IO。db.writeAsync()方法是纯异步的,提交请求后返回 Promise 对象;调用者调用Promise.await()注册回调,当收到响应数据后触发回调。

在《原理篇》中,我们看到了 Promise API 可以基于线程池或响应式模型实现;不论哪种方式,回调函数可以在接收响应的线程中执行,而不需要调用者线程阻塞地等待响应数据。

/* 提交请求*/
Promise resultPromise1 = db.writeAsync("data1");
Promise resultPromise2 = db.writeAsync("data2");
Promise resultPromise3 = db.writeAsync("data3");

/* 获取响应:异步*/
resultPromise1.await(result1 -> process(result1));
resultPromise2.await(result2 -> process(result2));
resultPromise3.await(result3 -> process(result3));

接下来我们看看以上几种模型中,调用者线程状态随时间变化的过程,如图2-1所示。

a.同步 IO。调用者一次只能提交一个请求,在收到响应之前不能提交下一个请求。

b.基于线程池的异步 IO。同一组请求(请求1-3,以及请求4-6)可以连续提交,而不需要等待前一条请求返回。然而,一旦调用者使用 Future.get() 获取响应数据(result1-3),就会阻塞而无法再提交下一组请求(请求4-6),直至实际收到响应数据。

c.基于 Promise 的异步 IO。 调用者随时可以提交请求,并向Promise注册对响应数据的回调函数;稍后接收线程向Promise通知响应数据,以触发回调函数。上述过程中,调用者线程不需要等待响应数据,始终不会阻塞。

图2-1a 线程时间线:同步IO

图2-1b 线程时间线:基于线程池的异步IO

图2-1c 线程时间线:基于 Promise 的异步IO

2.2 Promise结合线程池

和 ExecutorService+Future相比,Promise 具有纯异步的优点;然而在某些场景下也需要把 Promise 和线程池结合使用。例如:1.底层 API 只支持同步阻塞模型,不支持纯异步;此时只能在线程池中调用 API,才能做到非阻塞。2.需要重构一段遗留代码,将其线程模型从线程池模型改为响应式模型;可以先将对外接口改为 Promise API,而底层实现暂时使用线程池。

下面的代码片段展示了 Promise 和线程池结合的用法:

public PromiseOrException writeAsync() {
// 1.创建Promise对象
    PromiseOrException resultPromise = new PromiseOrException<>();
    try {
        executor.execute(() -> {
            String result = db.writeSync("data"); // 2a.执行请求。只支持同步阻塞
            resultPromise.signalAllWithResult(result); // 2b.通知Promise
        });

    }catch (RejectedExecutionException e){ // 3.异常:线程池满
        resultPromise.signalAllWithException(e); 
    }

    return resultPromise;
}

3.异常处理:PromiseOrException

Java 程序有时会遇到不可避免的异常情况,如网络连接断开;因此,程序员需要设计适当的异常处理机制,以提升程序的容错性。本章介绍异步 API 的异常处理,首先介绍 Java 语言异常处理规范;然后介绍 Promise 的变体 PromiseOrException,使得 Promise API 支持规范的异常处理。

3.1异常处理规范

个人认为,Java 代码的异常处理应当符合下列规范:

区分正常出口和异常出口

异常是 Java 语言的重要特性,是一种基本的控制流。Java 语言中,一个函数允许有一个返回值,以及抛出多个不同类型的异常。函数的返回值是正常出口,函数返回说明函数能够正常工作,并计算出正确的结果;相反,一旦函数遇到异常情况无法继续工作,如网络连接断开、请求非法等,就要抛出相应的异常。

虽然 if-else 和异常都是控制流,但是程序员必须辨析二者的使用场景。if-else 的各个分支一般是对等的,都用于处理正常情况;而函数的返回值和异常是不对等的,抛出异常表示函数遇到无法处理的故障,已经无法正常计算结果,其与函数正常工作所产生的返回值有本质区别。在 API 设计中,混淆正常出口(返回值)与异常出口(抛出异常),或者在无法继续工作时不抛异常,都是严重的设计缺陷。

以数据库访问为例,下面的代码对比了API进行异常处理的两种形式。数据库访问过程中,如果网络连接顺畅,并且服务端能够正确处理请求,那么db.write()应该返回服务端的响应数据,如服务端为所写数据生成的自增id、条件更新实际影响的数据条数等;如果网络连接断开,或者客户端和服务端版本不匹配导致请求无法解析,从而无法正常工作,那么db.write()应该抛出异常以说明具体原因。从“是否正常工作”的角度看,上述两种情况的性质是截然不同的,显然应该选用异常作为控制流,而不是 if-else。

/* 正确*/
try {
    String result = db.write("data");
    process(result); // 正常出口
} catch (Exception e) {
    log.error("write fails", e); // 异常出口
}

/* 错误*/
String resultOrError = db.write("data");
if (resultOrError.equals("OK")) {
    process(resultOrError); // 正常出口
} else {
    log.error("write fails, error: " + resultOrError); // 异常出口
} 

强制处理不可避免的异常

Java 语言的异常处理体系中,异常主要分为以下几类:Exception、RuntimeException、Error;三者都是Throwable 的子类,即可以被函数抛出。注意,由于 RuntimeException 是 Exception 的子类,本文为避免混淆,“Exception”特指“是 Exception 但不是 RuntimeException ”的那些异常。

个人认为,几种异常类型分别用于下列场景

1. Exception:程序外部的不可抗力造成的异常情况,如网络连接断开。即使 Java 代码完美无瑕,也绝对不可能避免这类异常(拔掉网线试试!)。既然无法避免,这种异常就应当强制处理,以提升系统的容错能力。

2. RuntimeException:编程错误造成的异常情况,如数组下标越界ArrayOutOfBoundException、参数不符合取值范围 IllegalArgumentException 等。如果程序员对 API 的输入约束了如指掌,并在调用 API 之前对函数参数进行适当校验,那么 RuntimeException 是可以绝对避免的(除非被调 API 在应当抛 Exception 处,实际抛出了RuntimeException)。既然可以避免,这种异常就没有必要强制处理。

当然,人无完人。假设程序员真的违背了某些约束,函数抛出 RuntimeException 且未被处理,那么作为惩罚,线程或进程会退出,从而提醒程序员改正错误代码。如果线程或进程必须常驻,就要对 RuntimeException 进行兜底,如下面的代码所示。这里将代码缺陷视为无法避免的异常情况,捕获异常后可以记录日志、触发告警,提醒稍后来修正缺陷。

new Thread(()->{
   while (true){
       try{
           doSomething();
       }catch (RuntimeException e){  // 对RuntimeException进行兜底,以防线程中断
 log.error("error occurs", e);
       }
   }
});

3. Error:jvm内部定义的异常,如 OutOfMemoryError。业务逻辑一般不抛出 Error,而是抛出某种Exception或 RuntimeException。

上述几类异常中,只有 Exception 是强制处理的,称为 checked exception[C]。如下所示是一个 checked exception的例子。数据库访问DB.write()抛出Exception,表示遇到网络断开、消息解析失败等不可抗情况。异常类型为Exception而不是RuntimeException,以强制调用者添加catch子句处理上述情况;如调用者遗漏了 catch子句,则编译器会报错,从而提示调用者“这里一定会遇到异常情况,必须进行处理”,以完善程序容错能力。

/**
 * 抛出异常,如果:
 * 1.网络连接断开
 * 2.消息无法解析
 * 3.业务逻辑相关,如服务端扣款时发现余额不足
 * 4.…… // 任何无法避免的情况,都应该抛出Exception!
 */
public String write(Object data) throws Exception {
    return "";
}

/**
 * 处理异常
 */
try {
    String result = db.write("data");
    process(result); 
} catch (Exception e) { // 如遗漏catch子句,则编译不通过
    log.error("write fails, db: ..., data: ...", e);
}

3.2 Promise API的异常处理

上一小节讨论了异常处理的规范:

  • 显式区分正常出口和异常出口;

  • 不可抗的异常,要在编译时刻强制处理。下面的代码展示了 Promise API 要如何设计异常处理机制,以符合上述规范。

// extends Promise>
 PromiseOrException resultPromsie = db.writeAsync("data");

 /* 重载1*/ 
resultPromsie.await(resultOrException -> { 
    try {
        String result = resultOrException.get();
        process(result);  // 正常出口
     } catch (Exception e) {
        log.error("write fails", e);  // 异常出口
 }
});
 
 /* 重载2*/
 resultPromsie.await((result, e) -> {
    if (e == null) {
        process(result);  // 正常出口
 } else {
        log.error("write fails", e);  // 异常出口
 }
});

PromiseOrException符合上一小节提出的异常处理规范,具有如下优点

最后,在异常处理结构方面个人提出一个建议:全部异常通过 PromiseOrException 来通知,而 API 本身不要抛出异常。以数据库访问 API writeAsync()为例,面的代码对比了两种抛异常的方式。正确的做法是PromiseOrException 作为唯一出口,如果 API 底层实现抛出异常(submit() throws Exception),则应该将异常封装于 PromiseOrException 对象,而不应该直接从API函数抛出(writeAsync() throws Exception)。

/* 正确:唯一出口PromiseOrException*/
public PromiseOrException writeAsync(Object data)  {
    try {
        submit(data);  // throws exception
    } catch (Exception e) {
        return PromiseOrException.immediatelyException(e);
    }
    PromiseOrException resultPromise = new PromiseOrException<>();
    return resultPromise;
}

/* 错误:两个出口throws Exception和PromiseOrException*/
public PromiseOrException writeAsync(Object data) throws Exception {
    submit(data);  // throws exception

    PromiseOrException resultPromise = new PromiseOrException<>();
    return resultPromise;
}

如果错误地设计了含有两个异常出口的 API,调用者就不得不重复书写异常处理逻辑,如下面的代码所示。

try {
    PromiseOrException resultPromise = db.writeAsync("data");
    resultPromise.await((result, e) -> {
        if (e == null) {
            process(result);  // 正常出口
 } else {
            log.error("write fails", e);  // 异常出口2
 }
    });
} catch (Exception e) {
    log.error("write fails", e);  // 异常出口1
}

4.请求调度

Java 程序中有时需要提交多条异步请求,且这些请求之间存在一定的关联关系。在异步非阻塞场景下,这些关联关系都可以借助 Promise 来实现。

1.顺序请求,如图4-1所示。后一条请求依赖前一条请求的响应数据;因此,必须等待前一条请求返回,才能构造并提交下一条请求。

图4-1 顺序请求

2.并行请求,如图4-2所示。一次提交多条请求,然后等待全部请求返回。所提交的请求之间没有依赖关系,因此可以同时执行;但是必须收到每条请求的响应数据(发生channelRead()事件,事件参数为响应数据),才能执行实际的处理process(result1,2,3)。

图4-2 并行请求

3.批量请求,如图4-3所示。调用者连续提交多条请求,但是暂存在队列中(offer()),而不是立刻执行。一段时间后,从队列中取出若干条请求,组装为批量请求来提交(writeBulk());当收到批量请求的响应消息时,可以从中取出每条请求的响应数据。由于每次网络IO都带来额外开销,故实际应用中经常使用批量请求来减少网络IO频率,以提升总体吞吐量。

图4-3 批量请求

4.1.顺序请求:Promise.then()

假设一系列操作需要依次完成,即前一操作完成后,才能开始执行下一操作;如果这些操作均表现为 Promise API,我们可以对 Promise.await(listener)进行封装,使代码结构更加简洁。

如下所示是一个异步 Promise API。submit 方法提交请求 request 并返回 Promise 对象;当收到响应数据时, 该 Promise 对象被通知。

/**
 * 异步Promise API
 */
 public static Promise submit(Object request) {
    Promise resultPromise = new Promise<>();
     // ……
 return resultPromise;
}

现假设有5个请求称为“A”-“E”,这些请求必须依次提交。例如,由于请求B的参数依赖请求 A 的响应数据,故提交请求A后必须先处理其响应数据 resultA,然后才能再提交请求B。这种场景可以用如下所示的代码来实现。某次调用submit("X")函数后,我们在其返回的 Promise 对象上注册回调;回调函数内处理响应数据 resultX,并调用submit("X+1")来提交下一请求。

这种方式虽然能实现功能需求,但是嵌套式的代码结构可读性非常差——每增加一个请求,代码就要多嵌套、缩进一个层级。当调用逻辑复杂、请求数较多时,代码会非常难以维护。

这种情况也称为 “回调地狱”[D],在 JavaScript 语言中相关讨论颇多,可以作为参考。

submit("A").await(resultA -> {
    submit("B").await(resultB -> {
        submit("C").await(resultC -> {
            submit("D").await(resultD -> {
                submit("E").await(resultE -> {
                    process(resultE);
                });
            });
        });
    });
});

为改进代码结构,我们对 Promise.await(Consumer) 方法进行封装,提供Promise.then(Function>)方法,如下所示。类似于await(),then()也可以注册一个回调函数resultX->submit("X+1"),回调函数处理响应数据resultX,并提交下一请求submit("X+1");then()的返回值即submit("X+1")的返回值,用于通知下一请求的响应数据resultX+1。

Promise resultPromiseA = submit("A");
Promise resultPromiseB = resultPromiseA.then(resultA -> submit("B"));
Promise resultPromiseC = resultPromiseB.then(resultB -> submit("C"));
Promise resultPromiseD = resultPromiseC.then(resultC -> submit("D"));
Promise resultPromiseE = resultPromiseD.then(resultD -> submit("E"));
resultPromiseE.await(resultE -> process(resultE));

接下来,我们将中间变量 resultPromiseA-E 内联,即得到基于then()的链式调用结构。相比于await(),then()消除了套娃般的嵌套回调。

submit("A")
        .then(resultA -> submit("B")) // 返回resultPromiseB
        .then(resultB -> submit("C")) // 返回resultPromiseC
        .then(resultC -> submit("D")) // 返回resultPromiseD
        .then(resultD -> submit("E")) // 返回resultPromiseE
        .await(resultE -> process(resultE));

最后,我们来看一下 Promise.then() 的一种简单实现,如下所示:

public  Promise then(Function> func) {
    Promise nextPromise = new Promise<>();
    await(result -> {
        Promise resultPromiseNext = func.apply(result);
        resultPromiseNext.await(nextPromise::signalAll);
    });
    return nextPromise;
}

注意,这里只展示了纯异步重载Promise.then(Function>)。根据回调函数是否有返回值、同步执行还是异步执行,Promise可以提供then()的更多种重载;受限于Java语法,如编译器无法辨析各个重载,则可以使用函数名称进行显式区别,如:

thenRun(Runnable)

thenAccept(Consumer)

thenApply(Function)

thenApplyAsync(Function>)

4.2.并行请求:LatchPromise

上一小节介绍了“顺序请求”的场景,即多条请求需要依次执行;而“并行请求”场景下,多条请求之间没有顺序约束,但是我们仍然需要等待全部请求返回,才能执行后续操作。例如,我们需要查询多张数据库表,这些查询语句可以同时执行;但是必须等待每条查询都返回,我们才能获得完整信息。jdk 提供CountDownLatch 来实现这一场景,但是其只支持同步等待;作为改进,我们采用 LatchPromise 实现相同的功能,并且支持纯异步API。

以数据库访问为例,如下所示的代码展示了 LatchPromise 的使用

/* 创建Promise对象*/
Promise resultPromise1 = db.writeAsync("a");
Promise resultPromise2 = db.writeAsync("b");
Promise resultPromise3 = db.writeAsync("c");

/* 向LatchPromise注册要等待的Promise*/
LatchPromise latch = new LatchPromise();
latch.add(resultPromise1);
latch.add(resultPromise2);
latch.add(resultPromise3);

/* 等待全部Promise被通知*/
VoidPromise allSignaled = latch.untilAllSignaled();
allSignaled.await(() -> {
    String result1 = resultPromise1.await();
    String result2 = resultPromise2.await();
    String result3 = resultPromise3.await();
    process(result1, result2, result3);
});

作为对比,下面的代码使用 CountDownLatch 实现相同功能,但是存在以下缺陷

CountDownLatch latch = new CountDownLatch(3);
resultPromise1.await(result1 -> latch.countDown());
resultPromise2.await(result2 -> latch.countDown());
resultPromise3.await(result3 -> latch.countDown());

latch.await();
String result1 = resultPromise1.await();
String result2 = resultPromise2.await();
String result3 = resultPromise3.await();
process(result1, result2, result3);  

最后,我们来看一下 LatchPromise 的参考实现。代码原理如下所示:

// private static class Lock。无成员,仅用于synchronized(lock)
private final Lock lock = new Lock();
private int countUnfinished = 0;
private final VoidPromise allSignaled = new VoidPromise();

public void add(Promise promise) {
    if (promise.isSignaled()) {
        return;
    }

    synchronized (lock) {
        countUnfinished++;
    }

    promise.await(unused -> {
        synchronized (lock) {
            countUnfinished--;
            if (countUnfinished == 0 && noMore) {
                allSignaled.signalAll();
            }
        }
    });
}

public VoidPromise untilAllSignaled() {
    synchronized (lock) {
        if (countUnfinished == 0) {
            allSignaled.signalAll();
        } else {
            noMore = true;
        }
    }

    return allSignaled;
}

4.3.批量请求:ExecutorAsync

批量请求的特性

“批量请求”(也称“bulk”、“batch”)是指发送一条消息即可携带多条请求,主要用于数据库访问和远程调用等场景。由于减少了网络 IO 次数、节约了构造和传输消息的开销,批量请求能有效提升吞吐量。

很多数据库 API 都支持批量读写,如JDBC PreparedStatement[E]、elasticsearch bulk API[F]、mongo DB insertMany()[G]、influx DB BatchPoints[H],读者可以查阅参考文献进一步了解。为了提升性能,部分API会牺牲易用性。其中,elasticsearch bulk API 对调用者的限制最少,允许混杂增删改等不同类型的请求,允许写入不同的数据库表(index);mongo DB、influx DB 次之,一个批量请求只能写入同一个数据库表,但是可以自定义每条数据的字段;PreparedStatement 的灵活性最低,其定义了 SQL 语句的模版,调用者只能填写模版参数,而不能修改语句结构。

虽然数据库 API 已经支持批量访问,但是很多原生 API 仍然需要调用者自己构造批量请求,需要调用者处理请求组装、批量大小、并发请求数等复杂的细节。

在此,我们设计出通用组件 ExecutorAsync,封装请求调度策略以提供更简洁的API。ExecutorAsync 的使用流程如下面的代码片段所示:

ExecutorAsync executor = new ExecutorAsync();
Promise<...> resultPromise1 = executor.submit(new Request("data1"));
Promise<...> resultPromise2 = executor.submit(new Request("data2"));
Promise<...> resultPromise3 = executor.submit(new Request("data3"));

具体而言,ExecutorAsync 支持如下调度策略:

1.排队,如图4-4a所示。调用者提交请求Request后不要立刻执行,而是将其缓存在队列queue中。

图4-4a ExecutorAsync特性:排队

2.批量,如图4-4b所示。每隔固定时间间隔,ExecutorAsync从队列中取出若干条请求,将其组装为批量请求bulk,并调用底层数据库API提交给服务端。如果队列长度增长得很快,我们也可以定义一个批量大小bulk size,当队列长度到达该值时立刻组装一个批量请求并提交。

图4-4b ExecutorAsync 特性:批量

3.并发,如图4-4c所示。如果底层数据库 API 支持异步提交请求,那么 ExecutorAsync 就可以充分利用这种特性,连续提交多个批量请求,而不需要等待之前的批量请求返回。为避免数据库服务器超载,我们可以定义并发度 parallelism,限制正在执行(in flight)的批量请求的数目;当达到限制时,如果调用者再提交新的请求,就暂存在队列 queue 中等待执行,而不会组装新的批量请求。

图4-4c ExecutorAsync 特性:并发

4.丢弃。如图4-4d所示。在上文提到的 bulk size 和 parallelism 的限制下,如果提交请求的速率远高于服务端响应的速率,那么大量请求就会堆积在队列中等待处理,最终导致超时失败。在这种情况下,将请求发送给服务端已经没有意义,因为调用者已经认定请求失败,而不再关心响应数据。

图4-4d 请求超时

因此,ExecutorAsync 应该及时从队列中移除无效请求,而剩余请求仍然“新鲜”。这种策略能够强制缩短队列长度,以降低后续请求在队列中的堆积时长、预防请求超时;同时,由于避免存储和发送无效请求,这种策略也能节约内存和 IO 开销。

图4-4e ExecutorAsync特性:丢弃

批量请求的实现

上一小节我们看到了 ExecutorAsync 的调度策略,包括排队、批量、并发、丢弃。如下面的代码所示,ExecutorAsync 只需对外提供 submit(Request) 方法,用于提交单条请求。请求以数据对象 Request 表示,其字段 Request.resultPromise 是 Promise 对象,用于通知响应数据;在需要进行异常处理的场景下,我们使用 PromiseOrException作为 Promise 的实现,其中泛型模版T改为响应数据的实际类型。

public class ExecutorAsync {

    public PromiseOrException submit(Request request) {
        return request.resultPromise;
    }
}

接下来我们来看看 ExecutorAsync 的实现原理。由于源码细节较多、篇幅较长,故本节用流程图的形式,来讲解更高层的设计,如图4-5所示。

图4-5 ExecutorAsync原理

1.提交请求。调用者调用 ExecutorAsync.submit(Request),每次调用提交一条请求。该条请求存入队列 queue,等待后续调度执行。参数 Request 的结构如下面的代码所示,包括下列字段:

predicate:函数,判断请求是否有效,无效请求(如超时的请求)将被丢弃。详见步骤2。

resultPromise:通知响应数据。

public class Request {
    public final PredicateE predicate; 
    public final PromiseOrException resultPromise;
}

2.每隔固定间隔,或者queue.size()达到bulk size,尝试组装批量请求。从队列queue中依次取出请求,每条请求执行函数 Request.predicate,以判断是否仍然要提交该请求;取出的有效请求的条数,不超过bulk size。predicate 是一个函数,类似于 jdk Predicate 接口,形式如下面的代码所示。接口函数test()可以正常返回,表示请求仍然有效;也可以抛出异常,说明请求无效的原因,如等待超时。如果抛出异常,则该条请求直接丢弃,并将发生的异常将通知给 Request.resultPromise,使得调用者执行异常处理逻辑。

public interface PredicateE {
    void test() throws Exception;
}

3.提交批量请求。第2步从队列 queue 中取出了至多 bulk size 条请求,将其作为参数调用RequestFunc.execute(requests),以提交批量请求。接口 RequestFunc 的形式如下面的代码所示。接口方法execute(requests)以若干条请求为参数,将其组装为批量请求,调用底层的数据库 API 来提交。

public interface RequestFunc{
    void execute(List> requests);
}

4.当收到响应后,对于每条请求,依次向 Request.resultPromise 通知响应数据

5.为防止服务端超载,ExecutorAsync 可限制并发请求数不超过 parallelism。我们设置计数变量 inFlight=0,以统计正在执行的批量请求数:

a.当尝试组装批量请求(步骤2)时,首先判断inFlight

b.当提交批量请求(步骤3,RequestFunc.execute())后,inFlight++。

c.当一批请求均收到响应数据(步骤4,Request.resultPromise被通知)后,inFlight--。此时如queue中的请求数仍超过bulk size,则回到步骤2,再取出一批请求来执行。

综上,ExecutorAsync使用队列queue来暂存待执行的请求;当需要提交批量请求时,以PredicateE筛选有效请求、丢弃无效请求;对于一批请求,调用RequestFunc.execute()来批量提交,收到响应数据后Request.resultPromise通知。上述过程满足约束,以防服务端超载:一批请求的数目至多为bulk size;同时正在执行的批量请求数不超过parallelism。以上即为ExecutorAsync的基本原理;实际应用中还需要处理配置参数、泛型模版等细节,限于篇幅原因本文不再详细讲解。

5.总结

本文介绍了异步模型Promise设计模式的实际应用场景,讨论了异步API的设计原则,并介绍了相应的解决方案。异步模式不仅仅是“提交请求-处理响应”的简单过程,有时也需要设计异常处理机制,以及根据请求之间的关联关系进行调度。在处理这些复杂场景的时候,API需要保持纯异步的特性,在提交请求、处理响应的过程中都不能阻塞;需要充分利用编译时刻检查,防止调用者遗漏分支,尤其是不可避免的异常分支;API需要封装复杂的、重复的实现细节,尽量保持调用者的代码结构简洁易懂。

本系列文章旨在对异步模式进行科普,希望能起到抛砖引玉的作用,帮助读者理解异步模式的基本原理,对有可能遇到的实际问题有所了解,并初步探索异步模式的实现机制。然而,现实中的项目和工具远比本文介绍的复杂,还请读者做好调研、选型工作。考察现有的各种异步API,读者会发现异步模式目前仍无统一标准。以数据库客户端为例,各种异步API的函数形式都不尽相同,listener API和Promise API都有采用,也有一些API形式上是异步的,但是在某些情况下会发生阻塞;异步API底层有基于线程池/连接池实现的,也有基于响应式模型(如netty)实现的。因此,请读者务必充分了解异步工具的API形式、阻塞特性、线程模型,然后才能在项目中应用;如果现有工具不符合开发规范,亦可大胆地进行封装,或者自行实现所需特性。关于异步工具的封装和实现,也非常欢迎读者交流与指正。

参考文献

[A] jdk CompletableFuture

https://www.baeldung.com/java-completablefuture

[B] netty DefaultPromise

https://www.tabnine.com/code/java/classes/io.netty.util.concurrent.DefaultPromise

[C] checked exception

https://www.geeksforgeeks.org/checked-vs-unchecked-exceptions-in-java/

[D] 回调地狱(JavaScript)

https://blog.avenuecode.com/callback-hell-promises-and-async/await

[E] 批量请求:JDBC PreparedStatement

https://www.tutorialspoint.com/jdbc/jdbc-batch-processing.htm

[F] 批量请求:elasticsearch bulk API

https://www.elastic.co/guide/en/elasticsearch/reference/current/docs-bulk.html

[G] 批量请求:mongo DB insertMany()

https://mongodb.github.io/mongo-java-driver/3.0/driver-async/getting-started/quick-tour/

[H] 批量请求:influx DB BatchPoints