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

Add a repository for client release information

Browse Source
This commit is contained in:
Jon Chambers
2023-07-25 17:28:24 -04:00
committed by Jon Chambers
parent 60cc0c482e
commit 10689843b0
5 changed files with 167 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
service/src/main/java/org/whispersystems/textsecuregcm/storage/ClientRelease.java Normal file
View File

@@ -0,0 +1,13 @@
/*
* Copyright 2023 Signal Messenger, LLC
* SPDX-License-Identifier: AGPL-3.0-only
*/
package org.whispersystems.textsecuregcm.storage;
import com.vdurmont.semver4j.Semver;
import org.whispersystems.textsecuregcm.util.ua.ClientPlatform;
import java.time.Instant;
public record ClientRelease(ClientPlatform platform, Semver version, Instant release, Instant expiration) {
}

69
service/src/main/java/org/whispersystems/textsecuregcm/storage/ClientReleases.java Normal file
View File

@@ -0,0 +1,69 @@
/*
* Copyright 2023 Signal Messenger, LLC
* SPDX-License-Identifier: AGPL-3.0-only
*/
package org.whispersystems.textsecuregcm.storage;
import com.vdurmont.semver4j.Semver;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.whispersystems.textsecuregcm.util.ua.ClientPlatform;
import reactor.core.publisher.Flux;
import reactor.util.function.Tuple2;
import reactor.util.function.Tuples;
import software.amazon.awssdk.services.dynamodb.DynamoDbAsyncClient;
import software.amazon.awssdk.services.dynamodb.model.AttributeValue;
import software.amazon.awssdk.services.dynamodb.model.ScanRequest;
import javax.annotation.Nullable;
import java.time.Instant;
import java.util.Collections;
import java.util.Map;
public class ClientReleases {
private final DynamoDbAsyncClient dynamoDbAsyncClient;
private final String tableName;
public static final String ATTR_PLATFORM = "P";
public static final String ATTR_VERSION = "V";
public static final String ATTR_RELEASE_TIMESTAMP = "T";
public static final String ATTR_EXPIRATION = "E";
private static final Logger logger = LoggerFactory.getLogger(ClientReleases.class);
public ClientReleases(final DynamoDbAsyncClient dynamoDbAsyncClient, final String tableName) {
this.dynamoDbAsyncClient = dynamoDbAsyncClient;
this.tableName = tableName;
}
public Map<ClientPlatform, Map<Semver, ClientRelease>> getClientReleases() {
return Collections.unmodifiableMap(
Flux.from(dynamoDbAsyncClient.scanPaginator(ScanRequest.builder()
.tableName(tableName)
.build())
.items())
.mapNotNull(ClientReleases::releaseFromItem)
.groupBy(ClientRelease::platform)
.flatMap(groupedFlux -> groupedFlux.collectMap(ClientRelease::version)
.map(releasesByVersion -> Tuples.of(groupedFlux.key(), releasesByVersion)))
.collectMap(Tuple2::getT1, Tuple2::getT2)
.blockOptional()
.orElseGet(Collections::emptyMap));
}
@Nullable
static ClientRelease releaseFromItem(final Map<String, AttributeValue> item) {
try {
final ClientPlatform platform = ClientPlatform.valueOf(item.get(ATTR_PLATFORM).s());
final Semver version = new Semver(item.get(ATTR_VERSION).s());
final Instant release = Instant.ofEpochSecond(Long.parseLong(item.get(ATTR_RELEASE_TIMESTAMP).n()));
final Instant expiration = Instant.ofEpochSecond(Long.parseLong(item.get(ATTR_EXPIRATION).n()));
return new ClientRelease(platform, version, release, expiration);
} catch (final Exception e) {
logger.warn("Failed to parse client release item", e);
return null;
}
}
}