diff --git a/pybtc/connector/block_loader.py b/pybtc/connector/block_loader.py index 21552e3..ae53ff8 100644 --- a/pybtc/connector/block_loader.py +++ b/pybtc/connector/block_loader.py @@ -10,7 +10,8 @@ import sys import aiojsonrpc import traceback from pybtc.connector.utils import decode_block_tx -import pickle +# import pickle +import msgpack as pickle from lru import LRU diff --git a/pybtc/connector/connector.py b/pybtc/connector/connector.py index 2beaa0b..240ef23 100644 --- a/pybtc/connector/connector.py +++ b/pybtc/connector/connector.py @@ -12,7 +12,7 @@ import zmq import zmq.asyncio import asyncio import time -from pickle import loads, dumps +from msgpack import loads, dumps class Connector: def __init__(self, node_rpc_url, node_zerromq_url, logger, diff --git a/pybtc/test/block.py b/pybtc/test/block.py index 827d27e..754a9e8 100644 --- a/pybtc/test/block.py +++ b/pybtc/test/block.py @@ -619,6 +619,17 @@ class BlockDeserializeTests(unittest.TestCase): p = pickle.loads(k) print("decoded block load", time.time() - qt) print(p[0]["hash"]) + + qt = time.time() + k = pickle.dumps(bt, protocol = pickle.HIGHEST_PROTOCOL) + print("decoded block dump", time.time() - qt) + qt = time.time() + p = pickle.loads(k) + print("decoded block load", time.time() - qt) + print(p[0]["hash"]) + + + import cProfile # cProfile.run("import pybtc;"