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

Count the prevalence of keys stored as strings or as bytes

This commit is contained in:
Jon Chambers 2023-05-25 10:57:39 -04:00 committed by Chris Eager
parent 1ab6bff54e
commit 70b54e227e
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
service/src/main/java/org/whispersystems/textsecuregcm/storage/Keys.java
View File

@ -62,6 +62,8 @@ public class Keys extends AbstractDynamoDbStore {
private static final DistributionSummary KEY_COUNT_DISTRIBUTION = Metrics.summary(name(Keys.class, "keyCount")); private static final DistributionSummary KEY_COUNT_DISTRIBUTION = Metrics.summary(name(Keys.class, "keyCount"));
private static final Counter KEYS_EMPTY_TAKE_COUNTER = Metrics.counter(name(Keys.class, "takeKeyEmpty")); private static final Counter KEYS_EMPTY_TAKE_COUNTER = Metrics.counter(name(Keys.class, "takeKeyEmpty"));
private static final Counter TOO_MANY_LAST_RESORT_KEYS_COUNTER = Metrics.counter(name(Keys.class, "tooManyLastResortKeys")); private static final Counter TOO_MANY_LAST_RESORT_KEYS_COUNTER = Metrics.counter(name(Keys.class, "tooManyLastResortKeys"));
private static final Counter PARSE_BYTES_FROM_STRING_COUNTER = Metrics.counter(name(Keys.class, "parseByteArray"), "format", "string");
private static final Counter READ_BYTES_FROM_BYTE_ARRAY_COUNTER = Metrics.counter(name(Keys.class, "parseByteArray"), "format", "bytes");
public Keys( public Keys(
final DynamoDbClient dynamoDB, final DynamoDbClient dynamoDB,
@ -404,8 +406,10 @@ public class Keys extends AbstractDynamoDbStore {
@VisibleForTesting @VisibleForTesting
static byte[] extractByteArray(final AttributeValue attributeValue) { static byte[] extractByteArray(final AttributeValue attributeValue) {
if (attributeValue.b() != null) { if (attributeValue.b() != null) {
READ_BYTES_FROM_BYTE_ARRAY_COUNTER.increment();
return attributeValue.b().asByteArray(); return attributeValue.b().asByteArray();
} else if (StringUtils.isNotBlank(attributeValue.s())) { } else if (StringUtils.isNotBlank(attributeValue.s())) {
PARSE_BYTES_FROM_STRING_COUNTER.increment();
return Base64.getDecoder().decode(attributeValue.s()); return Base64.getDecoder().decode(attributeValue.s());
} }