#188: add/discard message.reply_channel on generic group_send (#189)

This commit is contained in:
thewayiam 2016-06-03 07:52:35 +08:00 committed by Andrew Godwin
parent f8debafbd3
commit 6eaee8f522

View File

@ -49,7 +49,7 @@ class WebsocketConsumer(BaseConsumer):
need to call super() all the time.
"""
for group in self.connection_groups(**kwargs):
Group(group, channel_layer=message.channel_layer).add(message.channel)
Group(group, channel_layer=message.channel_layer).add(message.reply_channel)
self.connect(message, **kwargs)
def connect(self, message, **kwargs):
@ -99,7 +99,7 @@ class WebsocketConsumer(BaseConsumer):
need to call super() all the time.
"""
for group in self.connection_groups(**kwargs):
Group(group, channel_layer=message.channel_layer).discard(message.channel)
Group(group, channel_layer=message.channel_layer).discard(message.reply_channel)
self.disconnect(message, **kwargs)
def disconnect(self, message, **kwargs):