218 lines
11 KiB
Java
218 lines
11 KiB
Java
|
package io.grpc;
|
||
|
|
|
||
|
|
import com.google.common.base.Preconditions;
|
||
|
|
import io.grpc.MethodDescriptor;
|
||
|
|
import io.grpc.ServerCall;
|
||
|
|
import io.grpc.ServerServiceDefinition;
|
||
|
|
import io.grpc.ServiceDescriptor;
|
||
|
|
import java.io.BufferedInputStream;
|
||
|
|
import java.io.InputStream;
|
||
|
|
import java.util.ArrayList;
|
||
|
|
import java.util.Arrays;
|
||
|
|
import java.util.Collections;
|
||
|
|
import java.util.Iterator;
|
||
|
|
import java.util.List;
|
||
|
|
|
||
|
|
/* loaded from: classes6.dex */
|
||
|
|
public final class ServerInterceptors {
|
||
|
|
private ServerInterceptors() {
|
||
|
|
}
|
||
|
|
|
||
|
|
public static ServerServiceDefinition interceptForward(ServerServiceDefinition serverServiceDefinition, ServerInterceptor... serverInterceptorArr) {
|
||
|
|
return interceptForward(serverServiceDefinition, (List<? extends ServerInterceptor>) Arrays.asList(serverInterceptorArr));
|
||
|
|
}
|
||
|
|
|
||
|
|
public static ServerServiceDefinition interceptForward(BindableService bindableService, ServerInterceptor... serverInterceptorArr) {
|
||
|
|
return interceptForward(bindableService.bindService(), (List<? extends ServerInterceptor>) Arrays.asList(serverInterceptorArr));
|
||
|
|
}
|
||
|
|
|
||
|
|
public static ServerServiceDefinition interceptForward(ServerServiceDefinition serverServiceDefinition, List<? extends ServerInterceptor> list) {
|
||
|
|
ArrayList arrayList = new ArrayList(list);
|
||
|
|
Collections.reverse(arrayList);
|
||
|
|
return intercept(serverServiceDefinition, arrayList);
|
||
|
|
}
|
||
|
|
|
||
|
|
public static ServerServiceDefinition interceptForward(BindableService bindableService, List<? extends ServerInterceptor> list) {
|
||
|
|
return interceptForward(bindableService.bindService(), list);
|
||
|
|
}
|
||
|
|
|
||
|
|
public static ServerServiceDefinition intercept(ServerServiceDefinition serverServiceDefinition, ServerInterceptor... serverInterceptorArr) {
|
||
|
|
return intercept(serverServiceDefinition, (List<? extends ServerInterceptor>) Arrays.asList(serverInterceptorArr));
|
||
|
|
}
|
||
|
|
|
||
|
|
public static ServerServiceDefinition intercept(BindableService bindableService, ServerInterceptor... serverInterceptorArr) {
|
||
|
|
Preconditions.checkNotNull(bindableService, "bindableService");
|
||
|
|
return intercept(bindableService.bindService(), (List<? extends ServerInterceptor>) Arrays.asList(serverInterceptorArr));
|
||
|
|
}
|
||
|
|
|
||
|
|
public static ServerServiceDefinition intercept(ServerServiceDefinition serverServiceDefinition, List<? extends ServerInterceptor> list) {
|
||
|
|
Preconditions.checkNotNull(serverServiceDefinition, "serviceDef");
|
||
|
|
if (list.isEmpty()) {
|
||
|
|
return serverServiceDefinition;
|
||
|
|
}
|
||
|
|
ServerServiceDefinition.Builder builder = ServerServiceDefinition.builder(serverServiceDefinition.getServiceDescriptor());
|
||
|
|
Iterator<ServerMethodDefinition<?, ?>> it = serverServiceDefinition.getMethods().iterator();
|
||
|
|
while (it.hasNext()) {
|
||
|
|
wrapAndAddMethod(builder, it.next(), list);
|
||
|
|
}
|
||
|
|
return builder.build();
|
||
|
|
}
|
||
|
|
|
||
|
|
public static ServerServiceDefinition intercept(BindableService bindableService, List<? extends ServerInterceptor> list) {
|
||
|
|
Preconditions.checkNotNull(bindableService, "bindableService");
|
||
|
|
return intercept(bindableService.bindService(), list);
|
||
|
|
}
|
||
|
|
|
||
|
|
public static ServerServiceDefinition useInputStreamMessages(ServerServiceDefinition serverServiceDefinition) {
|
||
|
|
return useMarshalledMessages(serverServiceDefinition, new MethodDescriptor.Marshaller<InputStream>() { // from class: io.grpc.ServerInterceptors.1
|
||
|
|
@Override // io.grpc.MethodDescriptor.Marshaller
|
||
|
|
public InputStream stream(InputStream inputStream) {
|
||
|
|
return inputStream;
|
||
|
|
}
|
||
|
|
|
||
|
|
/* JADX WARN: Can't rename method to resolve collision */
|
||
|
|
@Override // io.grpc.MethodDescriptor.Marshaller
|
||
|
|
public InputStream parse(InputStream inputStream) {
|
||
|
|
if (inputStream.markSupported()) {
|
||
|
|
return inputStream;
|
||
|
|
}
|
||
|
|
if (inputStream instanceof KnownLength) {
|
||
|
|
return new KnownLengthBufferedInputStream(inputStream);
|
||
|
|
}
|
||
|
|
return new BufferedInputStream(inputStream);
|
||
|
|
}
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
/* JADX INFO: Access modifiers changed from: package-private */
|
||
|
|
/* loaded from: classes6.dex */
|
||
|
|
public static final class KnownLengthBufferedInputStream extends BufferedInputStream implements KnownLength {
|
||
|
|
public KnownLengthBufferedInputStream(InputStream inputStream) {
|
||
|
|
super(inputStream);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/* JADX WARN: Multi-variable type inference failed */
|
||
|
|
public static <T> ServerServiceDefinition useMarshalledMessages(ServerServiceDefinition serverServiceDefinition, MethodDescriptor.Marshaller<T> marshaller) {
|
||
|
|
ArrayList arrayList = new ArrayList();
|
||
|
|
ArrayList arrayList2 = new ArrayList();
|
||
|
|
for (ServerMethodDefinition<?, ?> serverMethodDefinition : serverServiceDefinition.getMethods()) {
|
||
|
|
MethodDescriptor build = serverMethodDefinition.getMethodDescriptor().toBuilder(marshaller, marshaller).build();
|
||
|
|
arrayList2.add(build);
|
||
|
|
arrayList.add(wrapMethod(serverMethodDefinition, build));
|
||
|
|
}
|
||
|
|
ServiceDescriptor.Builder schemaDescriptor = ServiceDescriptor.newBuilder(serverServiceDefinition.getServiceDescriptor().getName()).setSchemaDescriptor(serverServiceDefinition.getServiceDescriptor().getSchemaDescriptor());
|
||
|
|
Iterator it = arrayList2.iterator();
|
||
|
|
while (it.hasNext()) {
|
||
|
|
schemaDescriptor.addMethod((MethodDescriptor) it.next());
|
||
|
|
}
|
||
|
|
ServerServiceDefinition.Builder builder = ServerServiceDefinition.builder(schemaDescriptor.build());
|
||
|
|
Iterator it2 = arrayList.iterator();
|
||
|
|
while (it2.hasNext()) {
|
||
|
|
builder.addMethod((ServerMethodDefinition) it2.next());
|
||
|
|
}
|
||
|
|
return builder.build();
|
||
|
|
}
|
||
|
|
|
||
|
|
private static <ReqT, RespT> void wrapAndAddMethod(ServerServiceDefinition.Builder builder, ServerMethodDefinition<ReqT, RespT> serverMethodDefinition, List<? extends ServerInterceptor> list) {
|
||
|
|
ServerCallHandler<ReqT, RespT> serverCallHandler = serverMethodDefinition.getServerCallHandler();
|
||
|
|
Iterator<? extends ServerInterceptor> it = list.iterator();
|
||
|
|
while (it.hasNext()) {
|
||
|
|
serverCallHandler = InterceptCallHandler.create(it.next(), serverCallHandler);
|
||
|
|
}
|
||
|
|
builder.addMethod(serverMethodDefinition.withServerCallHandler(serverCallHandler));
|
||
|
|
}
|
||
|
|
|
||
|
|
/* JADX INFO: Access modifiers changed from: package-private */
|
||
|
|
/* loaded from: classes6.dex */
|
||
|
|
public static final class InterceptCallHandler<ReqT, RespT> implements ServerCallHandler<ReqT, RespT> {
|
||
|
|
private final ServerCallHandler<ReqT, RespT> callHandler;
|
||
|
|
private final ServerInterceptor interceptor;
|
||
|
|
|
||
|
|
public static <ReqT, RespT> InterceptCallHandler<ReqT, RespT> create(ServerInterceptor serverInterceptor, ServerCallHandler<ReqT, RespT> serverCallHandler) {
|
||
|
|
return new InterceptCallHandler<>(serverInterceptor, serverCallHandler);
|
||
|
|
}
|
||
|
|
|
||
|
|
private InterceptCallHandler(ServerInterceptor serverInterceptor, ServerCallHandler<ReqT, RespT> serverCallHandler) {
|
||
|
|
this.interceptor = (ServerInterceptor) Preconditions.checkNotNull(serverInterceptor, "interceptor");
|
||
|
|
this.callHandler = serverCallHandler;
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override // io.grpc.ServerCallHandler
|
||
|
|
public final ServerCall.Listener<ReqT> startCall(ServerCall<ReqT, RespT> serverCall, Metadata metadata) {
|
||
|
|
return this.interceptor.interceptCall(serverCall, metadata, this.callHandler);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/* JADX INFO: Access modifiers changed from: package-private */
|
||
|
|
public static <OReqT, ORespT, WReqT, WRespT> ServerMethodDefinition<WReqT, WRespT> wrapMethod(ServerMethodDefinition<OReqT, ORespT> serverMethodDefinition, MethodDescriptor<WReqT, WRespT> methodDescriptor) {
|
||
|
|
return ServerMethodDefinition.create(methodDescriptor, wrapHandler(serverMethodDefinition.getServerCallHandler(), serverMethodDefinition.getMethodDescriptor(), methodDescriptor));
|
||
|
|
}
|
||
|
|
|
||
|
|
/* JADX INFO: Access modifiers changed from: package-private */
|
||
|
|
/* JADX INFO: Add missing generic type declarations: [WRespT, WReqT] */
|
||
|
|
/* renamed from: io.grpc.ServerInterceptors$2, reason: invalid class name */
|
||
|
|
/* loaded from: classes6.dex */
|
||
|
|
public class AnonymousClass2<WReqT, WRespT> implements ServerCallHandler<WReqT, WRespT> {
|
||
|
|
final ServerCallHandler val$originalHandler;
|
||
|
|
final MethodDescriptor val$originalMethod;
|
||
|
|
final MethodDescriptor val$wrappedMethod;
|
||
|
|
|
||
|
|
AnonymousClass2(MethodDescriptor methodDescriptor, MethodDescriptor methodDescriptor2, ServerCallHandler serverCallHandler) {
|
||
|
|
this.val$originalMethod = methodDescriptor;
|
||
|
|
this.val$wrappedMethod = methodDescriptor2;
|
||
|
|
this.val$originalHandler = serverCallHandler;
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override // io.grpc.ServerCallHandler
|
||
|
|
public ServerCall.Listener<WReqT> startCall(ServerCall<WReqT, WRespT> serverCall, Metadata metadata) {
|
||
|
|
return new PartialForwardingServerCallListener<WReqT>(this, this.val$originalHandler.startCall(new PartialForwardingServerCall<OReqT, ORespT>(this, serverCall) { // from class: io.grpc.ServerInterceptors.2.1
|
||
|
|
final AnonymousClass2 this$0;
|
||
|
|
final ServerCall val$call;
|
||
|
|
|
||
|
|
{
|
||
|
|
this.this$0 = this;
|
||
|
|
this.val$call = serverCall;
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override // io.grpc.ServerCall
|
||
|
|
public void sendMessage(ORespT orespt) {
|
||
|
|
delegate().sendMessage(this.this$0.val$wrappedMethod.parseResponse(this.this$0.val$originalMethod.streamResponse(orespt)));
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override // io.grpc.ServerCall
|
||
|
|
public MethodDescriptor<OReqT, ORespT> getMethodDescriptor() {
|
||
|
|
return this.this$0.val$originalMethod;
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override // io.grpc.PartialForwardingServerCall
|
||
|
|
protected ServerCall<WReqT, WRespT> delegate() {
|
||
|
|
return this.val$call;
|
||
|
|
}
|
||
|
|
}, metadata)) { // from class: io.grpc.ServerInterceptors.2.2
|
||
|
|
final AnonymousClass2 this$0;
|
||
|
|
final ServerCall.Listener val$originalListener;
|
||
|
|
|
||
|
|
{
|
||
|
|
this.this$0 = this;
|
||
|
|
this.val$originalListener = r2;
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override // io.grpc.ServerCall.Listener
|
||
|
|
public void onMessage(WReqT wreqt) {
|
||
|
|
delegate().onMessage(this.this$0.val$originalMethod.parseRequest(this.this$0.val$wrappedMethod.streamRequest(wreqt)));
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override // io.grpc.PartialForwardingServerCallListener
|
||
|
|
protected ServerCall.Listener<OReqT> delegate() {
|
||
|
|
return this.val$originalListener;
|
||
|
|
}
|
||
|
|
};
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
private static <OReqT, ORespT, WReqT, WRespT> ServerCallHandler<WReqT, WRespT> wrapHandler(ServerCallHandler<OReqT, ORespT> serverCallHandler, MethodDescriptor<OReqT, ORespT> methodDescriptor, MethodDescriptor<WReqT, WRespT> methodDescriptor2) {
|
||
|
|
return new AnonymousClass2(methodDescriptor, methodDescriptor2, serverCallHandler);
|
||
|
|
}
|
||
|
|
}
|