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

Update InternalClient.java

Browse Source
This commit is contained in:
Andrea Cavalli 2020-11-12 13:01:22 +01:00
parent cfad096388
commit 73606957e7
1 changed files with 14 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/main/java/it/tdlight/common/InternalClient.java
View File

@ -1,10 +1,12 @@
package it.tdlight.common; package it.tdlight.common;
import it.tdlight.jni.TdApi.Error;
import it.tdlight.jni.TdApi.Function; import it.tdlight.jni.TdApi.Function;
import it.tdlight.jni.TdApi.Object; import it.tdlight.jni.TdApi.Object;
import it.unimi.dsi.fastutil.longs.LongArrayList; import it.unimi.dsi.fastutil.longs.LongArrayList;
import it.unimi.dsi.fastutil.objects.ObjectArrayList; import it.unimi.dsi.fastutil.objects.ObjectArrayList;
import java.util.concurrent.ConcurrentHashMap; import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.concurrent.locks.ReentrantReadWriteLock; import java.util.concurrent.locks.ReentrantReadWriteLock;
public class InternalClient implements ClientEventsHandler, TelegramClient { public class InternalClient implements ClientEventsHandler, TelegramClient {
@ -18,7 +20,7 @@ public class InternalClient implements ClientEventsHandler, TelegramClient {
private final MultiHandler updatesHandler; private final MultiHandler updatesHandler;
private final ExceptionHandler defaultExceptionHandler; private final ExceptionHandler defaultExceptionHandler;
private volatile boolean isClosed; private final AtomicBoolean isClosed = new AtomicBoolean();
public InternalClient(InternalClientManager clientManager, public InternalClient(InternalClientManager clientManager,
ResultHandler updateHandler, ResultHandler updateHandler,
@ -92,9 +94,18 @@ public class InternalClient implements ClientEventsHandler, TelegramClient {
} }
if (isClosed) { if (isClosed) {
this.isClosed = true; if (this.isClosed.compareAndSet(false, true)) {
handleClose();
} }
} }
}
private void handleClose() {
handlers.forEach((eventId, handler) -> {
handleResponse(eventId, new Error(500, "Instance closed"), handler);
});
handlers.clear();
}
/** /**
* Handles only a response (not an update!) * Handles only a response (not an update!)
@ -148,7 +159,7 @@ public class InternalClient implements ClientEventsHandler, TelegramClient {
} }
private void ensureOpen() { private void ensureOpen() {
if (isClosed) { if (isClosed.get()) {
throw new IllegalStateException("The client is closed!"); throw new IllegalStateException("The client is closed!");
} }
} }