39 lines
1.1 KiB
Python
39 lines
1.1 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
from .base import RedisSubscriber
|
|
|
|
if TYPE_CHECKING:
|
|
from app.router.chat.server import ChatServer
|
|
|
|
|
|
JOIN_CHANNEL = "chat:room:joined"
|
|
EXIT_CHANNEL = "chat:room:left"
|
|
|
|
|
|
class ChatSubscriber(RedisSubscriber):
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.room_subscriber: dict[int, list[int]] = {}
|
|
self.chat_server: "ChatServer | None" = None
|
|
|
|
async def start_subscribe(self):
|
|
await self.subscribe(JOIN_CHANNEL)
|
|
self.add_handler(JOIN_CHANNEL, self.on_join_room)
|
|
await self.subscribe(EXIT_CHANNEL)
|
|
self.add_handler(EXIT_CHANNEL, self.on_leave_room)
|
|
self.start()
|
|
|
|
async def on_join_room(self, c: str, s: str):
|
|
channel_id, user_id = s.split(":")
|
|
if self.chat_server is None:
|
|
return
|
|
await self.chat_server.join_room_channel(int(channel_id), int(user_id))
|
|
|
|
async def on_leave_room(self, c: str, s: str):
|
|
channel_id, user_id = s.split(":")
|
|
if self.chat_server is None:
|
|
return
|
|
await self.chat_server.leave_room_channel(int(channel_id), int(user_id))
|