modify logging

This commit is contained in:
kale4eat
2024-03-05 15:04:27 +09:00
parent 85f5b9bd25
commit 98ff976ec0
3 changed files with 11 additions and 5 deletions

View File

@@ -3,6 +3,8 @@ import socket
from .worker_common import RequestType, receive_data, send_data
from common.log import logger
class WorkerClient:
def __init__(self, port: int) -> None:
@@ -28,8 +30,12 @@ class WorkerClient:
"args": args,
"kwargs": kwargs,
}
logger.trace(f"client sends request: {data}")
send_data(self.sock, data)
return receive_data(self.sock).get("return")
logger.trace("client sent request successfully")
response = receive_data(self.sock)
logger.trace(f"client received response: {response}")
return response.get("return")
def status(self):
send_data(self.sock, {"request-type": RequestType.STATUS})