gRPC客户端超时配置：基于截止日期实现弹性调用

import io.grpc.ManagedChannel;
import io.grpc.ManagedChannelBuilder;
import io.grpc.StatusRuntimeException;
import com.example.grpc.GreeterGrpc; // 假设您的gRPC服务名为Greeter
import com.example.grpc.HelloRequest; // 假设您的请求消息名为HelloRequest
import com.example.grpc.HelloReply;   // 假设您的响应消息名为HelloReply
import java.util.concurrent.TimeUnit;

public class GrpcClientWithTimeout {

    public static void main(String[] args) {
        // 配置 gRPC 通道
        ManagedChannel channel = ManagedChannelBuilder.forAddress("localhost", 50051)
                .usePlaintext() // 仅用于示例，生产环境请使用TLS
                .build();

        try {
            // 创建同步阻塞的 gRPC 存根
            GreeterGrpc.GreeterBlockingStub blockingStub = GreeterGrpc.newBlockingStub(channel);

            // 设置 5 秒的超时（截止日期）
            // withDeadlineAfter 方法返回一个新的 Stub 实例，原 Stub 不受影响
            GreeterGrpc.GreeterBlockingStub stubWithTimeout = blockingStub.withDeadlineAfter(5, TimeUnit.SECONDS);

            // 构建请求
            HelloRequest request = HelloRequest.newBuilder().setName("World").build();

            // 发起带有截止日期的 RPC 调用
            HelloReply reply = stubWithTimeout.sayHello(request);

            System.out.println("成功收到响应: " + reply.getMessage());

        } catch (StatusRuntimeException e) {
            // 处理 RPC 调用中可能出现的异常
            if (e.getStatus().getCode() == io.grpc.Status.Code.DEADLINE_EXCEEDED) {
                System.err.println("RPC 调用超时 (DEADLINE_EXCEEDED): " + e.getMessage());
            } else {
                System.err.println("RPC 调用失败: " + e.getStatus().getDescription());
            }
        } finally {
            // 关闭通道，释放资源
            if (channel != null) {
                channel.shutdown();
                try {
                    channel.awaitTermination(5, TimeUnit.SECONDS);
                } catch (InterruptedException e) {
                    System.err.println("通道关闭中断: " + e.getMessage());
                }
            }
        }
    }
}