oci/zeronet
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Keep track of websocket connections

Browse source
This commit is contained in:
shortcutme 2017-11-05 23:40:59 +01:00
parent b39d86d020
commit 2781e31be0
No known key found for this signature in database
GPG key ID: 5B63BAE6CB9613AE
2 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/Ui/UiRequest.py
View file

@ -605,7 +605,9 @@ class UiRequest(object):
return self.error403() return self.error403()
ui_websocket = UiWebsocket(ws, site, self.server, user, self) ui_websocket = UiWebsocket(ws, site, self.server, user, self)
site.websockets.append(ui_websocket) # Add to site websockets to allow notify on events site.websockets.append(ui_websocket) # Add to site websockets to allow notify on events
self.server.websockets.append(ui_websocket)
ui_websocket.start() ui_websocket.start()
self.server.websockets.remove(ui_websocket)
for site_check in self.server.sites.values(): for site_check in self.server.sites.values():
# Remove websocket from every site (admin sites allowed to join other sites event channels) # Remove websocket from every site (admin sites allowed to join other sites event channels)
if ui_websocket in site_check.websockets: if ui_websocket in site_check.websockets:

1
src/Ui/UiServer.py
View file

@ -70,6 +70,7 @@ class UiServer:
self.wrapper_nonces = [] self.wrapper_nonces = []
self.add_nonces = [] self.add_nonces = []
self.websockets = []
self.site_manager = SiteManager.site_manager self.site_manager = SiteManager.site_manager
self.sites = SiteManager.site_manager.list() self.sites = SiteManager.site_manager.list()
self.log = logging.getLogger(__name__) self.log = logging.getLogger(__name__)