X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Ffcp%2Fquelaton%2FFcpReplySequence.java;h=106bd8abceec60bd1054615215f0f9449247a930;hb=dbd390e996a8b0d94d5e57906b996a95f4a22f4e;hp=9b2db2ae9b83fd34c223b3d17c71079cdfe36da6;hpb=e556f854462736f67977c60978a13cc1e1f7ee88;p=jFCPlib.git diff --git a/src/main/java/net/pterodactylus/fcp/quelaton/FcpReplySequence.java b/src/main/java/net/pterodactylus/fcp/quelaton/FcpReplySequence.java index 9b2db2a..106bd8a 100644 --- a/src/main/java/net/pterodactylus/fcp/quelaton/FcpReplySequence.java +++ b/src/main/java/net/pterodactylus/fcp/quelaton/FcpReplySequence.java @@ -1,6 +1,8 @@ package net.pterodactylus.fcp.quelaton; import java.io.IOException; +import java.util.Queue; +import java.util.concurrent.ConcurrentLinkedQueue; import java.util.concurrent.ExecutorService; import java.util.function.Consumer; @@ -59,6 +61,7 @@ public abstract class FcpReplySequence implements AutoCloseable, FcpListener private final Object syncObject = new Object(); private final ListeningExecutorService executorService; private final FcpConnection fcpConnection; + private final Queue messages = new ConcurrentLinkedQueue<>(); public FcpReplySequence(ExecutorService executorService, FcpConnection fcpConnection) { this.executorService = MoreExecutors.listeningDecorator(executorService); @@ -69,10 +72,17 @@ public abstract class FcpReplySequence implements AutoCloseable, FcpListener public ListenableFuture send(FcpMessage fcpMessage) throws IOException { fcpConnection.addFcpListener(this); - fcpConnection.sendMessage(fcpMessage); + messages.add(fcpMessage); return executorService.submit(() -> { synchronized (syncObject) { - while (!isFinished()) { + while (!isFinished() || !messages.isEmpty()) { + while (messages.peek() != null) { + FcpMessage message = messages.poll(); + fcpConnection.sendMessage(message); + } + if (isFinished()) { + continue; + } syncObject.wait(); } } @@ -80,6 +90,17 @@ public abstract class FcpReplySequence implements AutoCloseable, FcpListener }); } + protected void sendMessage(FcpMessage fcpMessage) { + messages.add(fcpMessage); + notifySyncObject(); + } + + private void notifySyncObject() { + synchronized (syncObject) { + syncObject.notifyAll(); + } + } + protected R getResult() { return null; } @@ -91,23 +112,17 @@ public abstract class FcpReplySequence implements AutoCloseable, FcpListener private void consume(Consumer consumer, M message) { consumer.accept(message); - synchronized (syncObject) { - syncObject.notifyAll(); - } + notifySyncObject(); } private void consumeUnknown(FcpMessage fcpMessage) { consumeUnknownMessage(fcpMessage); - synchronized (syncObject) { - syncObject.notifyAll(); - } + notifySyncObject(); } private void consumeClose(Throwable throwable) { consumeConnectionClosed(throwable); - synchronized (syncObject) { - syncObject.notifyAll(); - } + notifySyncObject(); } @Override