fix: 补齐候选诊断环境与证据保护(09-06-release-gate-integrity)
这个提交包含在:
@@ -0,0 +1,31 @@
|
||||
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())
|
||||
在新工单中引用
屏蔽一个用户