wisplite/distant-horizons-core-sharded
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Close network request tacker properly

Browse Source
This commit is contained in:
s809
2023-07-19 17:40:38 +05:00
parent 218411902e
commit 79408c081a
1 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
core/src/main/java/com/seibel/distanthorizons/core/network/future/NetworkRequestTracker.java
+10 -10
View File
@@ -44,16 +44,7 @@ public class NetworkRequestTracker
public CompletableFuture<TResponse> sendRequest(Channel channel, IFutureTrackableNetworkMessage<TKey> msg)
{
if (knownChannels.add(channel))
{
channel.closeFuture().addListener(closeFuture ->
{
pendingFutures.row(channel).values().removeIf(responseFuture ->
{
responseFuture.completeExceptionally(closeFuture.cause());
return true;
});
});
}
channel.closeFuture().addListener(closeFuture -> completeAllExceptionally(channel, closeFuture.cause()));
CompletableFuture<TResponse> responseFuture = new CompletableFuture<>();
pendingFutures.put(channel, msg.getRequestKey(), responseFuture);
@@ -66,8 +57,17 @@ public class NetworkRequestTracker
return responseFuture;
}
private void completeAllExceptionally(Channel channel, Throwable cause) {
for (CompletableFuture<TResponse> responseFuture : pendingFutures.row(channel).values()) {
responseFuture.completeExceptionally(cause);
};
pendingFutures.row(channel).clear();
}
@Override public void close()
{
this.eventSource.close();
for (Channel channel : pendingFutures.rowKeySet())
this.completeAllExceptionally(channel, new Exception(this.getClass().getSimpleName()+" is closed."));
}
}