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

UDP...

Browse Source
This commit is contained in:
Maxim Khomutov 2023-07-18 04:10:56 +03:00
parent 3d33eec5fd
commit 041883644c
2 changed files with 23 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
src/core/udp_server.py
View File

@ -10,42 +10,36 @@ import traceback
from core import utils
class UDPServer:
class UDPServer(asyncio.DatagramTransport):
def __init__(self, core, host, port):
def __init__(self, core, host=None, port=None):
super().__init__()
self.log = utils.get_logger("UDPServer")
self.loop = asyncio.get_event_loop()
self.Core = core
self.host = host
self.port = port
self.run = False
self.transport = None
async def handle_client(self, reader, writer):
while True:
try:
data = await reader.read(1)
if not data:
break
code = data.decode()
self.log.debug(f"Received {code!r} from {writer.get_extra_info('sockname')!r}")
# await self.handle_code(code, reader, writer)
# 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:
self.log.error("Error while connecting..")
self.log.error(f"Error: {e}")
traceback.print_exc()
break
def connection_made(self, transport):
self.transport = transport
def datagram_received(self, data, addr):
message = data.decode()
print('Received %r from %s' % (message, addr))
print('Send %r to %s' % (message, addr))
self.transport.sendto(data, addr)
async def start(self):
self.log.debug("Starting UDP server.")
self.run = True
try:
pass
self.transport, _ = await self.loop.create_datagram_endpoint(
lambda: self,
local_addr=(self.host, self.port),
reuse_port=True
)
except OSError as e:
self.log.error("Cannot bind port or other error")
raise e
@ -58,3 +52,4 @@ class UDPServer:
def stop(self):
self.log.debug("Stopping UDP server")
self.transport.close()

8
src/core/udp_server.pyi
View File

@ -5,20 +5,22 @@
# Licence: FPA
# (c) kuitoi.su 2023
import asyncio
from typing import Tuple
from core import utils
class UDPServer:
def __init__(self, core, host, port):
def __init__(self, core, host=None, port=None):
self.log = utils.get_logger("UDPServer")
self.loop = asyncio.get_event_loop()
self.Core = core
self.host = host
self.port = port
self.run = False
async def handle_client(self, srv_sock) -> None: ...
self.transport = None
def connection_made(self, transport: asyncio.DatagramTransport): ...
def datagram_received(self, data: bytes, addr: Tuple[str, int]): ...
async def start(self) -> None: ...
async def stop(self) -> None: ...