From 090c35fa358852bc3e51ed2e9e53b9f335c9ea9f Mon Sep 17 00:00:00 2001
From: Leifur Halldor Asgeirsson <lasgeirsson@zerofail.com>
Date: Thu, 1 Sep 2016 12:30:39 -0400
Subject: [PATCH] allow passing traversal_source to Goblin.open
---
goblin/app.py | 6 ++++--
goblin/driver/connection.py | 1 +
2 files changed, 5 insertions(+), 2 deletions(-)
diff --git a/goblin/app.py b/goblin/app.py
index 1b2eec2..8f87bb7 100644
--- a/goblin/app.py
+++ b/goblin/app.py
@@ -58,9 +58,11 @@ class Goblin:
self._get_hashable_id = get_hashable_id
@classmethod
- async def open(cls, loop, *, get_hashable_id=None, **config):
+ async def open(cls, loop, *, get_hashable_id=None, traversal_source=None,
+ **config):
cluster = await driver.Cluster.open(loop, **config)
- app = Goblin(cluster, get_hashable_id=get_hashable_id)
+ app = Goblin(cluster, get_hashable_id=get_hashable_id,
+ traversal_source=traversal_source)
await app.supports_transactions()
return app
diff --git a/goblin/driver/connection.py b/goblin/driver/connection.py
index 65c7245..2196390 100644
--- a/goblin/driver/connection.py
+++ b/goblin/driver/connection.py
@@ -208,6 +208,7 @@ class Connection(AbstractConnection):
request_id = str(uuid.uuid4())
message = self._message_serializer.serialize_message(
request_id, processor, op, **args)
+ logger.debug(message)
response_queue = asyncio.Queue(loop=self._loop)
self._response_queues[request_id] = response_queue
if self._ws.closed:
--
GitLab