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

Improve performance

Browse Source
This commit is contained in:
Andrea Cavalli 2022-09-13 22:37:12 +02:00
parent 73fbd89de0
commit c7c2b03ef1
1 changed files with 18 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -44,7 +44,9 @@ public final class InternalReactiveClient implements ClientEventsHandler, Reacti
private final InternalClientManager clientManager; private final InternalClientManager clientManager;
private final AtomicBoolean alreadyReceivedClosed = new AtomicBoolean(); private final AtomicBoolean alreadyReceivedClosed = new AtomicBoolean();
private final AtomicReference<SignalListener> signalListener = new AtomicReference<>(new ReplayStartupUpdatesListener()); // This field is not volatile, but it's not problematic, because ReplayStartupUpdatesListener is able to forward
// updates to the right listener
private SignalListener signalListener = new ReplayStartupUpdatesListener();
public InternalReactiveClient(InternalClientManager clientManager) { public InternalReactiveClient(InternalClientManager clientManager) {
this.clientManager = clientManager; this.clientManager = clientManager;
@ -98,7 +100,7 @@ public final class InternalReactiveClient implements ClientEventsHandler, Reacti
} catch (InterruptedException e) { } catch (InterruptedException e) {
logger.debug(TG_MARKER, "Interrupted", e); logger.debug(TG_MARKER, "Interrupted", e);
} }
SignalListener signalListener = this.signalListener.get(); SignalListener signalListener = this.signalListener;
// Close the signal listener if it still exists // Close the signal listener if it still exists
if (signalListener != null) { if (signalListener != null) {
signalListener.onSignal(Signal.ofClosed()); signalListener.onSignal(Signal.ofClosed());
@ -260,24 +262,17 @@ public final class InternalReactiveClient implements ClientEventsHandler, Reacti
public void setListener(SignalListener listener) { public void setListener(SignalListener listener) {
logger.debug(TG_MARKER, "Setting handler of client {}", clientId); logger.debug(TG_MARKER, "Setting handler of client {}", clientId);
// Keep in mind that this lambda could be called multiple times var prevSignalListener = this.signalListener;
SignalListener resultListener = this.signalListener.updateAndGet(previousListener -> { if (!(prevSignalListener instanceof ReplayStartupUpdatesListener)) {
if (previousListener instanceof ReplayStartupUpdatesListener) {
ReplayStartupUpdatesListener replayListener = (ReplayStartupUpdatesListener) previousListener;
replayListener.setNewListener(listener);
return replayListener;
} else if (previousListener != null) {
throw new IllegalStateException("Already subscribed"); throw new IllegalStateException("Already subscribed");
} else {
return listener;
} }
}); var replayStartupUpdatesListener = (ReplayStartupUpdatesListener) prevSignalListener;
// Set the new listener into the startup listener, then drain its startup queue
// Drain startup queue replayStartupUpdatesListener.setNewListener(listener);
if (resultListener instanceof ReplayStartupUpdatesListener) {
ReplayStartupUpdatesListener replayStartupUpdatesListener = (ReplayStartupUpdatesListener) resultListener;
replayStartupUpdatesListener.drain(); replayStartupUpdatesListener.drain();
}
// Set the new listener
this.signalListener = listener;
TdApi.GetAuthorizationState query = new TdApi.GetAuthorizationState(); TdApi.GetAuthorizationState query = new TdApi.GetAuthorizationState();
long queryId = clientManager.getNextQueryId(); long queryId = clientManager.getNextQueryId();
@ -346,7 +341,7 @@ public final class InternalReactiveClient implements ClientEventsHandler, Reacti
private void onDefaultException(Throwable updateEx) { private void onDefaultException(Throwable updateEx) {
Signal item = Signal.ofUpdateException(updateEx); Signal item = Signal.ofUpdateException(updateEx);
SignalListener signalListener = this.signalListener.get(); SignalListener signalListener = this.signalListener;
if (signalListener != null) { if (signalListener != null) {
signalListener.onSignal(item); signalListener.onSignal(item);
} else { } else {
@ -356,7 +351,7 @@ public final class InternalReactiveClient implements ClientEventsHandler, Reacti
private void onUpdateException(Throwable updateEx) { private void onUpdateException(Throwable updateEx) {
Signal item = Signal.ofUpdateException(updateEx); Signal item = Signal.ofUpdateException(updateEx);
SignalListener signalListener = this.signalListener.get(); SignalListener signalListener = this.signalListener;
if (signalListener != null) { if (signalListener != null) {
signalListener.onSignal(item); signalListener.onSignal(item);
} else { } else {
@ -366,7 +361,7 @@ public final class InternalReactiveClient implements ClientEventsHandler, Reacti
private void onUpdateFromHandler(TdApi.Object updateItem) { private void onUpdateFromHandler(TdApi.Object updateItem) {
Signal item = Signal.ofUpdate(updateItem); Signal item = Signal.ofUpdate(updateItem);
SignalListener signalListener = this.signalListener.get(); SignalListener signalListener = this.signalListener;
if (signalListener != null) { if (signalListener != null) {
signalListener.onSignal(item); signalListener.onSignal(item);
} else { } else {
@ -387,7 +382,7 @@ public final class InternalReactiveClient implements ClientEventsHandler, Reacti
assert queue.isEmpty(); assert queue.isEmpty();
listener.onSignal(signal); listener.onSignal(signal);
// Replace itself with the child signal listener, to reduce overhead permanently // Replace itself with the child signal listener, to reduce overhead permanently
InternalReactiveClient.this.signalListener.set(listener); InternalReactiveClient.this.signalListener = listener;
} else { } else {
queue.add(signal); queue.add(signal);
} }