django/daphne
1
1
Fork 0
mirror of https://github.com/django/daphne.git synced 2024-11-21 23:46:33 +03:00
Code Issues Packages Projects Releases Wiki Activity

Send HTTP disconnect messages

Browse Source
This commit is contained in:
Andrew Godwin 2016-02-20 22:48:31 +00:00
parent f1c2a3b9f7
commit b9412f8e82
1 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
daphne/http_protocol.py
View File

@ -97,11 +97,21 @@ class WebRequest(http.Request):
"server": [self.host.host, self.host.port],
})
def send_disconnect(self):
"""
Sends a disconnect message on the http.disconnect channel.
Useful only really for long-polling.
"""
self.factory.channel_layer.send("http.disconnect", {
"reply_channel": self.reply_channel,
})
def connectionLost(self, reason):
"""
Cleans up reply channel on close.
"""
if self.reply_channel and self.reply_channel in self.channel.factory.reply_protocols:
self.send_disconnect()
del self.channel.factory.reply_protocols[self.reply_channel]
logger.debug("HTTP disconnect for %s", self.reply_channel)
http.Request.connectionLost(self, reason)
@ -110,7 +120,8 @@ class WebRequest(http.Request):
"""
Cleans up reply channel on close.
"""
if self.reply_channel:
if self.reply_channel and self.reply_channel in self.channel.factory.reply_protocols:
self.send_disconnect()
del self.channel.factory.reply_protocols[self.reply_channel]
logger.debug("HTTP close for %s", self.reply_channel)
http.Request.finish(self)