def __create_from_args_internal():
check(
is_p2(args.num_shards_per_chain),
"--num_shards_per_chain must be power of 2",
)
check(is_p2(args.num_slaves), "--num_slaves must be power of 2")
config = ClusterConfig()
config.LOG_LEVEL = args.log_level
config.DB_PATH_ROOT = args.db_path_root
config.P2P_PORT = args.p2p_port
config.JSON_RPC_PORT = args.json_rpc_port
config.PRIVATE_JSON_RPC_PORT = args.json_rpc_private_port
config.CLEAN = args.clean
config.START_SIMULATED_MINING = args.start_simulated_mining
config.ENABLE_TRANSACTION_HISTORY = args.enable_transaction_history
config.QUARKCHAIN.update(
args.num_chains,
args.num_shards_per_chain,
args.root_block_interval_sec,
args.minor_block_interval_sec,
)
config.QUARKCHAIN.NETWORK_ID = args.network_id
config.GENESIS_DIR = args.genesis_dir
config.MONITORING.KAFKA_REST_ADDRESS = args.monitoring_kafka_rest_address
if args.p2p:
config.SIMPLE_NETWORK = None
config.P2P = P2PConfig()
# p2p module
config.P2P.BOOT_NODES = args.bootnodes
config.P2P.PRIV_KEY = args.privkey
config.P2P.MAX_PEERS = args.max_peers
config.P2P.UPNP = args.upnp
else:
config.P2P = None
config.SIMPLE_NETWORK = SimpleNetworkConfig()
config.SIMPLE_NETWORK.BOOTSTRAP_HOST = (
args.simple_network_bootstrap_host)
config.SIMPLE_NETWORK.BOOTSTRAP_PORT = (
args.simple_network_bootstrap_port)
config.SLAVE_LIST = []
for i in range(args.num_slaves):
slave_config = SlaveConfig()
slave_config.PORT = args.port_start + i
slave_config.ID = "S{}".format(i)
slave_config.CHAIN_MASK_LIST = [ChainMask(i | args.num_slaves)]
config.SLAVE_LIST.append(slave_config)
fd, config.json_filepath = tempfile.mkstemp()
with os.fdopen(fd, "w") as tmp:
tmp.write(config.to_json())
return config
def __init__(self):
self.QUARKCHAIN = QuarkChainConfig()
self.MASTER = MasterConfig()
self.SLAVE_LIST = [] # type: List[SlaveConfig]
self.SIMPLE_NETWORK = SimpleNetworkConfig()
self._json_filepath = None
self.MONITORING = MonitoringConfig()
self.kafka_logger = KafkaSampleLogger(self)
slave_config = SlaveConfig()
slave_config.PORT = 38000
slave_config.ID = "S0"
slave_config.CHAIN_MASK_LIST = [ChainMask(1)]
self.SLAVE_LIST.append(slave_config)
fd, self.json_filepath = tempfile.mkstemp()
with os.fdopen(fd, "w") as tmp:
tmp.write(self.to_json())
def test_chain_mask(self):
sm0 = ChainMask(0b1)
self.assertTrue(sm0.contain_full_shard_id(0))
self.assertTrue(sm0.contain_full_shard_id(1 << 16))
self.assertTrue(sm0.contain_full_shard_id(0b1111111))
sm1 = ChainMask(0b101)
self.assertFalse(sm1.contain_full_shard_id(0))
self.assertTrue(sm1.contain_full_shard_id(0b1 << 16))
self.assertFalse(sm1.contain_full_shard_id(0b10 << 16))
self.assertFalse(sm1.contain_full_shard_id(0b11 << 16))
self.assertFalse(sm1.contain_full_shard_id(0b100 << 16))
self.assertTrue(sm1.contain_full_shard_id(0b101 << 16))
self.assertFalse(sm1.contain_full_shard_id(0b110 << 16))
self.assertFalse(sm1.contain_full_shard_id(0b111 << 16))
self.assertFalse(sm1.contain_full_shard_id(0b1000 << 16))
self.assertTrue(sm1.contain_full_shard_id(0b1001 << 16))
def create_test_clusters(
num_cluster,
genesis_account,
chain_size,
shard_size,
num_slaves,
genesis_root_heights,
remote_mining=False,
small_coinbase=False,
loadtest_accounts=None,
connect=True, # connect the bootstrap node by default
should_set_gas_price_limit=False,
mblock_coinbase_amount=None,
):
# so we can have lower minimum diff
easy_diff_calc = EthDifficultyCalculator(
cutoff=45, diff_factor=2048, minimum_diff=10
)
bootstrap_port = get_next_port() # first cluster will listen on this port
cluster_list = []
loop = asyncio.get_event_loop()
for i in range(num_cluster):
env = get_test_env(
genesis_account,
genesis_minor_quarkash=1000000,
chain_size=chain_size,
shard_size=shard_size,
genesis_root_heights=genesis_root_heights,
remote_mining=remote_mining,
)
env.cluster_config.P2P_PORT = bootstrap_port if i == 0 else get_next_port()
env.cluster_config.JSON_RPC_PORT = get_next_port()
env.cluster_config.PRIVATE_JSON_RPC_PORT = get_next_port()
env.cluster_config.SIMPLE_NETWORK = SimpleNetworkConfig()
env.cluster_config.SIMPLE_NETWORK.BOOTSTRAP_PORT = bootstrap_port
env.quark_chain_config.loadtest_accounts = loadtest_accounts or []
if should_set_gas_price_limit:
env.quark_chain_config.MIN_TX_POOL_GAS_PRICE = 10
env.quark_chain_config.MIN_MINING_GAS_PRICE = 10
if small_coinbase:
# prevent breaking previous tests after tweaking default rewards
env.quark_chain_config.ROOT.COINBASE_AMOUNT = 5
for c in env.quark_chain_config.shards.values():
c.COINBASE_AMOUNT = 5
if mblock_coinbase_amount is not None:
for c in env.quark_chain_config.shards.values():
c.COINBASE_AMOUNT = mblock_coinbase_amount
env.cluster_config.SLAVE_LIST = []
check(is_p2(num_slaves))
for j in range(num_slaves):
slave_config = SlaveConfig()
slave_config.ID = "S{}".format(j)
slave_config.PORT = get_next_port()
slave_config.CHAIN_MASK_LIST = [ChainMask(num_slaves | j)]
env.cluster_config.SLAVE_LIST.append(slave_config)
slave_server_list = []
for j in range(num_slaves):
slave_env = env.copy()
slave_env.db = InMemoryDb()
slave_env.slave_config = env.cluster_config.get_slave_config(
"S{}".format(j)
)
slave_server = SlaveServer(slave_env, name="cluster{}_slave{}".format(i, j))
slave_server.start()
slave_server_list.append(slave_server)
root_state = RootState(env, diff_calc=easy_diff_calc)
master_server = MasterServer(env, root_state, name="cluster{}_master".format(i))
master_server.start()
# Wait until the cluster is ready
loop.run_until_complete(master_server.cluster_active_future)
# Substitute diff calculate with an easier one
for slave in slave_server_list:
for shard in slave.shards.values():
shard.state.diff_calc = easy_diff_calc
# Start simple network and connect to seed host
network = SimpleNetwork(env, master_server, loop)
network.start_server()
if connect and i != 0:
peer = call_async(network.connect("127.0.0.1", bootstrap_port))
else:
peer = None
cluster_list.append(Cluster(master_server, slave_server_list, network, peer))
return cluster_list
def from_dict(cls, d):
config = super().from_dict(d)
config.CHAIN_MASK_LIST = [ChainMask(v) for v in config.CHAIN_MASK_LIST]
return config
