From 64932b4dae621b10d4f6fd37927e5e1fb61275fb Mon Sep 17 00:00:00 2001
From: Benjamin Braatz <benjamin.braatz@graph-it.com>
Date: Wed, 14 Jul 2021 14:48:40 +0200
Subject: [PATCH] Handle ConnectionResetError.

---
 controlpi_plugins/graph.py | 20 +++++++++++++-------
 1 file changed, 13 insertions(+), 7 deletions(-)

diff --git a/controlpi_plugins/graph.py b/controlpi_plugins/graph.py
index 9bb9e47..c4fbb0a 100644
--- a/controlpi_plugins/graph.py
+++ b/controlpi_plugins/graph.py
@@ -45,11 +45,7 @@ class Graph(BasePlugin):
         banner_size = await self._reader.readexactly(4)
         banner_size = struct.unpack('<i', banner_size)[0]
         banner_message = await self._reader.readexactly(banner_size)
-        # Get node instance:
         self._call_id = 0
-        self._coroot_guid = await self._call('attributsknoten',
-                                             ['coroot_name',
-                                              self.conf['name']])
 
     async def _call(self, method, params):
         await self._lock.acquire()
@@ -60,9 +56,15 @@ class Graph(BasePlugin):
                    'params': params, 'id': self._call_id}
         message = msgpack.packb(request)
         size = struct.pack('<i', len(message))
-        self._writer.write(size)
-        self._writer.write(message)
-        await self._writer.drain()
+        success = False
+        while not success:
+            self._writer.write(size)
+            self._writer.write(message)
+            try:
+                await self._writer.drain()
+                success = True
+            except ConnectionResetError:
+                await self._open_connection()
         size = await self._reader.readexactly(4)
         size = struct.unpack('<i', size)[0]
         message = await self._reader.readexactly(size)
@@ -105,3 +107,7 @@ class Graph(BasePlugin):
         """Open connection and get coroot instance for name."""
         self._lock = asyncio.Lock()
         await self._open_connection()
+        # Get coroot instance:
+        self._coroot_guid = await self._call('attributsknoten',
+                                             ['coroot_name',
+                                              self.conf['name']])
-- 
2.34.1