Skip to content

Commit f576e30

Browse files
author
Kim Gustyr
committed
implement providable session in rest_producer
1 parent ceb3fc0 commit f576e30

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

asyncnsq/http/rest_producer.py

+5-4
Original file line numberDiff line numberDiff line change
@@ -7,9 +7,10 @@
77
class NsqHTTPProducer(BaseNsqProducer):
88

99
def __init__(self, nsqd_http_addresses, selector_factory=RandomSelector,
10-
loop=None):
10+
client_session=None, loop=None):
1111
self._endpoints = nsqd_http_addresses
12-
self._loop = loop or asyncio.get_event_loop()
12+
self._client_session = client_session
13+
self._loop = loop or getattr(client_session, '_loop', None) or asyncio.get_event_loop()
1314
self._selector = selector_factory()
1415
self._connections = {}
1516

@@ -21,9 +22,9 @@ def _get_connection(self):
2122
def connect(self):
2223
for endpoint in set(self._endpoints):
2324
if len(endpoint) == 2:
24-
conn = Nsqd(*endpoint, loop=self._loop)
25+
conn = Nsqd(*endpoint, loop=self._loop, session=self._client_session)
2526
else:
26-
conn = Nsqd(base_url=endpoint, loop=self._loop)
27+
conn = Nsqd(base_url=endpoint, loop=self._loop, session=self._client_session)
2728
self._connections[conn.endpoint] = conn
2829

2930
async def publish(self, topic, message):

0 commit comments

Comments
 (0)