MirrorMirror
/
Signal-Server
mirror of https://github.com/signalapp/Signal-Server.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Reorganize account manager timers

This commit is contained in:
Moxie Marlinspike 2018-11-01 03:16:29 -07:00
parent dc691daf54
commit 54a41b4f0a
1 changed files with 61 additions and 71 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

132
src/main/java/org/whispersystems/textsecuregcm/storage/AccountsManager.java
View File

@ -54,10 +54,6 @@ public class AccountsManager {
private static final Timer redisSetTimer = metricRegistry.timer(name(AccountsManager.class, "redisSet" )); private static final Timer redisSetTimer = metricRegistry.timer(name(AccountsManager.class, "redisSet" ));
private static final Timer redisGetTimer = metricRegistry.timer(name(AccountsManager.class, "redisGet" )); private static final Timer redisGetTimer = metricRegistry.timer(name(AccountsManager.class, "redisGet" ));
private static final Timer databaseCreateTimer = metricRegistry.timer(name(AccountsManager.class, "databaseCreate"));
private static final Timer databaseGetTimer = metricRegistry.timer(name(AccountsManager.class, "databaseGet" ));
private static final Timer databaseUpdateTimer = metricRegistry.timer(name(AccountsManager.class, "databaseUpdate"));
private final Logger logger = LoggerFactory.getLogger(AccountsManager.class); private final Logger logger = LoggerFactory.getLogger(AccountsManager.class);
private final Accounts accounts; private final Accounts accounts;
@ -139,92 +135,86 @@ public class AccountsManager {
} }
private void redisSet(String number, Account account, boolean optional) { private void redisSet(String number, Account account, boolean optional) {
try (Timer.Context context = redisSetTimer.time()) { new HystrixCommand<Boolean>(Setter.withGroupKey(HystrixCommandGroupKey.Factory.asKey(GroupKeys.REDIS_CACHE))
new HystrixCommand<Boolean>(Setter.withGroupKey(HystrixCommandGroupKey.Factory.asKey(GroupKeys.REDIS_CACHE)) .andCommandKey(HystrixCommandKey.Factory.asKey(AccountsManager.class.getSimpleName() + ".redisSet"))) {
.andCommandKey(HystrixCommandKey.Factory.asKey(AccountsManager.class.getSimpleName() + ".redisSet"))) { @Override
@Override protected Boolean run() {
protected Boolean run() { try (Jedis jedis = cacheClient.getWriteResource();
try (Jedis jedis = cacheClient.getWriteResource()) { Timer.Context timer = redisSetTimer.time())
jedis.set(getKey(number), mapper.writeValueAsString(account)); {
} catch (JsonProcessingException e) { jedis.set(getKey(number), mapper.writeValueAsString(account));
throw new HystrixBadRequestException("Json processing error", e); } catch (JsonProcessingException e) {
} throw new HystrixBadRequestException("Json processing error", e);
return true;
} }
@Override return true;
protected Boolean getFallback() { }
if (optional) return true;
else return super.getFallback(); @Override
} protected Boolean getFallback() {
}.execute(); if (optional) return true;
} else return super.getFallback();
}
}.execute();
} }
private Optional<Account> redisGet(String number) { private Optional<Account> redisGet(String number) {
try (Timer.Context context = redisGetTimer.time()) { return new HystrixCommand<Optional<Account>>(Setter.withGroupKey(HystrixCommandGroupKey.Factory.asKey(GroupKeys.REDIS_CACHE))
return new HystrixCommand<Optional<Account>>(Setter.withGroupKey(HystrixCommandGroupKey.Factory.asKey(GroupKeys.REDIS_CACHE)) .andCommandKey(HystrixCommandKey.Factory.asKey(AccountsManager.class.getSimpleName() + ".redisGet")))
.andCommandKey(HystrixCommandKey.Factory.asKey(AccountsManager.class.getSimpleName() + ".redisGet"))) {
{ @Override
@Override protected Optional<Account> run() {
protected Optional<Account> run() { try (Jedis jedis = cacheClient.getReadResource();
try (Jedis jedis = cacheClient.getReadResource()) { Timer.Context timer = redisGetTimer.time())
String json = jedis.get(getKey(number)); {
String json = jedis.get(getKey(number));
if (json != null) return Optional.of(mapper.readValue(json, Account.class));
else return Optional.empty(); if (json != null) return Optional.of(mapper.readValue(json, Account.class));
} catch (IOException e) { else return Optional.empty();
logger.warn("AccountsManager", "Deserialization error", e); } catch (IOException e) {
return Optional.empty(); logger.warn("AccountsManager", "Deserialization error", e);
}
}
@Override
protected Optional<Account> getFallback() {
return Optional.empty(); return Optional.empty();
} }
}.execute(); }
}
@Override
protected Optional<Account> getFallback() {
return Optional.empty();
}
}.execute();
} }
private Optional<Account> databaseGet(String number) { private Optional<Account> databaseGet(String number) {
try (Timer.Context context = databaseGetTimer.time()) { return new HystrixCommand<Optional<Account>>(Setter.withGroupKey(HystrixCommandGroupKey.Factory.asKey(GroupKeys.DATABASE_ACCOUNTS))
return new HystrixCommand<Optional<Account>>(Setter.withGroupKey(HystrixCommandGroupKey.Factory.asKey(GroupKeys.DATABASE_ACCOUNTS)) .andCommandKey(HystrixCommandKey.Factory.asKey(AccountsManager.class.getSimpleName() + ".databaseGet")))
.andCommandKey(HystrixCommandKey.Factory.asKey(AccountsManager.class.getSimpleName() + ".databaseGet"))) {
{ @Override
@Override protected Optional<Account> run() {
protected Optional<Account> run() {
return Optional.ofNullable(accounts.get(number)); return Optional.ofNullable(accounts.get(number));
} }
}.execute(); }.execute();
}
} }
private boolean databaseCreate(Account account) { private boolean databaseCreate(Account account) {
try (Timer.Context context = databaseCreateTimer.time()) { return new HystrixCommand<Boolean>(Setter.withGroupKey(HystrixCommandGroupKey.Factory.asKey(GroupKeys.DATABASE_ACCOUNTS))
return new HystrixCommand<Boolean>(Setter.withGroupKey(HystrixCommandGroupKey.Factory.asKey(GroupKeys.DATABASE_ACCOUNTS)) .andCommandKey(HystrixCommandKey.Factory.asKey(AccountsManager.class.getSimpleName() + ".databaseCreate")))
.andCommandKey(HystrixCommandKey.Factory.asKey(AccountsManager.class.getSimpleName() + ".databaseCreate"))) {
{ @Override
@Override protected Boolean run() {
protected Boolean run() {
return accounts.create(account); return accounts.create(account);
} }
}.execute(); }.execute();
}
} }
private void databaseUpdate(Account account) { private void databaseUpdate(Account account) {
try (Timer.Context context = databaseUpdateTimer.time()) { new HystrixCommand<Void>(Setter.withGroupKey(HystrixCommandGroupKey.Factory.asKey(GroupKeys.DATABASE_ACCOUNTS))
new HystrixCommand<Void>(Setter.withGroupKey(HystrixCommandGroupKey.Factory.asKey(GroupKeys.DATABASE_ACCOUNTS)) .andCommandKey(HystrixCommandKey.Factory.asKey(AccountsManager.class.getSimpleName() + ".databaseUpdate")))
.andCommandKey(HystrixCommandKey.Factory.asKey(AccountsManager.class.getSimpleName() + ".databaseUpdate"))) {
{ @Override
@Override protected Void run() {
protected Void run() { accounts.update(account);
accounts.update(account); return null;
return null; }
} }.execute();
}.execute();
}
} }
} }