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

Reduce overhead

Browse Source
This commit is contained in:
Andrea Cavalli 2022-01-11 15:39:25 +01:00
parent 67c581897b
commit 0390481825
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/main/java/it/tdlight/common/internal/InternalReactiveClient.java
View File

@ -259,6 +259,8 @@ public final class InternalReactiveClient implements ClientEventsHandler, Reacti
@Override
public void setListener(SignalListener listener) {
logger.debug(TG_MARKER, "Setting handler of client {}", clientId);
// Keep in mind that this lambda could be called multiple times
SignalListener resultListener = this.signalListener.updateAndGet(previousListener -> {
if (previousListener instanceof ReplayStartupUpdatesListener) {
ReplayStartupUpdatesListener replayListener = (ReplayStartupUpdatesListener) previousListener;
@ -372,7 +374,7 @@ public final class InternalReactiveClient implements ClientEventsHandler, Reacti
}
}
private static class ReplayStartupUpdatesListener implements SignalListener {
private class ReplayStartupUpdatesListener implements SignalListener {
private final ConcurrentLinkedQueue<Signal> queue = new ConcurrentLinkedQueue<>();
private final AtomicReference<SignalListener> listener = new AtomicReference<>(null);
@ -384,11 +386,16 @@ public final class InternalReactiveClient implements ClientEventsHandler, Reacti
drainQueue(listener);
assert queue.isEmpty();
listener.onSignal(signal);
// Replace itself with the child signal listener, to reduce overhead permanently
InternalReactiveClient.this.signalListener.set(listener);
} else {
queue.add(signal);
}
}
/**
* This method could be called multiple times
*/
public void setNewListener(SignalListener listener) {
this.listener.set(listener);
}