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-07 03:24:07 +04:00
parent 5d116adab2
commit c29877c1ae
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
pybtc/connector/block_loader.py
View File

@ -8,6 +8,7 @@ import signal
import sys import sys
import aiojsonrpc import aiojsonrpc
import traceback import traceback
import pickle
class BlockLoader: class BlockLoader:
def __init__(self, parent, workers=4): def __init__(self, parent, workers=4):
self.worker = dict() self.worker = dict()
@ -54,7 +55,7 @@ class BlockLoader:
self.worker_busy[i] = True self.worker_busy[i] = True
try: try:
self.log.warning("<<<<<") self.log.warning("<<<<<")
self.pipe_sent_msg(self.worker[i].writer, b'get', batch) self.pipe_sent_msg(self.worker[i].writer, b'get', pickle.dumps(batch))
self.log.warning("ok<") self.log.warning("ok<")
except: except:
self.log.warning(str(traceback.format_exc())) self.log.warning(str(traceback.format_exc()))
@ -92,7 +93,7 @@ class BlockLoader:
def pipe_sent_msg(self, writer, msg_type, msg): def pipe_sent_msg(self, writer, msg_type, msg):
msg_type = msg_type[:20].ljust(20) msg_type = msg_type[:20].ljust(20)
msg = msg_type + msg.encode() 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() writer.flush()