Signal/Signal-Server
1
0
Fork 0
mirror of https://github.com/signalapp/Signal-Server synced 2026-04-20 08:58:13 +01:00
Code Issues Packages Projects Releases Wiki Activity

Simplify optimistic write logic

Browse Source
This commit is contained in:
Jon Chambers
2021-08-03 11:54:26 -04:00
committed by GitHub
parent 23f9199439
commit bcb89924b4
2 changed files with 13 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -293,47 +293,21 @@ public class AccountsManager {
try (Timer.Context ignored = updateTimer.time()) {
if (!updater.apply(account)) {
return account;
}
{
// optimistically increment version
final int originalVersion = account.getVersion();
account.setVersion(originalVersion + 1);
redisSet(account);
account.setVersion(originalVersion);
}
redisDelete(account);
final UUID uuid = account.getUuid();
updatedAccount = updateWithRetries(account, updater, this::databaseUpdate, () -> databaseGet(uuid).get());
if (dynamoWriteEnabled()) {
runSafelyAndRecordMetrics(() -> {
final Optional<Account> dynamoAccount = dynamoGet(uuid);
if (dynamoAccount.isPresent()) {
if (!updater.apply(dynamoAccount.get())) {
return dynamoAccount;
}
Account dynamoUpdatedAccount = updateWithRetries(dynamoAccount.get(),
updater,
this::dynamoUpdate,
() -> dynamoGet(uuid).get());
return Optional.of(dynamoUpdatedAccount);
}
return Optional.empty();
}, Optional.of(uuid), Optional.of(updatedAccount),
runSafelyAndRecordMetrics(() -> dynamoGet(uuid).map(dynamoAccount ->
updateWithRetries(dynamoAccount, updater, this::dynamoUpdate, () -> dynamoGet(uuid).get())),
Optional.of(uuid),
Optional.of(updatedAccount),
this::compareAccounts,
"update");
}
// set the cache again, so that all updates are coalesced
redisSet(updatedAccount);
}
@@ -349,6 +323,10 @@ public class AccountsManager {
private Account updateWithRetries(Account account, Function<Account, Boolean> updater, Consumer<Account> persister,
Supplier<Account> retriever) {
if (!updater.apply(account)) {
return account;
}
final int maxTries = 10;
int tries = 0;