102 lines
4.1 KiB
Java
102 lines
4.1 KiB
Java
package io.grpc;
|
|
|
|
import com.google.common.base.Preconditions;
|
|
import io.grpc.ForwardingServerCallListener;
|
|
import io.grpc.ServerCall;
|
|
import io.grpc.Status;
|
|
import java.util.concurrent.TimeoutException;
|
|
|
|
/* loaded from: classes6.dex */
|
|
public final class Contexts {
|
|
private Contexts() {
|
|
}
|
|
|
|
public static <ReqT, RespT> ServerCall.Listener<ReqT> interceptCall(Context context, ServerCall<ReqT, RespT> serverCall, Metadata metadata, ServerCallHandler<ReqT, RespT> serverCallHandler) {
|
|
Context attach = context.attach();
|
|
try {
|
|
return new ContextualizedServerCallListener(serverCallHandler.startCall(serverCall, metadata), context);
|
|
} finally {
|
|
context.detach(attach);
|
|
}
|
|
}
|
|
|
|
/* loaded from: classes6.dex */
|
|
static class ContextualizedServerCallListener<ReqT> extends ForwardingServerCallListener.SimpleForwardingServerCallListener<ReqT> {
|
|
private final Context context;
|
|
|
|
public ContextualizedServerCallListener(ServerCall.Listener<ReqT> listener, Context context) {
|
|
super(listener);
|
|
this.context = context;
|
|
}
|
|
|
|
@Override // io.grpc.ForwardingServerCallListener, io.grpc.ServerCall.Listener
|
|
public void onMessage(ReqT reqt) {
|
|
Context attach = this.context.attach();
|
|
try {
|
|
super.onMessage(reqt);
|
|
} finally {
|
|
this.context.detach(attach);
|
|
}
|
|
}
|
|
|
|
@Override // io.grpc.ForwardingServerCallListener.SimpleForwardingServerCallListener, io.grpc.ForwardingServerCallListener, io.grpc.PartialForwardingServerCallListener, io.grpc.ServerCall.Listener
|
|
public void onHalfClose() {
|
|
Context attach = this.context.attach();
|
|
try {
|
|
super.onHalfClose();
|
|
} finally {
|
|
this.context.detach(attach);
|
|
}
|
|
}
|
|
|
|
@Override // io.grpc.ForwardingServerCallListener.SimpleForwardingServerCallListener, io.grpc.ForwardingServerCallListener, io.grpc.PartialForwardingServerCallListener, io.grpc.ServerCall.Listener
|
|
public void onCancel() {
|
|
Context attach = this.context.attach();
|
|
try {
|
|
super.onCancel();
|
|
} finally {
|
|
this.context.detach(attach);
|
|
}
|
|
}
|
|
|
|
@Override // io.grpc.ForwardingServerCallListener.SimpleForwardingServerCallListener, io.grpc.ForwardingServerCallListener, io.grpc.PartialForwardingServerCallListener, io.grpc.ServerCall.Listener
|
|
public void onComplete() {
|
|
Context attach = this.context.attach();
|
|
try {
|
|
super.onComplete();
|
|
} finally {
|
|
this.context.detach(attach);
|
|
}
|
|
}
|
|
|
|
@Override // io.grpc.ForwardingServerCallListener.SimpleForwardingServerCallListener, io.grpc.ForwardingServerCallListener, io.grpc.PartialForwardingServerCallListener, io.grpc.ServerCall.Listener
|
|
public void onReady() {
|
|
Context attach = this.context.attach();
|
|
try {
|
|
super.onReady();
|
|
} finally {
|
|
this.context.detach(attach);
|
|
}
|
|
}
|
|
}
|
|
|
|
public static Status statusFromCancelled(Context context) {
|
|
Preconditions.checkNotNull(context, "context must not be null");
|
|
if (!context.isCancelled()) {
|
|
return null;
|
|
}
|
|
Throwable cancellationCause = context.cancellationCause();
|
|
if (cancellationCause == null) {
|
|
return Status.CANCELLED.withDescription("io.grpc.Context was cancelled without error");
|
|
}
|
|
if (cancellationCause instanceof TimeoutException) {
|
|
return Status.DEADLINE_EXCEEDED.withDescription(cancellationCause.getMessage()).withCause(cancellationCause);
|
|
}
|
|
Status fromThrowable = Status.fromThrowable(cancellationCause);
|
|
if (Status.Code.UNKNOWN.equals(fromThrowable.getCode()) && fromThrowable.getCause() == cancellationCause) {
|
|
return Status.CANCELLED.withDescription("Context cancelled").withCause(cancellationCause);
|
|
}
|
|
return fromThrowable.withCause(cancellationCause);
|
|
}
|
|
}
|