mirror of
https://github.com/0glabs/0g-storage-node.git
synced 2024-12-28 17:25:18 +00:00
90 lines
2.9 KiB
Python
90 lines
2.9 KiB
Python
import os
|
|
import subprocess
|
|
import tempfile
|
|
|
|
from test_framework.blockchain_node import BlockChainNodeType, BlockchainNode
|
|
from utility.utils import blockchain_rpc_port, arrange_port
|
|
from utility.build_binary import build_zg
|
|
|
|
ZGNODE_PORT_CATEGORY_WS = 0
|
|
ZGNODE_PORT_CATEGORY_P2P = 1
|
|
ZGNODE_PORT_CATEGORY_RPC = 2
|
|
ZGNODE_PORT_CATEGORY_PPROF = 3
|
|
|
|
def zg_node_init_genesis(binary: str, root_dir: str, num_nodes: int):
|
|
assert num_nodes > 0, "Invalid number of blockchain nodes: %s" % num_nodes
|
|
|
|
if not os.path.exists(binary):
|
|
build_zg(os.path.dirname(binary))
|
|
|
|
shell_script = os.path.join(
|
|
os.path.dirname(os.path.realpath(__file__)), # test_framework folder
|
|
"..", "config", "0gchain-init-genesis.sh"
|
|
)
|
|
|
|
zgchaind_dir = os.path.join(root_dir, "0gchaind")
|
|
os.mkdir(zgchaind_dir)
|
|
|
|
log_file = tempfile.NamedTemporaryFile(dir=zgchaind_dir, delete=False, prefix="init_genesis_", suffix=".log")
|
|
p2p_port_start = arrange_port(ZGNODE_PORT_CATEGORY_P2P, 0)
|
|
|
|
ret = subprocess.run(
|
|
args=["bash", shell_script, zgchaind_dir, str(num_nodes), str(p2p_port_start)],
|
|
stdout=log_file,
|
|
stderr=log_file,
|
|
)
|
|
|
|
log_file.close()
|
|
|
|
assert ret.returncode == 0, "Failed to init 0gchain genesis, see more details in log file: %s" % log_file.name
|
|
|
|
class ZGNode(BlockchainNode):
|
|
def __init__(
|
|
self,
|
|
index,
|
|
root_dir,
|
|
binary,
|
|
updated_config,
|
|
contract_path,
|
|
log,
|
|
rpc_timeout=10,
|
|
):
|
|
data_dir = os.path.join(root_dir, "0gchaind", "node" + str(index))
|
|
rpc_url = "http://127.0.0.1:%s" % blockchain_rpc_port(index)
|
|
|
|
super().__init__(
|
|
index,
|
|
data_dir,
|
|
rpc_url,
|
|
binary,
|
|
{},
|
|
contract_path,
|
|
log,
|
|
BlockChainNodeType.ZG,
|
|
rpc_timeout,
|
|
)
|
|
|
|
self.config_file = None
|
|
self.args = [
|
|
binary, "start",
|
|
"--home", data_dir,
|
|
# overwrite json rpc http port: 8545
|
|
"--json-rpc.address", "127.0.0.1:%s" % blockchain_rpc_port(index),
|
|
# overwrite json rpc ws port: 8546
|
|
"--json-rpc.ws-address", "127.0.0.1:%s" % arrange_port(ZGNODE_PORT_CATEGORY_WS, index),
|
|
# overwrite p2p port: 26656
|
|
"--p2p.laddr", "tcp://127.0.0.1:%s" % arrange_port(ZGNODE_PORT_CATEGORY_P2P, index),
|
|
# overwrite rpc port: 26657
|
|
"--rpc.laddr", "tcp://127.0.0.1:%s" % arrange_port(ZGNODE_PORT_CATEGORY_RPC, index),
|
|
# overwrite pprof port: 6060
|
|
"--rpc.pprof_laddr", "127.0.0.1:%s" % arrange_port(ZGNODE_PORT_CATEGORY_PPROF, index),
|
|
]
|
|
|
|
for k, v in updated_config.items():
|
|
if type(k) == str and k.startswith("--"):
|
|
self.args.append(k)
|
|
self.args.append(str(v))
|
|
|
|
def setup_config(self):
|
|
""" Already batch initialized by shell script in framework """
|