what-the-bank/sources/io/grpc/Context.java

695 lines
25 KiB
Java
Raw Normal View History

2024-07-27 18:17:47 +07:00
package io.grpc;
import io.grpc.PersistentHashArrayMappedTrie;
import java.io.Closeable;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.concurrent.Callable;
import java.util.concurrent.Executor;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.ScheduledFuture;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.TimeoutException;
import java.util.concurrent.atomic.AtomicReference;
import java.util.logging.Level;
import java.util.logging.Logger;
/* loaded from: classes6.dex */
public class Context {
static final int CONTEXT_DEPTH_WARN_THRESH = 1000;
final CancellableContext cancellableAncestor;
final int generation;
final PersistentHashArrayMappedTrie.Node<Key<?>, Object> keyValueEntries;
static final Logger log = Logger.getLogger(Context.class.getName());
public static final Context ROOT = new Context();
/* loaded from: classes.dex */
@interface CanIgnoreReturnValue {
}
/* loaded from: classes6.dex */
public interface CancellationListener {
void cancelled(Context context);
}
/* loaded from: classes.dex */
@interface CheckReturnValue {
}
static Storage storage() {
return LazyStorage.storage;
}
/* JADX INFO: Access modifiers changed from: package-private */
/* loaded from: classes6.dex */
public static final class LazyStorage {
static final Storage storage;
private LazyStorage() {
}
static {
AtomicReference atomicReference = new AtomicReference();
storage = createStorage(atomicReference);
Throwable th = (Throwable) atomicReference.get();
if (th != null) {
Context.log.log(Level.FINE, "Storage override doesn't exist. Using default", th);
}
}
private static Storage createStorage(AtomicReference<? super ClassNotFoundException> atomicReference) {
try {
return (Storage) Class.forName("io.grpc.override.ContextStorageOverride").asSubclass(Storage.class).getConstructor(new Class[0]).newInstance(new Object[0]);
} catch (ClassNotFoundException e) {
atomicReference.set(e);
return new ThreadLocalContextStorage();
} catch (Exception e2) {
throw new RuntimeException("Storage override failed to initialize", e2);
}
}
}
public static <T> Key<T> key(String str) {
return new Key<>(str);
}
public static <T> Key<T> keyWithDefault(String str, T t) {
return new Key<>(str, t);
}
public static Context current() {
Context current = storage().current();
return current == null ? ROOT : current;
}
private Context(PersistentHashArrayMappedTrie.Node<Key<?>, Object> node, int i) {
this.cancellableAncestor = null;
this.keyValueEntries = node;
this.generation = i;
validateGeneration(i);
}
private Context(Context context, PersistentHashArrayMappedTrie.Node<Key<?>, Object> node) {
this.cancellableAncestor = cancellableAncestor(context);
this.keyValueEntries = node;
int i = context.generation + 1;
this.generation = i;
validateGeneration(i);
}
private Context() {
this.cancellableAncestor = null;
this.keyValueEntries = null;
this.generation = 0;
validateGeneration(0);
}
public CancellableContext withCancellation() {
return new CancellableContext();
}
public CancellableContext withDeadlineAfter(long j, TimeUnit timeUnit, ScheduledExecutorService scheduledExecutorService) {
return withDeadline(Deadline.after(j, timeUnit), scheduledExecutorService);
}
public CancellableContext withDeadline(Deadline deadline, ScheduledExecutorService scheduledExecutorService) {
boolean z;
checkNotNull(deadline, "deadline");
checkNotNull(scheduledExecutorService, "scheduler");
Deadline deadline2 = getDeadline();
if (deadline2 == null || deadline2.compareTo(deadline) > 0) {
z = true;
} else {
z = false;
deadline = deadline2;
}
CancellableContext cancellableContext = new CancellableContext(deadline);
if (z) {
cancellableContext.setUpDeadlineCancellation(deadline, scheduledExecutorService);
}
return cancellableContext;
}
public <V> Context withValue(Key<V> key, V v) {
return new Context(this, (PersistentHashArrayMappedTrie.Node<Key<?>, Object>) PersistentHashArrayMappedTrie.put(this.keyValueEntries, key, v));
}
public <V1, V2> Context withValues(Key<V1> key, V1 v1, Key<V2> key2, V2 v2) {
return new Context(this, (PersistentHashArrayMappedTrie.Node<Key<?>, Object>) PersistentHashArrayMappedTrie.put(PersistentHashArrayMappedTrie.put(this.keyValueEntries, key, v1), key2, v2));
}
public <V1, V2, V3> Context withValues(Key<V1> key, V1 v1, Key<V2> key2, V2 v2, Key<V3> key3, V3 v3) {
return new Context(this, (PersistentHashArrayMappedTrie.Node<Key<?>, Object>) PersistentHashArrayMappedTrie.put(PersistentHashArrayMappedTrie.put(PersistentHashArrayMappedTrie.put(this.keyValueEntries, key, v1), key2, v2), key3, v3));
}
public <V1, V2, V3, V4> Context withValues(Key<V1> key, V1 v1, Key<V2> key2, V2 v2, Key<V3> key3, V3 v3, Key<V4> key4, V4 v4) {
return new Context(this, (PersistentHashArrayMappedTrie.Node<Key<?>, Object>) PersistentHashArrayMappedTrie.put(PersistentHashArrayMappedTrie.put(PersistentHashArrayMappedTrie.put(PersistentHashArrayMappedTrie.put(this.keyValueEntries, key, v1), key2, v2), key3, v3), key4, v4));
}
public Context fork() {
return new Context(this.keyValueEntries, this.generation + 1);
}
public Context attach() {
Context doAttach = storage().doAttach(this);
return doAttach == null ? ROOT : doAttach;
}
public void detach(Context context) {
checkNotNull(context, "toAttach");
storage().detach(this, context);
}
boolean isCurrent() {
return current() == this;
}
public boolean isCancelled() {
CancellableContext cancellableContext = this.cancellableAncestor;
if (cancellableContext == null) {
return false;
}
return cancellableContext.isCancelled();
}
public Throwable cancellationCause() {
CancellableContext cancellableContext = this.cancellableAncestor;
if (cancellableContext == null) {
return null;
}
return cancellableContext.cancellationCause();
}
public Deadline getDeadline() {
CancellableContext cancellableContext = this.cancellableAncestor;
if (cancellableContext == null) {
return null;
}
return cancellableContext.getDeadline();
}
public void addListener(CancellationListener cancellationListener, Executor executor) {
checkNotNull(cancellationListener, "cancellationListener");
checkNotNull(executor, "executor");
CancellableContext cancellableContext = this.cancellableAncestor;
if (cancellableContext == null) {
return;
}
cancellableContext.addListenerInternal(new ExecutableListener(executor, cancellationListener, this));
}
public void removeListener(CancellationListener cancellationListener) {
CancellableContext cancellableContext = this.cancellableAncestor;
if (cancellableContext == null) {
return;
}
cancellableContext.removeListenerInternal(cancellationListener, this);
}
int listenerCount() {
CancellableContext cancellableContext = this.cancellableAncestor;
if (cancellableContext == null) {
return 0;
}
return cancellableContext.listenerCount();
}
public void run(Runnable runnable) {
Context attach = attach();
try {
runnable.run();
} finally {
detach(attach);
}
}
public <V> V call(Callable<V> callable) throws Exception {
Context attach = attach();
try {
return callable.call();
} finally {
detach(attach);
}
}
public Runnable wrap(Runnable runnable) {
return new Runnable(this, runnable) { // from class: io.grpc.Context.1
final Context this$0;
final Runnable val$r;
{
this.this$0 = this;
this.val$r = runnable;
}
@Override // java.lang.Runnable
public void run() {
Context attach = this.this$0.attach();
try {
this.val$r.run();
} finally {
this.this$0.detach(attach);
}
}
};
}
public <C> Callable<C> wrap(Callable<C> callable) {
return new Callable<C>(this, callable) { // from class: io.grpc.Context.2
final Context this$0;
final Callable val$c;
{
this.this$0 = this;
this.val$c = callable;
}
@Override // java.util.concurrent.Callable
public C call() throws Exception {
Context attach = this.this$0.attach();
try {
return (C) this.val$c.call();
} finally {
this.this$0.detach(attach);
}
}
};
}
public Executor fixedContextExecutor(Executor executor) {
return new Executor(this, executor) { // from class: io.grpc.Context.1FixedContextExecutor
final Context this$0;
final Executor val$e;
{
this.this$0 = this;
this.val$e = executor;
}
@Override // java.util.concurrent.Executor
public final void execute(Runnable runnable) {
this.val$e.execute(this.this$0.wrap(runnable));
}
};
}
public static Executor currentContextExecutor(Executor executor) {
return new Executor(executor) { // from class: io.grpc.Context.1CurrentContextExecutor
final Executor val$e;
{
this.val$e = executor;
}
@Override // java.util.concurrent.Executor
public final void execute(Runnable runnable) {
this.val$e.execute(Context.current().wrap(runnable));
}
};
}
/* loaded from: classes6.dex */
public static final class CancellableContext extends Context implements Closeable {
private Throwable cancellationCause;
private boolean cancelled;
private final Deadline deadline;
private ArrayList<ExecutableListener> listeners;
private CancellationListener parentListener;
private ScheduledFuture<?> pendingDeadline;
private final Context uncancellableSurrogate;
/* JADX WARN: Illegal instructions before constructor call */
/*
Code decompiled incorrectly, please refer to instructions dump.
To view partially-correct add '--show-bad-code' argument
*/
private CancellableContext(io.grpc.Context r3) {
/*
r2 = this;
io.grpc.PersistentHashArrayMappedTrie$Node<io.grpc.Context$Key<?>, java.lang.Object> r0 = r3.keyValueEntries
r1 = 0
r2.<init>(r0)
io.grpc.Deadline r3 = r3.getDeadline()
r2.deadline = r3
io.grpc.Context r3 = new io.grpc.Context
io.grpc.PersistentHashArrayMappedTrie$Node<io.grpc.Context$Key<?>, java.lang.Object> r0 = r2.keyValueEntries
r3.<init>(r0)
r2.uncancellableSurrogate = r3
return
*/
throw new UnsupportedOperationException("Method not decompiled: io.grpc.Context.CancellableContext.<init>(io.grpc.Context):void");
}
/* JADX WARN: Illegal instructions before constructor call */
/*
Code decompiled incorrectly, please refer to instructions dump.
To view partially-correct add '--show-bad-code' argument
*/
private CancellableContext(io.grpc.Context r3, io.grpc.Deadline r4) {
/*
r2 = this;
io.grpc.PersistentHashArrayMappedTrie$Node<io.grpc.Context$Key<?>, java.lang.Object> r0 = r3.keyValueEntries
r1 = 0
r2.<init>(r0)
r2.deadline = r4
io.grpc.Context r3 = new io.grpc.Context
io.grpc.PersistentHashArrayMappedTrie$Node<io.grpc.Context$Key<?>, java.lang.Object> r4 = r2.keyValueEntries
r3.<init>(r4)
r2.uncancellableSurrogate = r3
return
*/
throw new UnsupportedOperationException("Method not decompiled: io.grpc.Context.CancellableContext.<init>(io.grpc.Context, io.grpc.Deadline):void");
}
/* JADX INFO: Access modifiers changed from: private */
public void setUpDeadlineCancellation(Deadline deadline, ScheduledExecutorService scheduledExecutorService) {
if (!deadline.isExpired()) {
synchronized (this) {
this.pendingDeadline = deadline.runOnExpiration(new Runnable(this) { // from class: io.grpc.Context.CancellableContext.1CancelOnExpiration
final CancellableContext this$0;
{
this.this$0 = this;
}
@Override // java.lang.Runnable
public final void run() {
try {
this.this$0.cancel(new TimeoutException("context timed out"));
} catch (Throwable th) {
Context.log.log(Level.SEVERE, "Cancel threw an exception, which should not happen", th);
}
}
}, scheduledExecutorService);
}
} else {
cancel(new TimeoutException("context timed out"));
}
}
@Override // io.grpc.Context
public final Context attach() {
return this.uncancellableSurrogate.attach();
}
@Override // io.grpc.Context
public final void detach(Context context) {
this.uncancellableSurrogate.detach(context);
}
@Override // io.grpc.Context
public final void addListener(CancellationListener cancellationListener, Executor executor) {
checkNotNull(cancellationListener, "cancellationListener");
checkNotNull(executor, "executor");
addListenerInternal(new ExecutableListener(executor, cancellationListener, this));
}
/* JADX INFO: Access modifiers changed from: private */
public void addListenerInternal(ExecutableListener executableListener) {
synchronized (this) {
if (isCancelled()) {
executableListener.deliver();
} else {
ArrayList<ExecutableListener> arrayList = this.listeners;
if (arrayList == null) {
ArrayList<ExecutableListener> arrayList2 = new ArrayList<>();
this.listeners = arrayList2;
arrayList2.add(executableListener);
if (this.cancellableAncestor != null) {
this.parentListener = new CancellationListener(this) { // from class: io.grpc.Context.CancellableContext.1
final CancellableContext this$0;
{
this.this$0 = this;
}
@Override // io.grpc.Context.CancellationListener
public void cancelled(Context context) {
this.this$0.cancel(context.cancellationCause());
}
};
this.cancellableAncestor.addListenerInternal(new ExecutableListener(DirectExecutor.INSTANCE, this.parentListener, this));
}
} else {
arrayList.add(executableListener);
}
}
}
}
@Override // io.grpc.Context
public final void removeListener(CancellationListener cancellationListener) {
removeListenerInternal(cancellationListener, this);
}
/* JADX INFO: Access modifiers changed from: private */
public void removeListenerInternal(CancellationListener cancellationListener, Context context) {
synchronized (this) {
ArrayList<ExecutableListener> arrayList = this.listeners;
if (arrayList != null) {
int size = arrayList.size() - 1;
while (true) {
if (size < 0) {
break;
}
ExecutableListener executableListener = this.listeners.get(size);
if (executableListener.listener == cancellationListener && executableListener.context == context) {
this.listeners.remove(size);
break;
}
size--;
}
if (this.listeners.isEmpty()) {
if (this.cancellableAncestor != null) {
this.cancellableAncestor.removeListener(this.parentListener);
}
this.parentListener = null;
this.listeners = null;
}
}
}
}
@Override // io.grpc.Context
@Deprecated
public final boolean isCurrent() {
return this.uncancellableSurrogate.isCurrent();
}
public final boolean cancel(Throwable th) {
ScheduledFuture<?> scheduledFuture;
boolean z;
synchronized (this) {
scheduledFuture = null;
if (this.cancelled) {
z = false;
} else {
z = true;
this.cancelled = true;
ScheduledFuture<?> scheduledFuture2 = this.pendingDeadline;
if (scheduledFuture2 != null) {
this.pendingDeadline = null;
scheduledFuture = scheduledFuture2;
}
this.cancellationCause = th;
}
}
if (scheduledFuture != null) {
scheduledFuture.cancel(false);
}
if (z) {
notifyAndClearListeners();
}
return z;
}
private void notifyAndClearListeners() {
synchronized (this) {
ArrayList<ExecutableListener> arrayList = this.listeners;
if (arrayList == null) {
return;
}
CancellationListener cancellationListener = this.parentListener;
this.parentListener = null;
this.listeners = null;
Iterator<ExecutableListener> it = arrayList.iterator();
while (it.hasNext()) {
ExecutableListener next = it.next();
if (next.context == this) {
next.deliver();
}
}
Iterator<ExecutableListener> it2 = arrayList.iterator();
while (it2.hasNext()) {
ExecutableListener next2 = it2.next();
if (next2.context != this) {
next2.deliver();
}
}
if (this.cancellableAncestor != null) {
this.cancellableAncestor.removeListener(cancellationListener);
}
}
}
@Override // io.grpc.Context
final int listenerCount() {
int size;
synchronized (this) {
ArrayList<ExecutableListener> arrayList = this.listeners;
size = arrayList == null ? 0 : arrayList.size();
}
return size;
}
public final void detachAndCancel(Context context, Throwable th) {
try {
detach(context);
} finally {
cancel(th);
}
}
@Override // io.grpc.Context
public final boolean isCancelled() {
synchronized (this) {
if (this.cancelled) {
return true;
}
if (!super.isCancelled()) {
return false;
}
cancel(super.cancellationCause());
return true;
}
}
@Override // io.grpc.Context
public final Throwable cancellationCause() {
if (isCancelled()) {
return this.cancellationCause;
}
return null;
}
@Override // java.io.Closeable, java.lang.AutoCloseable
public final void close() {
cancel(null);
}
@Override // io.grpc.Context
public final Deadline getDeadline() {
return this.deadline;
}
}
/* loaded from: classes6.dex */
public static final class Key<T> {
private final T defaultValue;
private final String name;
Key(String str) {
this(str, null);
}
Key(String str, T t) {
this.name = (String) Context.checkNotNull(str, "name");
this.defaultValue = t;
}
public final T get() {
return get(Context.current());
}
public final T get(Context context) {
T t = (T) PersistentHashArrayMappedTrie.get(context.keyValueEntries, this);
return t == null ? this.defaultValue : t;
}
public final String toString() {
return this.name;
}
}
/* loaded from: classes6.dex */
public static abstract class Storage {
public abstract Context current();
public abstract void detach(Context context, Context context2);
@Deprecated
public void attach(Context context) {
throw new UnsupportedOperationException("Deprecated. Do not call.");
}
public Context doAttach(Context context) {
Context current = current();
attach(context);
return current;
}
}
/* JADX INFO: Access modifiers changed from: package-private */
/* loaded from: classes6.dex */
public static final class ExecutableListener implements Runnable {
private final Context context;
private final Executor executor;
final CancellationListener listener;
ExecutableListener(Executor executor, CancellationListener cancellationListener, Context context) {
this.executor = executor;
this.listener = cancellationListener;
this.context = context;
}
final void deliver() {
try {
this.executor.execute(this);
} catch (Throwable th) {
Context.log.log(Level.INFO, "Exception notifying context listener", th);
}
}
@Override // java.lang.Runnable
public final void run() {
this.listener.cancelled(this.context);
}
}
static <T> T checkNotNull(T t, Object obj) {
if (t != null) {
return t;
}
throw new NullPointerException(String.valueOf(obj));
}
/* JADX INFO: Access modifiers changed from: package-private */
/* loaded from: classes6.dex */
public enum DirectExecutor implements Executor {
INSTANCE;
@Override // java.util.concurrent.Executor
public final void execute(Runnable runnable) {
runnable.run();
}
@Override // java.lang.Enum
public final String toString() {
return "Context.DirectExecutor";
}
}
static CancellableContext cancellableAncestor(Context context) {
if (context instanceof CancellableContext) {
return (CancellableContext) context;
}
return context.cancellableAncestor;
}
private static void validateGeneration(int i) {
if (i == 1000) {
log.log(Level.SEVERE, "Context ancestry chain length is abnormally long. This suggests an error in application code. Length exceeded: 1000", (Throwable) new Exception());
}
}
}