diff --git a/main.py b/main.py index 83ba67c..50cf968 100644 --- a/main.py +++ b/main.py @@ -7,7 +7,9 @@ from src.recv_handler import recv_handler from src.send_handler import send_handler from src.config import global_config from src.mmc_com_layer import mmc_start_com, mmc_stop_com, router -from src.message_queue import message_queue, put_response, check_timeout_response +from src.response_pool import put_response, check_timeout_response + +message_queue = asyncio.Queue() async def message_recv(server_connection: Server.ServerConnection): diff --git a/src/recv_handler.py b/src/recv_handler.py index 2c02c54..7e031ec 100644 --- a/src/recv_handler.py +++ b/src/recv_handler.py @@ -28,7 +28,7 @@ from .utils import ( get_stranger_info, get_message_detail, ) -from .message_queue import get_response +from .response_pool import get_response class RecvHandler: diff --git a/src/message_queue.py b/src/response_pool.py similarity index 97% rename from src/message_queue.py rename to src/response_pool.py index 3720590..1fa592c 100644 --- a/src/message_queue.py +++ b/src/response_pool.py @@ -6,8 +6,6 @@ from .logger import logger response_dict: Dict = {} response_time_dict: Dict = {} -message_queue = asyncio.Queue() - async def get_response(request_id: str) -> dict: retry_count = 0 diff --git a/src/send_handler.py b/src/send_handler.py index 87ca64d..baf6fe7 100644 --- a/src/send_handler.py +++ b/src/send_handler.py @@ -12,7 +12,7 @@ from typing import Dict, Any, Tuple from . import CommandType from .config import global_config -from .message_queue import get_response +from .response_pool import get_response from .logger import logger from .utils import get_image_format, convert_image_to_gif diff --git a/src/utils.py b/src/utils.py index f85ad9a..fb0b5b4 100644 --- a/src/utils.py +++ b/src/utils.py @@ -3,7 +3,7 @@ import json import base64 import uuid from .logger import logger -from .message_queue import get_response +from .response_pool import get_response import urllib3 import ssl