SantaSpeen/KuiToi-Server
1
0
Fork 0
mirror of https://github.com/kuitoi/kuitoi-Server.git synced 2025-08-18 00:35:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

Minor update

Browse Source
This commit is contained in:
Maxim Khomutov 2023-07-13 13:29:37 +03:00
parent cdf226ac5c
commit 13321fb9b5
2 changed files with 29 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

50
src/core/udp_server.py
View File

@ -14,43 +14,47 @@ class UDPServer:
def __init__(self, core, host, port): def __init__(self, core, host, port):
self.log = utils.get_logger("UDPServer") self.log = utils.get_logger("UDPServer")
self.loop = asyncio.get_event_loop()
self.Core = core self.Core = core
self.host = host self.host = host
self.port = port self.port = port
self.loop = asyncio.get_event_loop() self.run = False
async def handle_client(self, srv_sock): async def handle_client(self, reader, writer):
while True: while True:
try: try:
data, addr = await self.loop.sock_recv(srv_sock, 1024) data = await reader.read(1)
if not data: if not data:
break break
code = data.decode() code = data.decode()
self.log.debug(f"Received {code!r} from {addr!r}") self.log.debug(f"Received {code!r} from {writer.get_extra_info('sockname')!r}")
# if not await self.handle_code(code, sock): # await self.handle_code(code, reader, writer)
# break # task = asyncio.create_task(self.handle_code(code, reader, writer))
# await asyncio.wait([task], return_when=asyncio.FIRST_EXCEPTION)
if not writer.is_closing():
writer.close()
self.log.debug("Disconnected.")
break
except Exception as e: except Exception as e:
self.log.error("Error while connecting..")
self.log.error(f"Error: {e}") self.log.error(f"Error: {e}")
traceback.print_exc() traceback.print_exc()
break break
srv_sock.close()
self.log.error("Error while connecting..")
async def start(self): async def start(self):
pass self.log.debug("Starting UDP server.")
# self.log.debug("Starting UDP server.") self.run = True
# await self.stop() try:
# srv_sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) pass
# srv_sock.bind((self.host, self.port)) except OSError as e:
# self.log.debug(f"Serving on {srv_sock.getsockname()}") self.log.error("Cannot bind port or other error")
# try: raise e
# await self.handle_client(srv_sock) except BaseException as e:
# except Exception as e: self.log.error(f"Error: {e}")
# self.log.error(f"Error: {e}") raise e
# traceback.print_exc() finally:
# finally: self.run = False
# await self.stop() self.Core.run = False
def stop(self): def stop(self):
pass self.log.debug("Stopping UDP server")
# self.log.debug("Stopping UDP server")

3
src/core/udp_server.pyi
View File

@ -13,10 +13,11 @@ class UDPServer:
def __init__(self, core, host, port): def __init__(self, core, host, port):
self.log = utils.get_logger("UDPServer") self.log = utils.get_logger("UDPServer")
self.loop = asyncio.get_event_loop()
self.Core = core self.Core = core
self.host = host self.host = host
self.port = port self.port = port
self.loop = asyncio.get_event_loop() self.run = False
async def handle_client(self, srv_sock) -> None: ... async def handle_client(self, srv_sock) -> None: ...
async def start(self) -> None: ... async def start(self) -> None: ...