serverscanning/capnproto-java-rpc
16
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

move cleanup to end of message loop

Browse source
This commit is contained in:
Vaci Koblizek 2020-10-24 16:51:16 +01:00
parent 94ca2a04e6
commit 86ccdd5a55
1 changed files with 10 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
runtime/src/main/java/org/capnproto/RpcState.java
View file

@ -282,7 +282,7 @@ final class RpcState {
private final ReferenceQueue<Question> questionRefs = new ReferenceQueue<>(); private final ReferenceQueue<Question> questionRefs = new ReferenceQueue<>();
private final ReferenceQueue<ImportClient> importRefs = new ReferenceQueue<>(); private final ReferenceQueue<ImportClient> importRefs = new ReferenceQueue<>();
RpcState( Capability.Client bootstrapInterface, RpcState(Capability.Client bootstrapInterface,
VatNetwork.Connection connection, VatNetwork.Connection connection,
CompletableFuture<java.lang.Void> onDisconnect) { CompletableFuture<java.lang.Void> onDisconnect) {
this.bootstrapInterface = bootstrapInterface; this.bootstrapInterface = bootstrapInterface;
@ -295,6 +295,10 @@ final class RpcState {
return this.messageLoop; return this.messageLoop;
} }
public CompletableFuture<java.lang.Void> onDisconnect() {
return this.messageLoop;
}
CompletableFuture<java.lang.Void> disconnect(Throwable exc) { CompletableFuture<java.lang.Void> disconnect(Throwable exc) {
if (isDisconnected()) { if (isDisconnected()) {
return CompletableFuture.failedFuture(this.disconnected); return CompletableFuture.failedFuture(this.disconnected);
@ -420,21 +424,20 @@ final class RpcState {
} }
private CompletableFuture<java.lang.Void> doMessageLoop() { private CompletableFuture<java.lang.Void> doMessageLoop() {
this.cleanupImports();
this.cleanupQuestions();
if (isDisconnected()) { if (isDisconnected()) {
return CompletableFuture.failedFuture(this.disconnected); return CompletableFuture.failedFuture(this.disconnected);
} }
return connection.receiveIncomingMessage().thenCompose(message -> { return connection.receiveIncomingMessage().thenCompose(message -> {
try { try {
handleMessage(message); this.handleMessage(message);
} catch (Exception rpcExc) { } catch (Exception rpcExc) {
// either we received an Abort message from peer // either we received an Abort message from peer
// or internal RpcState is bad. // or internal RpcState is bad.
return this.disconnect(rpcExc); return this.disconnect(rpcExc);
} }
this.cleanupImports();
this.cleanupQuestions();
return this.doMessageLoop(); return this.doMessageLoop();
}).exceptionallyCompose(exc -> this.disconnect(exc)); }).exceptionallyCompose(exc -> this.disconnect(exc));