Question

我有一个会话列表，我必须调用webservice来为每个会话设置一些属性。

我正在尝试使用异步进程调用webservice并使用completablefuture，以便在完成所有操作后，我可以将它们全部保存在db中。

我该怎么做？到目前为止，我的代码如下，它不起作用。

sessions.stream()
        .forEach(s -> CompletableFuture.runAsync(() -> webServiceCall(s), executor));
sessionService.saveAll(sessions);

编辑：

我提出了这个解决方案，不确定这是否是正确的方法。

List<CompletableFuture<Void>> futures = sessions.stream()
            .map(s -> CompletableFuture.runAsync(() -> webServiceCall(s), executor))
            .collect(Collectors.toList());
        CompletableFuture.allOf(futures.toArray(new CompletableFuture[0]))
                        .join();
        sessionService.saveAll(sessions);

我正在使用join来确保在保存会话之前等待响应返回

Answer 1

简而言之 - 你只需要这样的东西 -

CompletableFuture.supplyAsync(this::supplySomething, ex).thenAccept(this::consumer);

您需要一个将在执行程序（线程池）中调用的方法。在我的情况下，我的池大小为100.接下来，您需要根据需要多次致电供应商。

每次调用“供应商”都会创建一个任务。我正在创建10000个任务。他们每个人都会并行运行，他们每个人在完成后都会打电话给我的“消费者”。

您的供应商应该返回某种对象，该对象包含来自Web服务的响应。然后，该对象将成为“消费者”方法的参数。

您可能希望在完成所有操作后（或中间）杀死池。

请参阅下面的示例 -

package com.sanjeev.java8.thread;

import java.io.BufferedReader;
import java.io.DataOutputStream;
import java.io.InputStreamReader;
import java.io.Reader;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.TimeUnit;

public class Caller {

public static ExecutorService ex = Executors.newFixedThreadPool(100);

public static void main(String[] args) throws InterruptedException {
    Caller caller = new Caller();

    caller.start();

    ex.shutdown();
    ex.awaitTermination(10, TimeUnit.MINUTES);
}

private void start() {
    for (int i = 0; i < 10000; i++) {
        CompletableFuture.supplyAsync(this::supplySomething, ex).thenAccept(this::consumer);
    }
}

private int supplySomething() {
    try {
        URL url = new URL("http://www.mywebservice.com");

        HttpURLConnection connection = (HttpURLConnection) url.openConnection();
        connection.setRequestMethod("POST");
        connection.setDoOutput(true);
        connection.setDoInput(true);

        connection.connect();

        try (DataOutputStream wr = new DataOutputStream(connection.getOutputStream())) {
            wr.write("supply-some-data".getBytes());
        }

        Reader in = new BufferedReader(new InputStreamReader(connection.getInputStream(), "UTF-8"));

        for (int c; (c = in.read()) >= 0;) {
            System.out.print((char) c);
        }

        in.close();

        // return the response code. I'm return 'int', you should return some sort of object.
        return 200;

    } catch (Exception e) {
        e.printStackTrace();
        throw new RuntimeException(e);
    }
}

public void consumer(Integer i) {
    // This parameter should be of type 'your object' that supplier returned.
    // I got the response; add it in the list or whatever....
}

}

另一个可能更适合您需求的例子 -

public class Caller2 {

public static ExecutorService ex = Executors.newFixedThreadPool(2);
private static Iterator<String> addresses = Stream.of("www.google.com", "www.yahoo.com", "www.abc.com").collect(Collectors.toList()).iterator();
private static ArrayList<String> results = new ArrayList<>();

public static void main(String[] args) throws InterruptedException {
    Caller2 caller = new Caller2();

    caller.start();

    ex.shutdown();
    ex.awaitTermination(1, TimeUnit.HOURS);

    System.out.println(results);
}

private void start() {
    while (addresses.hasNext()) {
        CompletableFuture.supplyAsync(this::supplyURL, ex).thenAccept(this::consumer);
    }
}

private String supplyURL() {
    String url = addresses.next();
    // call this URL and return response;
    return "Success";
}

public void consumer(String result) {
    results.add(result);
}

CompletableFuture用于进行Web服务调用并在一切完成后保存

1 个答案: