Unverified Commit 9daf0a31 authored by Leonardo Aramaki's avatar Leonardo Aramaki Committed by GitHub

Merge pull request #581 from RocketChat/fix/servers-logging-out

Fix servers logging out on app update
parents 975511e9 c20777bc
...@@ -20,8 +20,8 @@ android { ...@@ -20,8 +20,8 @@ android {
applicationId "chat.rocket.android" applicationId "chat.rocket.android"
minSdkVersion 16 minSdkVersion 16
targetSdkVersion rootProject.ext.targetSdkVersion targetSdkVersion rootProject.ext.targetSdkVersion
versionCode 50 versionCode 52
versionName "1.0.27" versionName "1.0.29"
testInstrumentationRunner "android.support.test.runner.AndroidJUnitRunner" testInstrumentationRunner "android.support.test.runner.AndroidJUnitRunner"
vectorDrawables.useSupportLibrary = true vectorDrawables.useSupportLibrary = true
multiDexEnabled true multiDexEnabled true
...@@ -144,7 +144,7 @@ dependencies { ...@@ -144,7 +144,7 @@ dependencies {
testCompile "org.jetbrains.kotlin:kotlin-test:$rootProject.ext.kotlinVersion" testCompile "org.jetbrains.kotlin:kotlin-test:$rootProject.ext.kotlinVersion"
testCompile "org.jetbrains.kotlin:kotlin-test-junit:$rootProject.ext.kotlinVersion" testCompile "org.jetbrains.kotlin:kotlin-test-junit:$rootProject.ext.kotlinVersion"
testCompile "org.jetbrains.kotlin:kotlin-reflect:$rootProject.ext.kotlinVersion" testCompile "org.jetbrains.kotlin:kotlin-reflect:$rootProject.ext.kotlinVersion"
testCompile "com.nhaarman:mockito-kotlin:1.1.0" testCompile "com.nhaarman:mockito-kotlin:1.5.0"
testCompile 'org.amshove.kluent:kluent:1.14' testCompile 'org.amshove.kluent:kluent:1.14'
} }
apply plugin: 'com.google.gms.google-services' apply plugin: 'com.google.gms.google-services'
package chat.rocket.android.service; package chat.rocket.android.service;
import android.annotation.SuppressLint;
import android.content.ComponentName; import android.content.ComponentName;
import android.content.Context; import android.content.Context;
import android.content.ServiceConnection; import android.content.ServiceConnection;
...@@ -68,6 +69,7 @@ import io.reactivex.subjects.PublishSubject; ...@@ -68,6 +69,7 @@ import io.reactivex.subjects.PublishSubject;
} }
} }
@SuppressLint("RxLeakedSubscription")
@DebugLog @DebugLog
@Override @Override
public void ensureConnections() { public void ensureConnections() {
...@@ -84,6 +86,7 @@ import io.reactivex.subjects.PublishSubject; ...@@ -84,6 +86,7 @@ import io.reactivex.subjects.PublishSubject;
}); });
} }
@SuppressLint("RxLeakedSubscription")
@Override @Override
public void addOrUpdateServer(String hostname, @Nullable String name, boolean insecure) { public void addOrUpdateServer(String hostname, @Nullable String name, boolean insecure) {
RealmBasedServerInfo.addOrUpdate(hostname, name, insecure); RealmBasedServerInfo.addOrUpdate(hostname, name, insecure);
...@@ -95,6 +98,7 @@ import io.reactivex.subjects.PublishSubject; ...@@ -95,6 +98,7 @@ import io.reactivex.subjects.PublishSubject;
}, RCLog::e); }, RCLog::e);
} }
@SuppressLint("RxLeakedSubscription")
@Override @Override
public void removeServer(String hostname) { public void removeServer(String hostname) {
RealmBasedServerInfo.remove(hostname); RealmBasedServerInfo.remove(hostname);
...@@ -166,7 +170,11 @@ import io.reactivex.subjects.PublishSubject; ...@@ -166,7 +170,11 @@ import io.reactivex.subjects.PublishSubject;
@DebugLog @DebugLog
private Single<Boolean> connectToServerIfNeeded(String hostname, boolean forceConnect) { private Single<Boolean> connectToServerIfNeeded(String hostname, boolean forceConnect) {
return Single.defer(() -> { return Single.defer(() -> {
final int connectivity = serverConnectivityList.get(hostname); Integer state = serverConnectivityList.get(hostname);
if (state == null) {
state = ServerConnectivity.STATE_DISCONNECTED;
}
final int connectivity = state;
if (!forceConnect && connectivity == ServerConnectivity.STATE_CONNECTED) { if (!forceConnect && connectivity == ServerConnectivity.STATE_CONNECTED) {
return Single.just(true); return Single.just(true);
} }
......
...@@ -50,418 +50,398 @@ import io.reactivex.disposables.CompositeDisposable; ...@@ -50,418 +50,398 @@ import io.reactivex.disposables.CompositeDisposable;
* Thread for handling WebSocket connection. * Thread for handling WebSocket connection.
*/ */
public class RocketChatWebSocketThread extends HandlerThread { public class RocketChatWebSocketThread extends HandlerThread {
private static final Class[] REGISTERABLE_CLASSES = { private static final Class[] REGISTERABLE_CLASSES = {
LoginServiceConfigurationSubscriber.class, LoginServiceConfigurationSubscriber.class,
ActiveUsersSubscriber.class, ActiveUsersSubscriber.class,
UserDataSubscriber.class, UserDataSubscriber.class,
MethodCallObserver.class, MethodCallObserver.class,
SessionObserver.class, SessionObserver.class,
LoadMessageProcedureObserver.class, LoadMessageProcedureObserver.class,
GetUsersOfRoomsProcedureObserver.class, GetUsersOfRoomsProcedureObserver.class,
NewMessageObserver.class, NewMessageObserver.class,
DeletedMessageObserver.class, DeletedMessageObserver.class,
CurrentUserObserver.class, CurrentUserObserver.class,
FileUploadingToUrlObserver.class, FileUploadingToUrlObserver.class,
FileUploadingWithUfsObserver.class, FileUploadingWithUfsObserver.class,
PushSettingsObserver.class, PushSettingsObserver.class,
GcmPushRegistrationObserver.class GcmPushRegistrationObserver.class
}; };
private static final long HEARTBEAT_PERIOD_MS = 20000; private static final long HEARTBEAT_PERIOD_MS = 20000;
private final Context appContext; private final Context appContext;
private final String hostname; private final String hostname;
private final RealmHelper realmHelper; private final RealmHelper realmHelper;
private final ConnectivityManagerInternal connectivityManager; private final ConnectivityManagerInternal connectivityManager;
private final ArrayList<Registrable> listeners = new ArrayList<>(); private final ArrayList<Registrable> listeners = new ArrayList<>();
private final CompositeDisposable hearbeatDisposable = new CompositeDisposable(); private final CompositeDisposable heartbeatDisposable = new CompositeDisposable();
private final CompositeDisposable reconnectSubscription = new CompositeDisposable(); private final CompositeDisposable reconnectDisposable = new CompositeDisposable();
private boolean listenersRegistered; private boolean listenersRegistered;
private static class KeepAliveTimer { private static class KeepAliveTimer {
private long lastTime; private long lastTime;
private final long thresholdMs; private final long thresholdMs;
public KeepAliveTimer(long thresholdMs) { public KeepAliveTimer(long thresholdMs) {
this.thresholdMs = thresholdMs; this.thresholdMs = thresholdMs;
lastTime = System.currentTimeMillis(); lastTime = System.currentTimeMillis();
}
public boolean shouldCheckPrecisely() {
return lastTime + thresholdMs < System.currentTimeMillis();
}
public void update() {
lastTime = System.currentTimeMillis();
}
}
private final KeepAliveTimer keepAliveTimer = new KeepAliveTimer(20000);
private RocketChatWebSocketThread(Context appContext, String hostname) {
super("RC_thread_" + hostname);
this.appContext = appContext;
this.hostname = hostname;
this.realmHelper = RealmStore.getOrCreate(hostname);
this.connectivityManager = ConnectivityManager.getInstanceForInternal(appContext);
}
/**
* build new Thread.
*/
@DebugLog
public static Single<RocketChatWebSocketThread> getStarted(Context appContext, String hostname) {
return Single.<RocketChatWebSocketThread>create(objectSingleEmitter -> {
new RocketChatWebSocketThread(appContext, hostname) {
@Override
protected void onLooperPrepared() {
try {
super.onLooperPrepared();
objectSingleEmitter.onSuccess(this);
} catch (Exception exception) {
objectSingleEmitter.onError(exception);
}
}
}.start();
}).flatMap(webSocket ->
webSocket.connectWithExponentialBackoff().map(_val -> webSocket));
} }
public boolean shouldCheckPrecisely() { @Override
return lastTime + thresholdMs < System.currentTimeMillis(); protected void onLooperPrepared() {
super.onLooperPrepared();
forceInvalidateTokens();
} }
public void update() { private void forceInvalidateTokens() {
lastTime = System.currentTimeMillis(); realmHelper.executeTransaction(realm -> {
RealmSession session = RealmSession.queryDefaultSession(realm).findFirst();
if (session != null
&& !TextUtils.isEmpty(session.getToken())
&& (session.isTokenVerified() || !TextUtils.isEmpty(session.getError()))) {
session.setTokenVerified(false);
session.setError(null);
}
return null;
}).continueWith(new LogIfError());
} }
}
/**
private final KeepAliveTimer keepAliveTimer = new KeepAliveTimer(20000); * terminate WebSocket thread.
*/
private RocketChatWebSocketThread(Context appContext, String hostname) { @DebugLog
super("RC_thread_" + hostname); public Single<Boolean> terminate() {
this.appContext = appContext; if (isAlive()) {
this.hostname = hostname; return Single.create(emitter -> {
this.realmHelper = RealmStore.getOrCreate(hostname); new Handler(getLooper()).post(() -> {
this.connectivityManager = ConnectivityManager.getInstanceForInternal(appContext); RCLog.d("thread %s: terminated()", Thread.currentThread().getId());
} unregisterListenersAndClose();
connectivityManager.notifyConnectionLost(hostname,
/** DDPClient.REASON_CLOSED_BY_USER);
* build new Thread. RocketChatWebSocketThread.super.quit();
*/ emitter.onSuccess(true);
@DebugLog });
public static Single<RocketChatWebSocketThread> getStarted(Context appContext, String hostname) { });
return Single.<RocketChatWebSocketThread>fromPublisher(objectSingleEmitter -> { } else {
new RocketChatWebSocketThread(appContext, hostname) { connectivityManager.notifyConnectionLost(hostname,
@Override DDPClient.REASON_NETWORK_ERROR);
protected void onLooperPrepared() { super.quit();
try { return Single.just(true);
super.onLooperPrepared();
objectSingleEmitter.onNext(this);
objectSingleEmitter.onComplete();
} catch (Exception exception) {
objectSingleEmitter.onError(exception);
}
} }
}.start();
}).flatMap(webSocket ->
webSocket.connectWithExponentialBackoff().map(_val -> webSocket));
}
@Override
protected void onLooperPrepared() {
super.onLooperPrepared();
forceInvalidateTokens();
}
private void forceInvalidateTokens() {
realmHelper.executeTransaction(realm -> {
RealmSession session = RealmSession.queryDefaultSession(realm).findFirst();
if (session != null
&& !TextUtils.isEmpty(session.getToken())
&& (session.isTokenVerified() || !TextUtils.isEmpty(session.getError()))) {
session.setTokenVerified(false);
session.setError(null);
}
return null;
}).continueWith(new LogIfError());
}
/**
* terminate WebSocket thread.
*/
@DebugLog
public Single<Boolean> terminate() {
if (isAlive()) {
return Single.fromPublisher(emitter -> {
new Handler(getLooper()).post(() -> {
RCLog.d("thread %s: terminated()", Thread.currentThread().getId());
unregisterListenersAndClose();
connectivityManager.notifyConnectionLost(hostname,
DDPClient.REASON_CLOSED_BY_USER);
RocketChatWebSocketThread.super.quit();
emitter.onNext(true);
emitter.onComplete();
});
});
} else {
connectivityManager.notifyConnectionLost(hostname,
DDPClient.REASON_NETWORK_ERROR);
super.quit();
return Single.just(true);
} }
}
/**
/** * THIS METHOD THROWS EXCEPTION!! Use terminate() instead!!
* THIS METHOD THROWS EXCEPTION!! Use terminate() instead!! */
*/ @Deprecated
@Deprecated @Override
@Override public final boolean quit() {
public final boolean quit() { throw new UnsupportedOperationException();
throw new UnsupportedOperationException();
}
/**
* synchronize the state of the thread with ServerConfig.
*/
@DebugLog
public Single<Boolean> keepAlive() {
return checkIfConnectionAlive()
.flatMap(alive -> alive ? Single.just(true) : connectWithExponentialBackoff());
}
@DebugLog
private Single<Boolean> checkIfConnectionAlive() {
if (DDPClient.get() == null) {
return Single.just(false);
} }
if (!keepAliveTimer.shouldCheckPrecisely()) { /**
return Single.just(true); * synchronize the state of the thread with ServerConfig.
*/
@DebugLog
public Single<Boolean> keepAlive() {
return checkIfConnectionAlive()
.flatMap(alive -> alive ? Single.just(true) : connectWithExponentialBackoff());
} }
keepAliveTimer.update();
@DebugLog
return Single.fromPublisher(emitter -> { private Single<Boolean> checkIfConnectionAlive() {
new Thread() { if (DDPClient.get() == null) {
@Override return Single.just(false);
public void run() {
DDPClient.get().ping().continueWith(task -> {
if (task.isFaulted()) {
Exception error = task.getError();
RCLog.e(error);
connectivityManager.notifyConnectionLost(
hostname, DDPClient.REASON_CLOSED_BY_USER);
emitter.onError(error);
} else {
keepAliveTimer.update();
emitter.onNext(true);
emitter.onComplete();
}
return null;
});
} }
}.start();
});
}
@DebugLog
private Flowable<Boolean> heartbeat(long interval) {
return Flowable.interval(interval, TimeUnit.MILLISECONDS)
.onBackpressureDrop()
.flatMap(tick -> DDPClient.get().doPing().toFlowable())
.map(callback -> {
if (callback instanceof DDPClientCallback.Ping) {
return true;
}
// ideally we should never get here. We should always receive a DDPClientCallback.Ping
// because we just received a pong. But maybe we received a pong from an unmatched
// ping id which we should ignore. In this case or any other random error, log and
// send false downstream
RCLog.d("heartbeat pong < %s", callback.toString());
return false;
});
}
private Single<Boolean> prepareDDPClient() {
// TODO: temporarily replaced checkIfConnectionAlive() call for this single checking if ddpClient is
// null or not. In case it is, build a new client, otherwise just keep connecting with existing one.
return Single.just(DDPClient.get() != null)
.doOnSuccess(alive -> {
if (!alive) {
RCLog.d("DDPClient#build");
}
});
}
private Single<Boolean> connectDDPClient() {
return prepareDDPClient()
.flatMap(_val -> Single.fromPublisher(emitter -> {
ServerInfo info = connectivityManager.getServerInfoForHost(hostname);
if (info == null) {
emitter.onNext(false);
emitter.onComplete();
return;
}
RCLog.d("DDPClient#connect");
DDPClient.get().connect(hostname, info.getSession(), info.isSecure())
.onSuccessTask(task -> {
final String newSession = task.getResult().session;
connectivityManager.notifyConnectionEstablished(hostname, newSession);
// handling WebSocket#onClose() callback.
task.getResult().client.getOnCloseCallback().onSuccess(_task -> {
RxWebSocketCallback.Close result = _task.getResult();
if (result.code == DDPClient.REASON_NETWORK_ERROR) {
reconnect();
}
return null;
});
return realmHelper.executeTransaction(realm -> { if (!keepAliveTimer.shouldCheckPrecisely()) {
RealmSession sessionObj = RealmSession.queryDefaultSession(realm).findFirst(); return Single.just(true);
if (sessionObj == null) { }
realm.createOrUpdateObjectFromJson(RealmSession.class, keepAliveTimer.update();
new JSONObject().put(RealmSession.ID, RealmSession.DEFAULT_ID));
} else { return Single.create(emitter -> {
// invalidate login token. new Thread() {
if (!TextUtils.isEmpty(sessionObj.getToken()) && sessionObj.isTokenVerified()) { @Override
sessionObj.setTokenVerified(false); public void run() {
sessionObj.setError(null); DDPClient.get().ping().continueWith(task -> {
if (task.isFaulted()) {
Exception error = task.getError();
RCLog.e(error);
connectivityManager.notifyConnectionLost(
hostname, DDPClient.REASON_CLOSED_BY_USER);
emitter.onError(error);
} else {
keepAliveTimer.update();
emitter.onSuccess(true);
} }
return null;
}
return null;
}); });
}) }
.continueWith(task -> { }.start();
if (task.isFaulted()) { });
emitter.onError(task.getError());
} else {
emitter.onNext(true);
emitter.onComplete();
}
return null;
});
}));
}
private void reconnect() {
// if we are already trying to reconnect then return.
if (reconnectSubscription.size() > 0) {
return;
} }
forceInvalidateTokens();
connectivityManager.notifyConnecting(hostname); @DebugLog
reconnectSubscription.add( private Flowable<Boolean> heartbeat(long interval) {
connectWithExponentialBackoff() return Flowable.interval(interval, TimeUnit.MILLISECONDS)
.subscribe( .onBackpressureDrop()
connected -> { .flatMap(tick -> DDPClient.get().doPing().toFlowable())
if (!connected) { .map(callback -> {
connectivityManager.notifyConnecting(hostname); if (callback instanceof DDPClientCallback.Ping) {
return true;
} }
reconnectSubscription.clear(); // ideally we should never get here. We should always receive a DDPClientCallback.Ping
}, // because we just received a pong. But maybe we received a pong from an unmatched
error -> { // ping id which we should ignore. In this case or any other random error, log and
logErrorAndUnsubscribe(reconnectSubscription, error); // send false downstream
connectivityManager.notifyConnectionLost(hostname, RCLog.d("heartbeat pong < %s", callback.toString());
DDPClient.REASON_NETWORK_ERROR); return false;
} });
)
);
}
private void logErrorAndUnsubscribe(CompositeDisposable disposables, Throwable err) {
RCLog.e(err);
disposables.clear();
}
private Single<Boolean> connectWithExponentialBackoff() {
return connect().retryWhen(RxHelper.exponentialBackoff(3, 500, TimeUnit.MILLISECONDS));
}
@DebugLog
private Single<Boolean> connect() {
return connectDDPClient()
.flatMap(_val -> Single.fromPublisher(emitter -> {
fetchPublicSettings();
fetchPermissions();
registerListeners();
emitter.onNext(true);
emitter.onComplete();
}));
}
private Task<Void> fetchPublicSettings() {
return new MethodCallHelper(appContext, realmHelper).getPublicSettings(hostname);
}
private Task<Void> fetchPermissions() {
return new MethodCallHelper(realmHelper).getPermissions();
}
@DebugLog
private void registerListeners() {
if (!Thread.currentThread().getName().equals("RC_thread_" + hostname)) {
// execute in Looper.
new Handler(getLooper()).post(this::registerListeners);
return;
} }
if (listenersRegistered) { private Single<Boolean> connectDDPClient() {
unregisterListeners(); return Single.create(emitter -> {
ServerInfo info = connectivityManager.getServerInfoForHost(hostname);
if (info == null) {
emitter.onSuccess(false);
return;
}
RCLog.d("DDPClient#connect");
DDPClient.get().connect(hostname, info.getSession(), info.isSecure())
.onSuccessTask(task -> {
final String newSession = task.getResult().session;
connectivityManager.notifyConnectionEstablished(hostname, newSession);
// handling WebSocket#onClose() callback.
task.getResult().client.getOnCloseCallback().onSuccess(_task -> {
RxWebSocketCallback.Close result = _task.getResult();
if (result.code == DDPClient.REASON_NETWORK_ERROR) {
reconnect();
}
return null;
});
return realmHelper.executeTransaction(realm -> {
RealmSession sessionObj = RealmSession.queryDefaultSession(realm).findFirst();
if (sessionObj == null) {
realm.createOrUpdateObjectFromJson(RealmSession.class,
new JSONObject().put(RealmSession.ID, RealmSession.DEFAULT_ID));
} else {
// invalidate login token.
if (!TextUtils.isEmpty(sessionObj.getToken()) && sessionObj.isTokenVerified()) {
sessionObj.setTokenVerified(false);
sessionObj.setError(null);
}
}
return null;
});
})
.continueWith(task -> {
if (task.isFaulted()) {
emitter.onError(task.getError());
} else {
emitter.onSuccess(true);
}
return null;
});
});
} }
List<RealmSession> sessions = realmHelper.executeTransactionForReadResults(realm -> private void reconnect() {
realm.where(RealmSession.class) // if we are already trying to reconnect then return.
.isNotNull(RealmSession.TOKEN) if (reconnectDisposable.size() > 0) {
.equalTo(RealmSession.TOKEN_VERIFIED, false) return;
.isNull(RealmSession.ERROR) }
.findAll()); forceInvalidateTokens();
connectivityManager.notifyConnecting(hostname);
if (sessions != null && sessions.size() > 0) { reconnectDisposable.add(
// if we have a session try to resume it. At this point we're probably recovering from connectWithExponentialBackoff()
// a disconnection state .subscribe(connected -> {
final CompositeDisposable disposables = new CompositeDisposable(); if (!connected) {
MethodCallHelper methodCall = new MethodCallHelper(realmHelper); connectivityManager.notifyConnecting(hostname);
disposables.add( }
Completable.defer(() -> { reconnectDisposable.clear();
Task<Void> result = methodCall.loginWithToken(sessions.get(0).getToken()); }, error -> {
if (result.isFaulted()) { logErrorAndUnsubscribe(reconnectDisposable, error);
return Completable.error(result.getError()); connectivityManager.notifyConnectionLost(hostname,
} else { DDPClient.REASON_NETWORK_ERROR);
return Completable.complete(); }
} )
}).retryWhen(RxHelper.exponentialBackoff(Integer.MAX_VALUE, 500, TimeUnit.MILLISECONDS)) );
.subscribe( }
() -> {
createObserversAndRegister(); private void logErrorAndUnsubscribe(CompositeDisposable disposables, Throwable err) {
disposables.clear(); RCLog.e(err);
}, disposables.clear();
error -> logErrorAndUnsubscribe(disposables, error) }
)
); private Single<Boolean> connectWithExponentialBackoff() {
} else { return connect().retryWhen(RxHelper.exponentialBackoff(3, 500, TimeUnit.MILLISECONDS));
// if we don't have any session then just build the observers and register normally }
createObserversAndRegister();
@DebugLog
private Single<Boolean> connect() {
return connectDDPClient()
.flatMap(_val -> Single.create(emitter -> {
fetchPublicSettings();
fetchPermissions();
registerListeners();
emitter.onSuccess(true);
}));
}
private Task<Void> fetchPublicSettings() {
return new MethodCallHelper(appContext, realmHelper).getPublicSettings(hostname);
}
private Task<Void> fetchPermissions() {
return new MethodCallHelper(realmHelper).getPermissions();
}
@DebugLog
private void registerListeners() {
if (!Thread.currentThread().getName().equals("RC_thread_" + hostname)) {
// execute in Looper.
new Handler(getLooper()).post(this::registerListeners);
return;
}
if (listenersRegistered) {
unregisterListeners();
}
List<RealmSession> sessions = realmHelper.executeTransactionForReadResults(realm ->
realm.where(RealmSession.class)
.isNotNull(RealmSession.TOKEN)
.equalTo(RealmSession.TOKEN_VERIFIED, false)
.isNull(RealmSession.ERROR)
.findAll());
if (sessions != null && sessions.size() > 0) {
// if we have a session try to resume it. At this point we're probably recovering from
// a disconnection state
final CompositeDisposable disposables = new CompositeDisposable();
MethodCallHelper methodCall = new MethodCallHelper(realmHelper);
disposables.add(
Completable.defer(() -> {
Task<Void> result = methodCall.loginWithToken(sessions.get(0).getToken());
if (result.isFaulted()) {
return Completable.error(result.getError());
} else {
return Completable.complete();
}
}).retryWhen(RxHelper.exponentialBackoff(3, 500, TimeUnit.MILLISECONDS))
.subscribe(
() -> {
createObserversAndRegister();
disposables.clear();
},
error -> logErrorAndUnsubscribe(disposables, error)
)
);
} else {
// if we don't have any session then just build the observers and register normally
createObserversAndRegister();
}
} }
}
@DebugLog
@DebugLog private void createObserversAndRegister() {
private void createObserversAndRegister() { for (Class clazz : REGISTERABLE_CLASSES) {
for (Class clazz : REGISTERABLE_CLASSES) { try {
try { Constructor ctor = clazz.getConstructor(Context.class, String.class, RealmHelper.class);
Constructor ctor = clazz.getConstructor(Context.class, String.class, RealmHelper.class); Object obj = ctor.newInstance(appContext, hostname, realmHelper);
Object obj = ctor.newInstance(appContext, hostname, realmHelper);
if (obj instanceof Registrable) {
if (obj instanceof Registrable) { Registrable registrable = (Registrable) obj;
Registrable registrable = (Registrable) obj; registrable.register();
registrable.register(); listeners.add(registrable);
listeners.add(registrable); }
} catch (Exception exception) {
RCLog.w(exception, "Failed to register listeners!!");
}
} }
} catch (Exception exception) { listenersRegistered = true;
RCLog.w(exception, "Failed to register listeners!!"); startHeartBeat();
}
} }
listenersRegistered = true;
startHeartBeat(); private void startHeartBeat() {
} heartbeatDisposable.clear();
heartbeatDisposable.add(
private void startHeartBeat() { heartbeat(HEARTBEAT_PERIOD_MS)
hearbeatDisposable.clear(); .subscribe(
hearbeatDisposable.add( ponged -> {
heartbeat(HEARTBEAT_PERIOD_MS) if (!ponged) {
.subscribe( RCLog.d("Pong received but didn't match ping id");
ponged -> { }
if (!ponged) { },
RCLog.d("Pong received but didn't match ping id"); error -> {
} RCLog.e(error);
}, // Stop pinging
error -> { heartbeatDisposable.clear();
RCLog.e(error); if (error instanceof DDPClientCallback.Closed || error instanceof TimeoutException) {
// Stop pinging RCLog.d("Hearbeat failure: retrying connection...");
hearbeatDisposable.clear(); reconnect();
if (error instanceof DDPClientCallback.Closed || error instanceof TimeoutException) { }
RCLog.d("Hearbeat failure: retrying connection..."); }
reconnect(); )
} );
} }
)
); @DebugLog
} private void unregisterListenersAndClose() {
unregisterListeners();
@DebugLog DDPClient.get().close();
private void unregisterListenersAndClose() { }
unregisterListeners();
DDPClient.get().close(); @DebugLog
} private void unregisterListeners() {
Iterator<Registrable> iterator = listeners.iterator();
@DebugLog while (iterator.hasNext()) {
private void unregisterListeners() { Registrable registrable = iterator.next();
Iterator<Registrable> iterator = listeners.iterator(); registrable.unregister();
while (iterator.hasNext()) { iterator.remove();
Registrable registrable = iterator.next(); }
registrable.unregister(); heartbeatDisposable.clear();
iterator.remove(); listenersRegistered = false;
} }
hearbeatDisposable.clear();
listenersRegistered = false;
}
} }
...@@ -15,9 +15,9 @@ buildscript { ...@@ -15,9 +15,9 @@ buildscript {
maven { url 'http://oss.jfrog.org/artifactory/oss-snapshot-local' } maven { url 'http://oss.jfrog.org/artifactory/oss-snapshot-local' }
} }
dependencies { dependencies {
classpath 'com.android.tools.build:gradle:3.0.0' classpath 'com.android.tools.build:gradle:3.0.1'
classpath "org.jetbrains.kotlin:kotlin-gradle-plugin:1.1.51" classpath "org.jetbrains.kotlin:kotlin-gradle-plugin:1.1.60"
classpath 'io.realm:realm-gradle-plugin:4.2.0-SNAPSHOT' classpath 'io.realm:realm-gradle-plugin:4.2.0'
classpath 'com.jakewharton.hugo:hugo-plugin:1.2.1' classpath 'com.jakewharton.hugo:hugo-plugin:1.2.1'
classpath 'com.google.gms:google-services:3.0.0' classpath 'com.google.gms:google-services:3.0.0'
classpath 'com.github.triplet.gradle:play-publisher:1.1.5' classpath 'com.github.triplet.gradle:play-publisher:1.1.5'
......
...@@ -6,6 +6,7 @@ import chat.rocket.persistence.realm.models.ddp.RealmRole; ...@@ -6,6 +6,7 @@ import chat.rocket.persistence.realm.models.ddp.RealmRole;
import chat.rocket.persistence.realm.models.ddp.RealmRoomRole; import chat.rocket.persistence.realm.models.ddp.RealmRoomRole;
import chat.rocket.persistence.realm.models.ddp.RealmSpotlightRoom; import chat.rocket.persistence.realm.models.ddp.RealmSpotlightRoom;
import chat.rocket.persistence.realm.models.ddp.RealmSpotlightUser; import chat.rocket.persistence.realm.models.ddp.RealmSpotlightUser;
import chat.rocket.persistence.realm.models.ddp.RealmUser;
import io.realm.DynamicRealm; import io.realm.DynamicRealm;
import io.realm.FieldAttribute; import io.realm.FieldAttribute;
import io.realm.RealmMigration; import io.realm.RealmMigration;
...@@ -67,6 +68,13 @@ public class Migration implements RealmMigration { ...@@ -67,6 +68,13 @@ public class Migration implements RealmMigration {
if (oldVersion == 4) { if (oldVersion == 4) {
RealmObjectSchema messageSchema = schema.get("RealmMessage"); RealmObjectSchema messageSchema = schema.get("RealmMessage");
messageSchema.addField(RealmMessage.EDITED_AT, long.class); messageSchema.addField(RealmMessage.EDITED_AT, long.class);
oldVersion++;
}
if (oldVersion == 5) {
RealmObjectSchema userSchema = schema.get("RealmUser");
userSchema.addField(RealmUser.NAME, String.class);
} }
} }
......
...@@ -15,9 +15,7 @@ public class RealmStore { ...@@ -15,9 +15,7 @@ public class RealmStore {
.name(name + ".realm") .name(name + ".realm")
.modules(new RocketChatLibraryModule()) .modules(new RocketChatLibraryModule())
.migration(new Migration()) .migration(new Migration())
.schemaVersion(5) .schemaVersion(6)
// Just in case
.deleteRealmIfMigrationNeeded()
.build(); .build();
} }
...@@ -49,7 +47,7 @@ public class RealmStore { ...@@ -49,7 +47,7 @@ public class RealmStore {
sStore.put(name, new RealmConfiguration.Builder() sStore.put(name, new RealmConfiguration.Builder()
.name(name + ".realm") .name(name + ".realm")
.modules(new RocketChatServerModule()) .modules(new RocketChatServerModule())
.deleteRealmIfMigrationNeeded().build()); .build());
} }
return new RealmHelper(sStore.get(name)); return new RealmHelper(sStore.get(name));
} }
......
...@@ -41,8 +41,7 @@ public class RealmMessageRepository extends RealmRepository implements MessageRe ...@@ -41,8 +41,7 @@ public class RealmMessageRepository extends RealmRepository implements MessageRe
return Flowable.empty(); return Flowable.empty();
} }
return return pair.first.where(RealmMessage.class)
pair.first.where(RealmMessage.class)
.equalTo(RealmMessage.ID, messageId) .equalTo(RealmMessage.ID, messageId)
.findAll() .findAll()
.<RealmResults<RealmMessage>>asFlowable(); .<RealmResults<RealmMessage>>asFlowable();
......
...@@ -7,6 +7,7 @@ import com.hadisatrio.optional.Optional; ...@@ -7,6 +7,7 @@ import com.hadisatrio.optional.Optional;
import chat.rocket.core.models.Session; import chat.rocket.core.models.Session;
import chat.rocket.core.repositories.SessionRepository; import chat.rocket.core.repositories.SessionRepository;
import chat.rocket.persistence.realm.RealmHelper;
import chat.rocket.persistence.realm.RealmStore; import chat.rocket.persistence.realm.RealmStore;
import chat.rocket.persistence.realm.models.internal.RealmSession; import chat.rocket.persistence.realm.models.internal.RealmSession;
import io.reactivex.Flowable; import io.reactivex.Flowable;
...@@ -44,7 +45,7 @@ public class RealmSessionRepository extends RealmRepository implements SessionRe ...@@ -44,7 +45,7 @@ public class RealmSessionRepository extends RealmRepository implements SessionRe
if (realmSessions.size() == 0) { if (realmSessions.size() == 0) {
return Optional.absent(); return Optional.absent();
} }
return Optional.of(realmSessions.get(0).asSession()); return Optional.of(realmSessions.get(0).asSession());
})); }));
} }
...@@ -73,14 +74,9 @@ public class RealmSessionRepository extends RealmRepository implements SessionRe ...@@ -73,14 +74,9 @@ public class RealmSessionRepository extends RealmRepository implements SessionRe
realmSession.setTokenVerified(session.isTokenVerified()); realmSession.setTokenVerified(session.isTokenVerified());
realmSession.setError(session.getError()); realmSession.setError(session.getError());
realm.beginTransaction(); return RealmHelper.copyToRealmOrUpdate(realm, realmSession)
return realm.copyToRealmOrUpdate(realmSession)
.asFlowable()
.filter(it -> it != null && it.isLoaded() && it.isValid()) .filter(it -> it != null && it.isLoaded() && it.isValid())
.firstElement() .firstElement()
.doOnSuccess(it -> realm.commitTransaction())
.doOnError(throwable -> realm.cancelTransaction())
.doOnEvent((realmObject, throwable) -> close(realm, looper)) .doOnEvent((realmObject, throwable) -> close(realm, looper))
.toSingle() .toSingle()
.map(realmObject -> true); .map(realmObject -> true);
......
...@@ -24,8 +24,8 @@ class RealmSpotlightRepository(private val hostname: String) : RealmRepository() ...@@ -24,8 +24,8 @@ class RealmSpotlightRepository(private val hostname: String) : RealmRepository()
} }
return@using pair.first.where(RealmSpotlight::class.java) return@using pair.first.where(RealmSpotlight::class.java)
.findAllSorted(Columns.TYPE, Sort.DESCENDING) .findAllSorted(Columns.TYPE, Sort.DESCENDING)
.asFlowable() .asFlowable()
}) { pair -> close(pair.first, pair.second) } }) { pair -> close(pair.first, pair.second) }
.unsubscribeOn(AndroidSchedulers.from(Looper.myLooper()!!)) .unsubscribeOn(AndroidSchedulers.from(Looper.myLooper()!!))
.filter { realmSpotlightResults -> realmSpotlightResults.isLoaded && realmSpotlightResults.isValid } .filter { realmSpotlightResults -> realmSpotlightResults.isLoaded && realmSpotlightResults.isValid }
......
...@@ -6,7 +6,7 @@ dependencies { ...@@ -6,7 +6,7 @@ dependencies {
compile extraDependencies.rxJava compile extraDependencies.rxJava
compile extraDependencies.optional compile extraDependencies.optional
compile "org.jetbrains.kotlin:kotlin-stdlib-jre7:$rootProject.ext.kotlinVersion" compile "org.jetbrains.kotlin:kotlin-stdlib-jre7:$rootProject.ext.kotlinVersion"
compile 'com.google.code.findbugs:jsr305:3.0.1' compile 'com.google.code.findbugs:jsr305:3.0.2'
compileOnly 'com.google.auto.value:auto-value:1.3' compileOnly 'com.google.auto.value:auto-value:1.3'
kapt 'com.google.auto.value:auto-value:1.3' kapt 'com.google.auto.value:auto-value:1.3'
kapt 'com.gabrielittner.auto.value:auto-value-with:1.0.0' kapt 'com.gabrielittner.auto.value:auto-value-with:1.0.0'
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment