77 lines
2.5 KiB
Java
77 lines
2.5 KiB
Java
package it.tdlight.common;
|
|
|
|
import it.tdlight.jni.TdApi.Object;
|
|
import java.util.concurrent.ConcurrentHashMap;
|
|
import java.util.concurrent.atomic.AtomicLong;
|
|
import java.util.concurrent.atomic.AtomicReference;
|
|
|
|
public class InternalClientManager implements AutoCloseable {
|
|
|
|
private static final AtomicReference<InternalClientManager> INSTANCE = new AtomicReference<>(null);
|
|
|
|
private final String implementationName;
|
|
private final ResponseReceiver responseReceiver = new ResponseReceiver(this::handleClientEvents);
|
|
private final ConcurrentHashMap<Integer, ClientEventsHandler> registeredClientEventHandlers = new ConcurrentHashMap<>();
|
|
private final ConcurrentHashMap<ClientEventsHandler, java.lang.Object> unregisteredClientEventHandlers = new ConcurrentHashMap<>();
|
|
private final AtomicLong currentQueryId = new AtomicLong();
|
|
|
|
private InternalClientManager(String implementationName) {
|
|
try {
|
|
Init.start();
|
|
} catch (Throwable ex) {
|
|
ex.printStackTrace();
|
|
System.exit(1);
|
|
}
|
|
this.implementationName = implementationName;
|
|
}
|
|
|
|
public static InternalClientManager get(String implementationName) {
|
|
return INSTANCE.updateAndGet(val -> val == null ? new InternalClientManager(implementationName) : val);
|
|
}
|
|
|
|
private void handleClientEvents(int clientId, boolean isClosed, long[] clientEventIds, Object[] clientEvents) {
|
|
ClientEventsHandler handler = registeredClientEventHandlers.get(clientId);
|
|
|
|
if (handler == null) {
|
|
handler = unregisteredClientEventHandlers
|
|
.keySet()
|
|
.stream()
|
|
.filter(item -> item.getClientId() == clientId)
|
|
.findAny()
|
|
.orElse(null);
|
|
}
|
|
|
|
if (handler != null) {
|
|
handler.handleEvents(isClosed, clientEventIds, clientEvents);
|
|
} else {
|
|
System.err.println("Unknown client id " + clientId + ", " + clientEvents.length + " events have been dropped!");
|
|
}
|
|
|
|
if (isClosed) {
|
|
registeredClientEventHandlers.remove(clientId);
|
|
}
|
|
}
|
|
|
|
public void preregisterClient(ClientEventsHandler client) {
|
|
this.unregisteredClientEventHandlers.put(client, new java.lang.Object());
|
|
}
|
|
|
|
public void registerClient(int clientId, InternalClient internalClient) {
|
|
registeredClientEventHandlers.put(clientId, internalClient);
|
|
unregisteredClientEventHandlers.remove(internalClient);
|
|
}
|
|
|
|
public String getImplementationName() {
|
|
return implementationName;
|
|
}
|
|
|
|
public long getNextQueryId() {
|
|
return currentQueryId.updateAndGet(value -> (value >= Long.MAX_VALUE ? 0 : value) + 1);
|
|
}
|
|
|
|
@Override
|
|
public void close() throws InterruptedException {
|
|
responseReceiver.close();
|
|
}
|
|
}
|