Java 使用阻塞队列拿取异步回调的值

 本文的场景是在一个线程中去调用另外一个系统，另外一个系统只能使用回调的方式来返回请求结果，但是当前线程却需要返回这次回调的结果。

这个场景相当于一个线程等待结果，另外一个线程去唤醒它。

利用阻塞队列的特性：在阻塞队列为空的时候，线程去拿取元素会等待队列有元素进来。而且可以设置超时时间。

示例：

主线程

Map<String, ArrayBlockingQueue<String>> blockingQueueMap = new ConcurrentHashMap<>();


void sync(){
    String taskId = IdUtil.fastUUID();
    System.out.println(taskId+"--------------");
    // 发送请求
    send(taskId);
    ArrayBlockingQueue<String> blockingQueue = new ArrayBlockingQueue<>(1);
    blockingQueueMap.put(taskId,blockingQueue);
    String result = null;
    try {
         // 等待回调结果
         result = blockingQueue.poll(1011, TimeUnit.MILLISECONDS);
    } catch (InterruptedException e) {
         e.printStackTrace();
    }
     if (result==null){
         System.out.println("超时了");
         blockingQueueMap.remove(taskId);
     }
     System.out.println(taskId+result);
}

 回调方法

void result2(String taskId,String result){
        try {
            ArrayBlockingQueue<String> blockingQueue = blockingQueueMap.get(taskId);
            if (blockingQueue!=null){
                blockingQueue.put(result);
                // 当 map 发生 resize 时，可能会删除失败。
                blockingQueueMap.remove(taskId);
            }
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
 }

这样就可以做到等待和唤醒的操作了。