diff --git a/pybtc/connector/block_loader.py b/pybtc/connector/block_loader.py index ae53ff8..b767b13 100644 --- a/pybtc/connector/block_loader.py +++ b/pybtc/connector/block_loader.py @@ -10,8 +10,8 @@ import sys import aiojsonrpc import traceback from pybtc.connector.utils import decode_block_tx -# import pickle -import msgpack as pickle + +import _pickle as pickle from lru import LRU diff --git a/pybtc/connector/connector.py b/pybtc/connector/connector.py index 240ef23..23563dc 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 msgpack import loads, dumps +from _pickle 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 754a9e8..da63c53 100644 --- a/pybtc/test/block.py +++ b/pybtc/test/block.py @@ -603,15 +603,8 @@ class BlockDeserializeTests(unittest.TestCase): bt = ( Block(fc[:-1], format="raw", keep_raw_tx=False), ) - print("decoded block", time.time() - qt ) - import msgpack - qt = time.time() - k = msgpack.dumps(bt) - print("decoded block dump", time.time() - qt) - qt = time.time() - p = msgpack.loads(k) - print("decoded block load", time.time() - qt) - import pickle + + import _pickle as pickle qt = time.time() k = pickle.dumps(bt) print("decoded block dump", time.time() - qt) @@ -620,14 +613,6 @@ class BlockDeserializeTests(unittest.TestCase): 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