discours.io/core
5
0
Fork 0
Code Pull Requests Actions Projects Activity

minor changes

Browse Source
This commit is contained in:
knst-kotov 2021-07-08 10:58:49 +03:00
parent a70421f243
commit cd8af4d761
1 changed files with 13 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
resolvers/inbox.py
View File

@ -7,9 +7,13 @@ from auth.authenticate import login_required
import asyncio import asyncio
new_message_queue = asyncio.Queue()
updated_message_queue = asyncio.Queue() class MessageQueue:
deleted_message_queue = asyncio.Queue()
new_message = asyncio.Queue()
updated_message = asyncio.Queue()
deleted_message = asyncio.Queue()
@mutation.field("createMessage") @mutation.field("createMessage")
@login_required @login_required
@ -23,7 +27,7 @@ async def create_message(_, info, input):
replyTo = input.get("replyTo") replyTo = input.get("replyTo")
) )
new_message_queue.put_nowait(new_message) MessageQueue.new_message.put_nowait(new_message)
return { return {
"status": True, "status": True,
@ -69,7 +73,7 @@ async def update_message(_, info, input):
message.body = input["body"] message.body = input["body"]
global_session.commit() global_session.commit()
updated_message_queue.put_nowait(message) MessageQueue.updated_message.put_nowait(message)
return { return {
"status" : True, "status" : True,
@ -93,7 +97,7 @@ async def delete_message(_, info, id):
global_session.delete(message) global_session.delete(message)
global_session.commit() global_session.commit()
deleted_message_queue.put_nowait(message) MessageQueue.deleted_message.put_nowait(message)
return { return {
"status" : True "status" : True
@ -103,19 +107,19 @@ async def delete_message(_, info, id):
@subscription.source("messageCreated") @subscription.source("messageCreated")
async def new_message_generator(obj, info): async def new_message_generator(obj, info):
while True: while True:
new_message = await new_message_queue.get() new_message = await MessageQueue.new_message.get()
yield new_message yield new_message
@subscription.source("messageUpdated") @subscription.source("messageUpdated")
async def updated_message_generator(obj, info): async def updated_message_generator(obj, info):
while True: while True:
message = await updated_message_queue.get() message = await MessageQueue.updated_message.get()
yield message yield message
@subscription.source("messageDeleted") @subscription.source("messageDeleted")
async def deleted_message_generator(obj, info): async def deleted_message_generator(obj, info):
while True: while True:
message = await deleted_message_queue.get() message = await MessageQueue.deleted_message.get()
yield new_message yield new_message
@subscription.field("messageCreated") @subscription.field("messageCreated")