Disconnect client on error

6 years ago · 0a7d8e8640
1 changed files with 5 additions and 0 deletions
--- a/common/edge-api/src/main/java/org/thingsboard/edge/rpc/EdgeGrpcClient.java
+++ b/common/edge-api/src/main/java/org/thingsboard/edge/rpc/EdgeGrpcClient.java
@ -133,6 +133,11 @@ public class EdgeGrpcClient implements EdgeRpcClient {
            @Override
            public void onError(Throwable t) {
                log.debug("[{}] The rpc session received an error!", edgeKey, t);
+                try {
+                    EdgeGrpcClient.this.disconnect(true);
+                } catch (InterruptedException e) {
+                    log.error("[{}] Got interruption during disconnect!", edgeKey, e);
+                }
                onError.accept(new RuntimeException(t));
            }