92 lines
3.7 KiB
Java
92 lines
3.7 KiB
Java
|
package io.grpc;
|
||
|
|
|
||
|
|
import com.google.common.base.Joiner;
|
||
|
|
import com.google.common.base.Preconditions;
|
||
|
|
import io.grpc.Codec;
|
||
|
|
import java.nio.charset.StandardCharsets;
|
||
|
|
import java.util.Collections;
|
||
|
|
import java.util.HashSet;
|
||
|
|
import java.util.LinkedHashMap;
|
||
|
|
import java.util.Map;
|
||
|
|
import java.util.Set;
|
||
|
|
|
||
|
|
/* loaded from: classes6.dex */
|
||
|
|
public final class DecompressorRegistry {
|
||
|
|
static final Joiner ACCEPT_ENCODING_JOINER = Joiner.on(',');
|
||
|
|
private static final DecompressorRegistry DEFAULT_INSTANCE = emptyInstance().with(new Codec.Gzip(), true).with(Codec.Identity.NONE, false);
|
||
|
|
private final byte[] advertisedDecompressors;
|
||
|
|
private final Map<String, DecompressorInfo> decompressors;
|
||
|
|
|
||
|
|
public static DecompressorRegistry emptyInstance() {
|
||
|
|
return new DecompressorRegistry();
|
||
|
|
}
|
||
|
|
|
||
|
|
public final DecompressorRegistry with(Decompressor decompressor, boolean z) {
|
||
|
|
return new DecompressorRegistry(decompressor, z, this);
|
||
|
|
}
|
||
|
|
|
||
|
|
private DecompressorRegistry(Decompressor decompressor, boolean z, DecompressorRegistry decompressorRegistry) {
|
||
|
|
String messageEncoding = decompressor.getMessageEncoding();
|
||
|
|
Preconditions.checkArgument(!messageEncoding.contains(","), "Comma is currently not allowed in message encoding");
|
||
|
|
int size = decompressorRegistry.decompressors.size();
|
||
|
|
LinkedHashMap linkedHashMap = new LinkedHashMap(decompressorRegistry.decompressors.containsKey(decompressor.getMessageEncoding()) ? size : size + 1);
|
||
|
|
for (DecompressorInfo decompressorInfo : decompressorRegistry.decompressors.values()) {
|
||
|
|
String messageEncoding2 = decompressorInfo.decompressor.getMessageEncoding();
|
||
|
|
if (!messageEncoding2.equals(messageEncoding)) {
|
||
|
|
linkedHashMap.put(messageEncoding2, new DecompressorInfo(decompressorInfo.decompressor, decompressorInfo.advertised));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
linkedHashMap.put(messageEncoding, new DecompressorInfo(decompressor, z));
|
||
|
|
this.decompressors = Collections.unmodifiableMap(linkedHashMap);
|
||
|
|
this.advertisedDecompressors = ACCEPT_ENCODING_JOINER.join(getAdvertisedMessageEncodings()).getBytes(StandardCharsets.US_ASCII);
|
||
|
|
}
|
||
|
|
|
||
|
|
private DecompressorRegistry() {
|
||
|
|
this.decompressors = new LinkedHashMap(0);
|
||
|
|
this.advertisedDecompressors = new byte[0];
|
||
|
|
}
|
||
|
|
|
||
|
|
public final Set<String> getKnownMessageEncodings() {
|
||
|
|
return this.decompressors.keySet();
|
||
|
|
}
|
||
|
|
|
||
|
|
public final Set<String> getAdvertisedMessageEncodings() {
|
||
|
|
HashSet hashSet = new HashSet(this.decompressors.size());
|
||
|
|
for (Map.Entry<String, DecompressorInfo> entry : this.decompressors.entrySet()) {
|
||
|
|
if (entry.getValue().advertised) {
|
||
|
|
hashSet.add(entry.getKey());
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return Collections.unmodifiableSet(hashSet);
|
||
|
|
}
|
||
|
|
|
||
|
|
public final Decompressor lookupDecompressor(String str) {
|
||
|
|
DecompressorInfo decompressorInfo = this.decompressors.get(str);
|
||
|
|
if (decompressorInfo != null) {
|
||
|
|
return decompressorInfo.decompressor;
|
||
|
|
}
|
||
|
|
return null;
|
||
|
|
}
|
||
|
|
|
||
|
|
/* JADX INFO: Access modifiers changed from: package-private */
|
||
|
|
/* loaded from: classes6.dex */
|
||
|
|
public static final class DecompressorInfo {
|
||
|
|
final boolean advertised;
|
||
|
|
final Decompressor decompressor;
|
||
|
|
|
||
|
|
DecompressorInfo(Decompressor decompressor, boolean z) {
|
||
|
|
this.decompressor = (Decompressor) Preconditions.checkNotNull(decompressor, "decompressor");
|
||
|
|
this.advertised = z;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/* JADX INFO: Access modifiers changed from: package-private */
|
||
|
|
public final byte[] getRawAdvertisedMessageEncodings() {
|
||
|
|
return this.advertisedDecompressors;
|
||
|
|
}
|
||
|
|
|
||
|
|
public static DecompressorRegistry getDefaultInstance() {
|
||
|
|
return DEFAULT_INSTANCE;
|
||
|
|
}
|
||
|
|
}
|