tdlight-team/tdlight-java
tdlight-team/tdlight-java
1
2
Fork 0
Code Issues Releases Wiki Activity

Add shutdown hook

Browse Source
This commit is contained in:
Andrea Cavalli 2021-09-28 00:19:34 +02:00
parent 88ca5ce8ac
commit ff712809b5
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/main/java/it/tdlight/client/SimpleTelegramClient.java
View File

@ -174,6 +174,18 @@ public final class SimpleTelegramClient implements Authenticable {
this.authenticationData = authenticationData; this.authenticationData = authenticationData;
createDirectories(); createDirectories();
client.initialize(this::handleUpdate, this::handleUpdateException, this::handleDefaultException); client.initialize(this::handleUpdate, this::handleUpdateException, this::handleDefaultException);
// Handle unexpected shutdown
Runtime.getRuntime().addShutdownHook(new Thread(() -> {
try {
// Send close function
this.client.send(new TdApi.Close(), ok -> {}, ex -> {});
// Wait until the client has been closed successfully
this.waitForExit();
} catch (Throwable ignored) {
// Ignore errors since we are shutting down everything
}
}));
} }
private void createDirectories() { private void createDirectories() {