Add a manager class for checking "liveness" of client versions

This commit is contained in:
Jon Chambers
2023-07-25 17:43:00 -04:00
committed by Jon Chambers
parent 10689843b0
commit 6f4801fd6f
2 changed files with 137 additions and 0 deletions

View File

@@ -0,0 +1,80 @@
/*
* Copyright 2023 Signal Messenger, LLC
* SPDX-License-Identifier: AGPL-3.0-only
*/
package org.whispersystems.textsecuregcm.storage;
import com.vdurmont.semver4j.Semver;
import java.time.Clock;
import java.time.Duration;
import java.util.Collections;
import java.util.Map;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.ScheduledFuture;
import java.util.concurrent.TimeUnit;
import io.dropwizard.lifecycle.Managed;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.whispersystems.textsecuregcm.util.ua.ClientPlatform;
import javax.annotation.Nullable;
public class ClientReleaseManager implements Managed {
private final ClientReleases clientReleases;
private final ScheduledExecutorService scheduledExecutorService;
private final Duration refreshInterval;
private final Clock clock;
@Nullable
private ScheduledFuture<?> refreshClientReleasesFuture;
private volatile Map<ClientPlatform, Map<Semver, ClientRelease>> clientReleasesByPlatform = Collections.emptyMap();
private static final Logger logger = LoggerFactory.getLogger(ClientReleaseManager.class);
public ClientReleaseManager(final ClientReleases clientReleases,
final ScheduledExecutorService scheduledExecutorService,
final Duration refreshInterval,
final Clock clock) {
this.clientReleases = clientReleases;
this.scheduledExecutorService = scheduledExecutorService;
this.refreshInterval = refreshInterval;
this.clock = clock;
}
public boolean isVersionActive(final ClientPlatform platform, final Semver version) {
final Map<Semver, ClientRelease> releasesByVersion = clientReleasesByPlatform.get(platform);
return releasesByVersion != null &&
releasesByVersion.containsKey(version) &&
releasesByVersion.get(version).expiration().isAfter(clock.instant());
}
@Override
public void start() throws Exception {
refreshClientVersions();
refreshClientReleasesFuture =
scheduledExecutorService.scheduleWithFixedDelay(this::refreshClientVersions,
refreshInterval.toMillis(),
refreshInterval.toMillis(),
TimeUnit.MILLISECONDS);
}
@Override
public void stop() throws Exception {
if (refreshClientReleasesFuture != null) {
refreshClientReleasesFuture.cancel(true);
}
}
void refreshClientVersions() {
try {
clientReleasesByPlatform = clientReleases.getClientReleases();
} catch (final Exception e) {
logger.warn("Failed to refresh client releases", e);
}
}
}