arno/gns3-server
1
0
Fork 0
mirror of https://github.com/GNS3/gns3-server synced 2025-05-31 13:18:51 +00:00
Code Issues Releases Wiki Activity

Fix issue with notification queue that prevented to properly close projects. Fix #1493

Browse Source
This commit is contained in:
grossmj 2019-01-12 16:02:36 +07:00
parent 161c05a310
commit a896346c77
4 changed files with 34 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
gns3server/controller/notification.py
View File

@ -17,7 +17,6 @@
import os
import aiohttp
import asyncio
from contextlib import contextmanager
from ..notification_queue import NotificationQueue
@ -43,7 +42,9 @@ class Notification:
queue = NotificationQueue()
self._project_listeners.setdefault(project.id, set())
self._project_listeners[project.id].add(queue)
try:
yield queue
finally:
self._project_listeners[project.id].remove(queue)
@contextmanager
@ -55,7 +56,9 @@ class Notification:
"""
queue = NotificationQueue()
self._controller_listeners.append(queue)
try:
yield queue
finally:
self._controller_listeners.remove(queue)
def controller_emit(self, action, event):

4
gns3server/handlers/api/compute/notification_handler.py
View File

@ -44,8 +44,9 @@ class NotificationHandler:
request.app['websockets'].add(ws)
asyncio.ensure_future(process_websocket(ws))
with notifications.queue() as queue:
try:
with notifications.queue() as queue:
while True:
notification = await queue.get_json(1)
if ws.closed:
@ -55,4 +56,5 @@ class NotificationHandler:
if not ws.closed:
await ws.close()
request.app['websockets'].discard(ws)
return ws

2
gns3server/handlers/api/controller/notification_handler.py
View File

@ -67,8 +67,8 @@ class NotificationHandler:
request.app['websockets'].add(ws)
asyncio.ensure_future(process_websocket(ws))
with controller.notification.controller_queue() as queue:
try:
with controller.notification.controller_queue() as queue:
while True:
notification = await queue.get_json(5)
if ws.closed:

2
gns3server/handlers/api/controller/project_handler.py
View File

@ -259,8 +259,8 @@ class ProjectHandler:
request.app['websockets'].add(ws)
asyncio.ensure_future(process_websocket(ws))
with controller.notification.project_queue(project) as queue:
try:
with controller.notification.project_queue(project) as queue:
while True:
notification = await queue.get_json(5)
if ws.closed: