ClientInterceptor拦截器

Question

实现一个GRPC的ClientInterceptor，当FullMethodName equal "path"的时候直接return response，而不是继续发送请求给server

Answer 1

import io.grpc.*;
import io.grpc.ClientInterceptors.CheckedForwardingClientCall.SimpleForwardingClientCall;
import io.grpc.ForwardingClientCallListener.SimpleForwardingClientCallListener;
import io.grpc.Metadata.Key;
import io.grpc.MethodDescriptor.MethodType;

public class CustomInterceptor implements ClientInterceptor {
    private static final Key<String> FULL_METHOD_NAME = Key.of("full_method_name", Metadata.ASCII_STRING_MARSHALLER);

    @Override
    public <ReqT, RespT> ClientCall<ReqT, RespT> interceptCall(MethodDescriptor<ReqT, RespT> method,
            CallOptions callOptions, Channel next) {
        String fullMethodName = method.getFullMethodName();
        
        if (fullMethodName.equals("path")) {
            return new SimpleForwardingClientCall<ReqT, RespT>(next.newCall(method, callOptions)) {
                @Override
                public void start(Listener<RespT> delegate, Metadata headers) {
                    // 创建一个固定的 response
                    // 这里的 MyResponse 是你定义的自定义消息类型
                    delegate.onMessage((RespT) MyResponse.newBuilder().build());
                    delegate.onClose(Status.OK, new Metadata());
                }
            };
        } else {
            return next.newCall(method, callOptions);
        }
    }
}

然后，在你的 gRPC 客户端代码中，将自定义的 Interceptor 添加到 Channel 上，如下所示：

import io.grpc.*;

public class MyGrpcClient {
    public static void main(String[] args) {
        ManagedChannel channel = ManagedChannelBuilder.forAddress("localhost", 50051)
                .usePlaintext()
                .build();

        // 创建 Interceptor
        CustomInterceptor interceptor = new CustomInterceptor();

        // 创建 Channel，并将 Interceptor 添加到 Channel
        Channel interceptedChannel = ClientInterceptors.intercept(channel, interceptor);

        // 创建你的 gRPC 客户端
        YourGrpcClient client = YourGrpcClient.newBlockingStub(interceptedChannel);

        try {
            // 调用服务中的方法，会根据 FullMethodName 进行拦截判断
            client.myMethod(request);
            
            // ...
        } catch (Exception e) {
            // 异常处理
        }
    }
}