From 220789daa6ec8e4be7c9373baefef8324c02e05b Mon Sep 17 00:00:00 2001
From: davebshow <davebshow@gmail.com>
Date: Sun, 9 Aug 2015 11:26:34 -0400
Subject: [PATCH] added GremlinServerError in try except to release socket
---
aiogremlin/client.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/aiogremlin/client.py b/aiogremlin/client.py
index c93c06f..c690977 100644
--- a/aiogremlin/client.py
+++ b/aiogremlin/client.py
@@ -6,7 +6,7 @@ import uuid
import aiohttp
from aiogremlin.response import GremlinClientWebSocketResponse
-from aiogremlin.exceptions import RequestError
+from aiogremlin.exceptions import RequestError, GremlinServerError
from aiogremlin.connector import GremlinConnector
from aiogremlin.subprotocol import gremlin_response_parser, GremlinWriter
@@ -320,7 +320,7 @@ class GremlinResponseStream:
asyncio.Task(self._ws.receive(), loop=self._loop)
try:
message = yield from self._stream.read()
- except RequestError:
+ except (RequestError, GremlinServerError):
yield from self._ws.release()
raise
return message
--
GitLab