Move directory refresh from a service over to a job.

// FREEBIE
This commit is contained in:
Moxie Marlinspike 2014-11-27 15:24:26 -08:00
parent 0fb7f3b970
commit 0d01feebc3
6 changed files with 76 additions and 86 deletions

View File

@ -22,6 +22,8 @@ import org.whispersystems.libaxolotl.state.SignedPreKeyRecord;
import org.whispersystems.libaxolotl.util.guava.Optional; import org.whispersystems.libaxolotl.util.guava.Optional;
import org.whispersystems.textsecure.api.push.ContactTokenDetails; import org.whispersystems.textsecure.api.push.ContactTokenDetails;
import org.whispersystems.textsecure.api.push.TrustStore; import org.whispersystems.textsecure.api.push.TrustStore;
import org.whispersystems.textsecure.api.push.exceptions.NonSuccessfulResponseCodeException;
import org.whispersystems.textsecure.api.push.exceptions.PushNetworkException;
import org.whispersystems.textsecure.internal.push.PushServiceSocket; import org.whispersystems.textsecure.internal.push.PushServiceSocket;
import org.whispersystems.textsecure.api.push.SignedPreKeyEntity; import org.whispersystems.textsecure.api.push.SignedPreKeyEntity;
@ -86,7 +88,9 @@ public class TextSecureAccountManager {
return Optional.fromNullable(this.pushServiceSocket.getContactTokenDetails(contactToken)); return Optional.fromNullable(this.pushServiceSocket.getContactTokenDetails(contactToken));
} }
public List<ContactTokenDetails> getContacts(Set<String> contactTokens) { public List<ContactTokenDetails> getContacts(Set<String> contactTokens)
throws IOException
{
return this.pushServiceSocket.retrieveDirectory(contactTokens); return this.pushServiceSocket.retrieveDirectory(contactTokens);
} }

View File

