aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--grpc/grpc-client/src/main/java/org/open/infc/grpc/client/OpenInterfaceGrpcClient.java31
1 files changed, 27 insertions, 4 deletions
diff --git a/grpc/grpc-client/src/main/java/org/open/infc/grpc/client/OpenInterfaceGrpcClient.java b/grpc/grpc-client/src/main/java/org/open/infc/grpc/client/OpenInterfaceGrpcClient.java
index 2d9d32fc..9a818d7a 100644
--- a/grpc/grpc-client/src/main/java/org/open/infc/grpc/client/OpenInterfaceGrpcClient.java
+++ b/grpc/grpc-client/src/main/java/org/open/infc/grpc/client/OpenInterfaceGrpcClient.java
@@ -37,6 +37,25 @@ public class OpenInterfaceGrpcClient {
private final ManagedChannel channel;
private final OpenInterfaceGrpc.OpenInterfaceBlockingStub blockingStub;
+ public static class OpenInterfaceGrpcExecption extends Exception {
+ private static final long serialVersionUID = -8755636432217894246L;
+
+ private int errorCode = -1;
+
+ public OpenInterfaceGrpcExecption(int errorCode, String message) {
+ super(message);
+ this.errorCode = errorCode;
+ }
+ }
+
+ public static class OpenInterfaceGrpcTimeoutExecption extends OpenInterfaceGrpcExecption {
+ private static int errorCode = 1;
+
+ public OpenInterfaceGrpcTimeoutExecption(String message) {
+ super(errorCode, message);
+ }
+ }
+
public OpenInterfaceGrpcClient(String host, int port) {
this(ManagedChannelBuilder.forAddress(host, port)
// Channels are secure by default (via SSL/TLS). For the example we disable TLS to avoid
@@ -54,27 +73,31 @@ public class OpenInterfaceGrpcClient {
channel.shutdown().awaitTermination(5, TimeUnit.SECONDS);
}
- public Output invoke(Input input) {
+ public Output invoke(Input input) throws OpenInterfaceGrpcTimeoutExecption {
logger.info("Input " + input.toString());
Output result = Output.newBuilder().build();
try {
- result = blockingStub.invoke(input);
+ result = blockingStub.withDeadlineAfter(10, TimeUnit.SECONDS).invoke(input);
} catch (StatusRuntimeException e) {
logger.warn("RPC failed: {0}", e.getStatus());
+ //Status{code=DEADLINE_EXCEEDED}
+ throw new OpenInterfaceGrpcTimeoutExecption(e.getMessage());
}
logger.info("Output: " + result.toString());
return result;
}
- public Result remoteCli(Args args) {
+ public Result remoteCli(Args args) throws OpenInterfaceGrpcTimeoutExecption {
logger.info(args.toString());
Result result = Result.newBuilder().setExitCode(1).build();
try {
- result = blockingStub.remoteCli(args);
+ result = blockingStub.withDeadlineAfter(10, TimeUnit.SECONDS).remoteCli(args);
} catch (StatusRuntimeException e) {
logger.warn("RPC failed: {0}", e.getStatus());
+ //Status{code=DEADLINE_EXCEEDED}
+ throw new OpenInterfaceGrpcTimeoutExecption(e.getMessage());
}
logger.info("Result: " + result.toString());