RanchiMall/pyflo
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

connector

Browse Source
This commit is contained in:
4tochka 2019-05-20 23:52:48 +03:00
parent 181b44d7af
commit 6be3f0a68a
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
pybtc/connector/block_loader.py
View File

@ -115,7 +115,7 @@ class BlockLoader:
out_writer.close() out_writer.close()
# get stream reader # get stream reader
worker.reader = await self.get_pipe_reader(out_reader) worker.reader = await self.get_pipe_reader(out_reader)
worker.writer = in_writer worker.writer = await self.get_pipe_writer(in_writer)
worker.name = str(index) worker.name = str(index)
self.worker[index] = worker self.worker[index] = worker
self.worker_busy[index] = False self.worker_busy[index] = False
@ -137,6 +137,14 @@ class BlockLoader:
return None return None
return reader return reader
async def get_pipe_writer(self, fd_writer):
try:
wt, wp = await self.loop.connect_write_pipe(asyncio.streams.FlowControlMixin, fd_writer)
writer = asyncio.streams.StreamWriter(wt, wp, None, self.loop)
except:
return None
return writer
async def pipe_get_msg(self, reader): async def pipe_get_msg(self, reader):
while True: while True:
try: try:
@ -159,7 +167,7 @@ class BlockLoader:
msg = msg_type + msg msg = msg_type + msg
msg = b''.join((b'ME', len(msg).to_bytes(4, byteorder='little'), msg)) msg = b''.join((b'ME', len(msg).to_bytes(4, byteorder='little'), msg))
writer.write(msg) writer.write(msg)
# writer.flush() await writer.drain()
@ -359,6 +367,6 @@ class Worker:
msg = msg_type + msg msg = msg_type + msg
msg = b''.join((b'ME', len(msg).to_bytes(4, byteorder='little'), msg)) msg = b''.join((b'ME', len(msg).to_bytes(4, byteorder='little'), msg))
self.out_writer.write(msg) self.out_writer.write(msg)
# self.out_writer.flush() self.out_writer.flush()