boldsuck/haveno
1
0
Fork 0
mirror of https://github.com/boldsuck/haveno.git synced 2024-12-23 04:29:22 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

handle connection disconnect and shut down off UserThread

Browse source
This commit is contained in:
woodser 2023-12-20 08:37:16 -05:00
parent 43f177d1f9
commit e1ca35172d
1 changed files with 4 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
p2p/src/main/java/haveno/network/p2p/network/Connection.java
View file

@ -32,7 +32,6 @@ import haveno.network.p2p.storage.payload.CapabilityRequiringPayload;
import haveno.network.p2p.storage.payload.PersistableNetworkPayload; import haveno.network.p2p.storage.payload.PersistableNetworkPayload;
import haveno.common.Proto; import haveno.common.Proto;
import haveno.common.UserThread;
import haveno.common.app.Capabilities; import haveno.common.app.Capabilities;
import haveno.common.app.HasCapabilities; import haveno.common.app.HasCapabilities;
import haveno.common.app.Version; import haveno.common.app.Version;
@ -502,7 +501,7 @@ public class Connection implements HasCapabilities, Runnable, MessageListener {
handleException(t); handleException(t);
} finally { } finally {
stopped = true; stopped = true;
UserThread.execute(() -> doShutDown(closeConnectionReason, shutDownCompleteHandler)); EXECUTOR.execute(() -> doShutDown(closeConnectionReason, shutDownCompleteHandler));
} }
}, "Connection:SendCloseConnectionMessage-" + this.uid).start(); }, "Connection:SendCloseConnectionMessage-" + this.uid).start();
} else { } else {
@ -512,12 +511,12 @@ public class Connection implements HasCapabilities, Runnable, MessageListener {
} else { } else {
//TODO find out why we get called that //TODO find out why we get called that
log.debug("stopped was already at shutDown call"); log.debug("stopped was already at shutDown call");
UserThread.execute(() -> doShutDown(closeConnectionReason, shutDownCompleteHandler)); EXECUTOR.execute(() -> doShutDown(closeConnectionReason, shutDownCompleteHandler));
} }
} }
private void doShutDown(CloseConnectionReason closeConnectionReason, @Nullable Runnable shutDownCompleteHandler) { private void doShutDown(CloseConnectionReason closeConnectionReason, @Nullable Runnable shutDownCompleteHandler) {
UserThread.execute(() -> connectionListener.onDisconnect(closeConnectionReason, this)); EXECUTOR.execute(() -> connectionListener.onDisconnect(closeConnectionReason, this));
try { try {
protoOutputStream.onConnectionShutdown(); protoOutputStream.onConnectionShutdown();
socket.close(); socket.close();
@ -540,7 +539,7 @@ public class Connection implements HasCapabilities, Runnable, MessageListener {
log.debug("Connection shutdown complete {}", this); log.debug("Connection shutdown complete {}", this);
if (shutDownCompleteHandler != null) if (shutDownCompleteHandler != null)
UserThread.execute(shutDownCompleteHandler); EXECUTOR.execute(shutDownCompleteHandler);
} }
} }