diff --git a/awx/main/dispatch/__init__.py b/awx/main/dispatch/__init__.py index 7fa4bd06f1..bbfd38b404 100644 --- a/awx/main/dispatch/__init__.py +++ b/awx/main/dispatch/__init__.py @@ -31,7 +31,7 @@ class PubSub(object): cur.execute('SELECT pg_notify(%s, %s);', (channel, payload)) def events(self, select_timeout=5, yield_timeouts=False): - if not pg_connection.get_autocommit(): + if not self.conn.autocommit: raise RuntimeError('Listening for events can only be done in autocommit mode') while True: diff --git a/awx/main/dispatch/control.py b/awx/main/dispatch/control.py index b1eb2281c9..cb7fabda94 100644 --- a/awx/main/dispatch/control.py +++ b/awx/main/dispatch/control.py @@ -46,7 +46,7 @@ class Control(object): reply_queue = Control.generate_reply_queue_name() self.result = None - with pg_bus_conn() as conn: + with pg_bus_conn(new_connection=True) as conn: conn.listen(reply_queue) conn.notify(self.queuename, json.dumps({'control': command, 'reply_to': reply_queue}))