47 lines
1.3 KiB
Python
47 lines
1.3 KiB
Python
import asyncio
|
|
|
|
|
|
class FollowingResult:
|
|
def __init__(self, event, kind, payload):
|
|
self.event = event
|
|
self.kind = kind
|
|
self.payload = payload
|
|
|
|
|
|
class Following:
|
|
queue = asyncio.Queue()
|
|
|
|
def __init__(self, kind, uid):
|
|
self.kind = kind # author topic shout chat
|
|
self.uid = uid
|
|
|
|
|
|
class FollowingManager:
|
|
lock = asyncio.Lock()
|
|
data = {"author": [], "topic": [], "shout": [], "chat": []}
|
|
|
|
@staticmethod
|
|
async def register(kind, uid):
|
|
async with FollowingManager.lock:
|
|
FollowingManager[kind].append(uid)
|
|
|
|
@staticmethod
|
|
async def remove(kind, uid):
|
|
async with FollowingManager.lock:
|
|
FollowingManager[kind].remove(uid)
|
|
|
|
@staticmethod
|
|
async def push(kind, payload):
|
|
try:
|
|
async with FollowingManager.lock:
|
|
if kind == "chat":
|
|
for chat in FollowingManager["chat"]:
|
|
if payload.message["chatId"] == chat.uid:
|
|
chat.queue.put_nowait(payload)
|
|
else:
|
|
for entity in FollowingManager[kind]:
|
|
if payload.shout["createdBy"] == entity.uid:
|
|
entity.queue.put_nowait(payload)
|
|
except Exception as e:
|
|
print(Exception(e))
|