32 行
776 B
Python
32 行
776 B
Python
import asyncio
|
|
import sys
|
|
|
|
|
|
async def copy_stream(reader, writer):
|
|
try:
|
|
while data := await reader.read(65536):
|
|
writer.write(data)
|
|
await writer.drain()
|
|
except (ConnectionError, asyncio.CancelledError):
|
|
pass
|
|
finally:
|
|
writer.close()
|
|
|
|
|
|
async def connect_database(reader, writer):
|
|
try:
|
|
database_reader, database_writer = await asyncio.open_unix_connection(sys.argv[1])
|
|
except OSError:
|
|
writer.close()
|
|
return
|
|
await asyncio.gather(copy_stream(reader, database_writer), copy_stream(database_reader, writer))
|
|
|
|
|
|
async def main():
|
|
server = await asyncio.start_server(connect_database, "127.0.0.1", 25432)
|
|
async with server:
|
|
await server.serve_forever()
|
|
|
|
|
|
asyncio.run(main())
|