from channels.generic.websocket import AsyncJsonWebsocketConsumer


class HealthConsumer(AsyncJsonWebsocketConsumer):
    group_name = "healthcheck"

    async def connect(self):
        try:
            await self.channel_layer.group_add(self.group_name, self.channel_name)
            await self.channel_layer.group_discard(self.group_name, self.channel_name)
        except Exception:
            await self.close(code=1011)
            return

        await self.accept()
        await self.send_json({"status": "ok", "channel_layer": "ok"})
        await self.close()