From 0a7d8e864091fc6ea645257f6205ec72e8ba537c Mon Sep 17 00:00:00 2001
From: Volodymyr Babak <volodymyr.babak@gmail.com>
Date: Fri, 9 Oct 2020 15:50:05 +0300
Subject: [PATCH] Disconnect client on error

---
 .../main/java/org/thingsboard/edge/rpc/EdgeGrpcClient.java   | 5 +++++
 1 file changed, 5 insertions(+)

diff --git 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
index 6232029a14..650fb89130 100644
--- 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));
             }