mirror of https://github.com/nocturn9x/giambio.git
58 lines
1.6 KiB
Python
58 lines
1.6 KiB
Python
import sys
|
|
from typing import Tuple
|
|
import giambio
|
|
import logging
|
|
|
|
|
|
async def sender(sock: giambio.socket.AsyncSocket, q: giambio.Queue):
|
|
while True:
|
|
await sock.send_all(b"yo")
|
|
await q.put((0, ""))
|
|
|
|
|
|
async def receiver(sock: giambio.socket.AsyncSocket, q: giambio.Queue):
|
|
data = b""
|
|
buffer = b""
|
|
while True:
|
|
while not data.endswith(b"\n"):
|
|
data += await sock.receive(1024)
|
|
data, rest = data.split(b"\n", maxsplit=2)
|
|
buffer = b"".join(rest)
|
|
await q.put((1, data.decode()))
|
|
data = buffer
|
|
|
|
|
|
async def main(host: Tuple[str, int]):
|
|
"""
|
|
Main client entry point
|
|
"""
|
|
|
|
queue = giambio.Queue()
|
|
async with giambio.create_pool() as pool:
|
|
async with giambio.socket.socket() as sock:
|
|
await sock.connect(host)
|
|
await pool.spawn(sender, sock, queue)
|
|
await pool.spawn(receiver, sock, queue)
|
|
while True:
|
|
op, data = await queue.get()
|
|
if op == 0:
|
|
print(f"Sent.")
|
|
else:
|
|
print(f"Received: {data}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
port = int(sys.argv[1]) if len(sys.argv) > 1 else 1501
|
|
logging.basicConfig(
|
|
level=20,
|
|
format="[%(levelname)s] %(asctime)s %(message)s",
|
|
datefmt="%d/%m/%Y %p",
|
|
)
|
|
try:
|
|
giambio.run(main, ("localhost", port))
|
|
except (Exception, KeyboardInterrupt) as error: # Exceptions propagate!
|
|
if isinstance(error, KeyboardInterrupt):
|
|
logging.info("Ctrl+C detected, exiting")
|
|
else:
|
|
logging.error(f"Exiting due to a {type(error).__name__}: {error}")
|