@ -321,17 +321,14 @@ public class PushServiceSocket {
downloadExternalFile(descriptor.getLocation(), destination); downloadExternalFile(descriptor.getLocation(), destination);
} }
public List<ContactTokenDetails> retrieveDirectory(Set<String> contactTokens) { public List<ContactTokenDetails> retrieveDirectory(Set<String> contactTokens)
try { throws NonSuccessfulResponseCodeException, PushNetworkException
ContactTokenList contactTokenList = new ContactTokenList(new LinkedList<String>(contactTokens)); {
String response = makeRequest(DIRECTORY_TOKENS_PATH, "PUT", new Gson().toJson(contactTokenList)); ContactTokenList contactTokenList = new ContactTokenList(new LinkedList<>(contactTokens));
ContactTokenDetailsList activeTokens = new Gson().fromJson(response, ContactTokenDetailsList.class); String response = makeRequest(DIRECTORY_TOKENS_PATH, "PUT", new Gson().toJson(contactTokenList));
ContactTokenDetailsList activeTokens = new Gson().fromJson(response, ContactTokenDetailsList.class);
return activeTokens.getContacts(); return activeTokens.getContacts();
} catch (IOException ioe) {
Log.w("PushServiceSocket", ioe);
return null;
}
} }
public ContactTokenDetails getContactTokenDetails(String contactToken) throws IOException { public ContactTokenDetails getContactTokenDetails(String contactToken) throws IOException {

View File

@ -0,0 +1,46 @@
package org.thoughtcrime.securesms.jobs;
import android.content.Context;
import android.os.PowerManager;
import org.thoughtcrime.securesms.util.DirectoryHelper;
import org.whispersystems.jobqueue.JobParameters;
import org.whispersystems.jobqueue.requirements.NetworkRequirement;
import org.whispersystems.textsecure.api.push.exceptions.PushNetworkException;
import java.io.IOException;
public class DirectoryRefreshJob extends ContextJob {
public DirectoryRefreshJob(Context context) {
super(context, JobParameters.newBuilder()
.withGroupId(DirectoryRefreshJob.class.getSimpleName())
.withRequirement(new NetworkRequirement(context))
.create());
}
@Override
public void onAdded() {}
@Override
public void onRun() throws IOException {
PowerManager powerManager = (PowerManager) context.getSystemService(Context.POWER_SERVICE);
PowerManager.WakeLock wakeLock = powerManager.newWakeLock(PowerManager.PARTIAL_WAKE_LOCK, "Directory Refresh");
try {
wakeLock.acquire();
DirectoryHelper.refreshDirectory(context);
} finally {
if (wakeLock.isHeld()) wakeLock.release();
}
}
@Override
public boolean onShouldRetry(Exception exception) {
if (exception instanceof PushNetworkException) return true;
return false;
}
@Override
public void onCanceled() {}
}

View File

@ -8,6 +8,8 @@ import android.content.Context;
import android.content.Intent; import android.content.Intent;
import android.util.Log; import android.util.Log;
import org.thoughtcrime.securesms.ApplicationContext;
import org.thoughtcrime.securesms.jobs.DirectoryRefreshJob;
import org.thoughtcrime.securesms.util.TextSecurePreferences; import org.thoughtcrime.securesms.util.TextSecurePreferences;
public class DirectoryRefreshListener extends BroadcastReceiver { public class DirectoryRefreshListener extends BroadcastReceiver {
@ -41,9 +43,9 @@ public class DirectoryRefreshListener extends BroadcastReceiver {
if (time <= System.currentTimeMillis()) { if (time <= System.currentTimeMillis()) {
if (time != 0) { if (time != 0) {
Intent serviceIntent = new Intent(context, DirectoryRefreshService.class); ApplicationContext.getInstance(context)
serviceIntent.setAction(DirectoryRefreshService.REFRESH_ACTION); .getJobManager()
context.startService(serviceIntent); .add(new DirectoryRefreshJob(context));
} }
time = System.currentTimeMillis() + INTERVAL; time = System.currentTimeMillis() + INTERVAL;

View File

@ -1,64 +0,0 @@
package org.thoughtcrime.securesms.service;
import android.app.Service;
import android.content.Context;
import android.content.Intent;
import android.os.IBinder;
import android.os.PowerManager;
import android.util.Log;
import org.thoughtcrime.securesms.util.DirectoryHelper;
import java.util.concurrent.Executor;
import java.util.concurrent.Executors;
public class DirectoryRefreshService extends Service {
public static final String REFRESH_ACTION = "org.whispersystems.whisperpush.REFRESH_ACTION";
private static final Executor executor = Executors.newSingleThreadExecutor();
@Override
public int onStartCommand (Intent intent, int flags, int startId) {
if (REFRESH_ACTION.equals(intent.getAction())) {
handleRefreshAction();
}
return START_NOT_STICKY;
}
@Override
public IBinder onBind(Intent intent) {
return null;
}
private void handleRefreshAction() {
PowerManager pm = (PowerManager) getSystemService(Context.POWER_SERVICE);
PowerManager.WakeLock wakeLock = pm.newWakeLock(PowerManager.PARTIAL_WAKE_LOCK, "Directory Refresh");
wakeLock.acquire();
executor.execute(new RefreshRunnable(wakeLock));
}
private class RefreshRunnable implements Runnable {
private final PowerManager.WakeLock wakeLock;
private final Context context;
public RefreshRunnable(PowerManager.WakeLock wakeLock) {
this.wakeLock = wakeLock;
this.context = DirectoryRefreshService.this.getApplicationContext();
}
public void run() {
try {
Log.w("DirectoryRefreshService", "Refreshing directory...");
DirectoryHelper.refreshDirectory(context);
Log.w("DirectoryRefreshService", "Directory refresh complete...");
} finally {
if (wakeLock != null && wakeLock.isHeld())
wakeLock.release();
}
}
}
}

View File

@ -13,6 +13,7 @@ import org.whispersystems.textsecure.api.TextSecureAccountManager;
import org.whispersystems.textsecure.api.push.ContactTokenDetails; import org.whispersystems.textsecure.api.push.ContactTokenDetails;
import org.whispersystems.textsecure.api.util.InvalidNumberException; import org.whispersystems.textsecure.api.util.InvalidNumberException;
import java.io.IOException;
import java.util.List; import java.util.List;
import java.util.Map; import java.util.Map;
import java.util.Set; import java.util.Set;
@ -20,10 +21,6 @@ import java.util.Set;
public class DirectoryHelper { public class DirectoryHelper {
private static final String TAG = DirectoryHelper.class.getSimpleName(); private static final String TAG = DirectoryHelper.class.getSimpleName();
public static void refreshDirectoryWithProgressDialog(final Context context) {
refreshDirectoryWithProgressDialog(context, null);
}
public static void refreshDirectoryWithProgressDialog(final Context context, final DirectoryUpdateFinishedListener listener) { public static void refreshDirectoryWithProgressDialog(final Context context, final DirectoryUpdateFinishedListener listener) {
if (!TextSecurePreferences.isPushRegistered(context)) { if (!TextSecurePreferences.isPushRegistered(context)) {
Toast.makeText(context.getApplicationContext(), Toast.makeText(context.getApplicationContext(),
@ -38,7 +35,11 @@ public class DirectoryHelper {
{ {
@Override @Override
protected Void doInBackground(Void... voids) { protected Void doInBackground(Void... voids) {
DirectoryHelper.refreshDirectory(context.getApplicationContext()); try {
DirectoryHelper.refreshDirectory(context.getApplicationContext());
} catch (IOException e) {
Log.w(TAG, e);
}
return null; return null;
} }
@ -51,15 +52,19 @@ public class DirectoryHelper {
} }
public static void refreshDirectory(final Context context) { public static void refreshDirectory(final Context context) throws IOException {
refreshDirectory(context, TextSecureCommunicationFactory.createManager(context)); refreshDirectory(context, TextSecureCommunicationFactory.createManager(context));
} }
public static void refreshDirectory(final Context context, final TextSecureAccountManager accountManager) { public static void refreshDirectory(final Context context, final TextSecureAccountManager accountManager)
throws IOException
{
refreshDirectory(context, accountManager, TextSecurePreferences.getLocalNumber(context)); refreshDirectory(context, accountManager, TextSecurePreferences.getLocalNumber(context));
} }
public static void refreshDirectory(final Context context, final TextSecureAccountManager accountManager, final String localNumber) { public static void refreshDirectory(final Context context, final TextSecureAccountManager accountManager, final String localNumber)
throws IOException
{
TextSecureDirectory directory = TextSecureDirectory.getInstance(context); TextSecureDirectory directory = TextSecureDirectory.getInstance(context);
Set<String> eligibleContactNumbers = directory.getPushEligibleContactNumbers(localNumber); Set<String> eligibleContactNumbers = directory.getPushEligibleContactNumbers(localNumber);
Map<String, String> tokenMap = DirectoryUtil.getDirectoryServerTokenMap(eligibleContactNumbers); Map<String, String> tokenMap = DirectoryUtil.getDirectoryServerTokenMap(eligibleContactNumbers);