68 lines
2.3 KiB
Java
68 lines
2.3 KiB
Java
import com.rabbitmq.client.AMQP;
|
|
import com.rabbitmq.client.Channel;
|
|
import com.rabbitmq.client.Connection;
|
|
import com.rabbitmq.client.ConnectionFactory;
|
|
|
|
import java.io.IOException;
|
|
import java.util.UUID;
|
|
import java.util.concurrent.*;
|
|
|
|
public class RPCClient implements AutoCloseable {
|
|
|
|
private Connection connection;
|
|
private Channel channel;
|
|
private String requestQueueName = "rpc_queue";
|
|
|
|
public RPCClient() throws IOException, TimeoutException {
|
|
ConnectionFactory factory = new ConnectionFactory();
|
|
factory.setHost("localhost");
|
|
|
|
connection = factory.newConnection();
|
|
channel = connection.createChannel();
|
|
}
|
|
|
|
public static void main(String[] argv) {
|
|
try (RPCClient fibonacciRpc = new RPCClient()) {
|
|
for (int i = 0; i < 32; i++) {
|
|
String i_str = Integer.toString(i);
|
|
System.out.println(" [x] Requesting fib(" + i_str + ")");
|
|
String response = fibonacciRpc.call(i_str);
|
|
System.out.println(" [.] Got '" + response + "'");
|
|
}
|
|
} catch (IOException | TimeoutException | InterruptedException | ExecutionException e) {
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
|
|
public String call(String message) throws IOException, InterruptedException, ExecutionException {
|
|
final String corrId = UUID.randomUUID().toString();
|
|
|
|
String replyQueueName = channel.queueDeclare().getQueue();
|
|
AMQP.BasicProperties props = new AMQP.BasicProperties
|
|
.Builder()
|
|
.correlationId(corrId)
|
|
.replyTo(replyQueueName)
|
|
.build();
|
|
|
|
channel.basicPublish("", requestQueueName, props, message.getBytes("UTF-8"));
|
|
|
|
final CompletableFuture<String> response = new CompletableFuture<>();
|
|
|
|
String ctag = channel.basicConsume(replyQueueName, true, (consumerTag, delivery) -> {
|
|
if (delivery.getProperties().getCorrelationId().equals(corrId)) {
|
|
response.complete(new String(delivery.getBody(), "UTF-8"));
|
|
}
|
|
}, consumerTag -> {
|
|
});
|
|
|
|
String result = response.get();
|
|
channel.basicCancel(ctag);
|
|
return result;
|
|
}
|
|
|
|
public void close() throws IOException {
|
|
connection.close();
|
|
}
|
|
}
|
|
|