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

Skip UDP Server

Browse Source
This commit is contained in:
Maxim Khomutov 2023-07-08 02:03:56 +03:00
parent 7fa3004ec1
commit 90ca3f7700
2 changed files with 17 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
src/core/udp_server.py
View File

@ -5,7 +5,6 @@
# Licence: FPA # Licence: FPA
# (c) kuitoi.su 2023 # (c) kuitoi.su 2023
import asyncio import asyncio
import socket
import traceback import traceback
from core import utils from core import utils
@ -18,12 +17,12 @@ class UDPServer:
self.Core = core self.Core = core
self.host = host self.host = host
self.port = port self.port = port
self.loop = asyncio.get_event_loop()
async def handle_client(self, srv_sock): async def handle_client(self, srv_sock):
loop = asyncio.get_event_loop()
while True: while True:
try: try:
data, addr = await loop.sock_recv(srv_sock, 1024) data, addr = await self.loop.sock_recv(srv_sock, 1024)
if not data: if not data:
break break
code = data.decode() code = data.decode()
@ -34,21 +33,22 @@ class UDPServer:
self.log.error(f"Error: {e}") self.log.error(f"Error: {e}")
traceback.print_exc() traceback.print_exc()
break break
# srv_sock.close() srv_sock.close()
self.log.error("Error while connecting..") self.log.error("Error while connecting..")
async def start(self): async def start(self):
self.log.debug("Starting UDP server.") self.log.debug("Starting UDP server.")
srv_sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) await self.stop()
srv_sock.bind((self.host, self.port)) # srv_sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
self.log.debug(f"Serving on {srv_sock.getsockname()}") # srv_sock.bind((self.host, self.port))
try: # self.log.debug(f"Serving on {srv_sock.getsockname()}")
await self.handle_client(srv_sock) # try:
except Exception as e: # await self.handle_client(srv_sock)
self.log.error(f"Error: {e}") # except Exception as e:
traceback.print_exc() # self.log.error(f"Error: {e}")
finally: # traceback.print_exc()
await self.stop() # finally:
# await self.stop()
async def stop(self): async def stop(self):
self.log.debug("Stopping UDP server") self.log.debug("Stopping UDP server")

3
src/core/udp_server.pyi
View File

@ -4,6 +4,8 @@
# Version 0.0 # Version 0.0
# Licence: FPA # Licence: FPA
# (c) kuitoi.su 2023 # (c) kuitoi.su 2023
import asyncio
from core import utils from core import utils
@ -14,6 +16,7 @@ class UDPServer:
self.Core = core self.Core = core
self.host = host self.host = host
self.port = port self.port = port
self.loop = asyncio.get_event_loop()
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: ...