2022-09-30 11:40:22 +03:00
|
|
|
from __future__ import annotations
|
|
|
|
|
2021-02-18 20:00:25 +03:00
|
|
|
import logging
|
2022-08-23 06:20:55 +03:00
|
|
|
import os
|
2020-10-09 01:56:00 +03:00
|
|
|
import pathlib
|
2022-07-13 19:43:48 +03:00
|
|
|
import sys
|
2020-06-06 01:20:09 +03:00
|
|
|
from multiprocessing import freeze_support
|
2022-07-13 19:43:48 +03:00
|
|
|
from typing import Dict, List, Optional, Tuple
|
2020-01-13 16:19:53 +03:00
|
|
|
|
2021-04-04 06:55:26 +03:00
|
|
|
from chia.consensus.constants import ConsensusConstants
|
|
|
|
from chia.consensus.default_constants import DEFAULT_CONSTANTS
|
|
|
|
from chia.full_node.full_node import FullNode
|
|
|
|
from chia.full_node.full_node_api import FullNodeAPI
|
|
|
|
from chia.rpc.full_node_rpc_api import FullNodeRpcApi
|
|
|
|
from chia.server.outbound_message import NodeType
|
2022-07-13 19:43:48 +03:00
|
|
|
from chia.server.start_service import RpcInfo, Service, async_run
|
2022-09-13 00:57:31 +03:00
|
|
|
from chia.util.chia_logging import initialize_service_logging
|
2022-07-08 03:57:08 +03:00
|
|
|
from chia.util.config import load_config, load_config_cli
|
2021-04-04 06:55:26 +03:00
|
|
|
from chia.util.default_root import DEFAULT_ROOT_PATH
|
2022-07-13 19:43:48 +03:00
|
|
|
from chia.util.ints import uint16
|
2022-11-06 01:46:40 +03:00
|
|
|
from chia.util.task_timing import maybe_manage_task_instrumentation
|
2019-07-26 11:23:50 +03:00
|
|
|
|
2020-06-13 00:37:19 +03:00
|
|
|
# See: https://bugs.python.org/issue29288
|
2020-11-30 10:29:15 +03:00
|
|
|
"".encode("idna")
|
2020-05-26 05:28:53 +03:00
|
|
|
|
2020-10-12 21:53:05 +03:00
|
|
|
SERVICE_NAME = "full_node"
|
2021-02-18 20:00:25 +03:00
|
|
|
log = logging.getLogger(__name__)
|
2020-10-12 21:53:05 +03:00
|
|
|
|
2020-05-26 05:28:53 +03:00
|
|
|
|
2022-07-13 19:43:48 +03:00
|
|
|
def create_full_node_service(
|
|
|
|
root_path: pathlib.Path,
|
|
|
|
config: Dict,
|
|
|
|
consensus_constants: ConsensusConstants,
|
|
|
|
connect_to_daemon: bool = True,
|
|
|
|
override_capabilities: List[Tuple[uint16, str]] = None,
|
2022-09-22 06:49:01 +03:00
|
|
|
) -> Service[FullNode]:
|
2022-07-08 03:57:08 +03:00
|
|
|
service_config = config[SERVICE_NAME]
|
2021-02-18 20:00:25 +03:00
|
|
|
|
2021-02-11 05:42:49 +03:00
|
|
|
full_node = FullNode(
|
2022-07-08 03:57:08 +03:00
|
|
|
service_config,
|
2021-02-11 05:42:49 +03:00
|
|
|
root_path=root_path,
|
2021-02-24 03:54:50 +03:00
|
|
|
consensus_constants=consensus_constants,
|
2021-02-11 05:42:49 +03:00
|
|
|
)
|
2020-10-16 04:03:46 +03:00
|
|
|
api = FullNodeAPI(full_node)
|
|
|
|
|
2020-11-30 10:29:15 +03:00
|
|
|
upnp_list = []
|
2022-07-08 03:57:08 +03:00
|
|
|
if service_config["enable_upnp"]:
|
|
|
|
upnp_list = [service_config["port"]]
|
|
|
|
network_id = service_config["selected_network"]
|
2022-07-13 19:43:48 +03:00
|
|
|
rpc_info: Optional[RpcInfo] = None
|
|
|
|
if service_config["start_rpc_server"]:
|
|
|
|
rpc_info = (FullNodeRpcApi, service_config["rpc_port"])
|
|
|
|
return Service(
|
2020-05-26 05:28:53 +03:00
|
|
|
root_path=root_path,
|
2022-07-08 03:57:08 +03:00
|
|
|
config=config,
|
2020-10-16 04:03:46 +03:00
|
|
|
node=api.full_node,
|
|
|
|
peer_api=api,
|
2020-05-26 05:28:53 +03:00
|
|
|
node_type=NodeType.FULL_NODE,
|
2022-07-08 03:57:08 +03:00
|
|
|
advertised_port=service_config["port"],
|
2020-10-12 21:53:05 +03:00
|
|
|
service_name=SERVICE_NAME,
|
2020-11-13 07:30:55 +03:00
|
|
|
upnp_ports=upnp_list,
|
2022-07-08 03:57:08 +03:00
|
|
|
server_listen_ports=[service_config["port"]],
|
2020-12-01 11:22:58 +03:00
|
|
|
on_connect_callback=full_node.on_connect,
|
2021-03-11 02:43:02 +03:00
|
|
|
network_id=network_id,
|
2022-07-13 19:43:48 +03:00
|
|
|
rpc_info=rpc_info,
|
|
|
|
connect_to_daemon=connect_to_daemon,
|
|
|
|
override_capabilities=override_capabilities,
|
2020-05-20 21:26:58 +03:00
|
|
|
)
|
2019-10-23 10:49:16 +03:00
|
|
|
|
2019-11-18 07:50:31 +03:00
|
|
|
|
2022-07-13 19:43:48 +03:00
|
|
|
async def async_main() -> int:
|
2022-07-08 03:57:08 +03:00
|
|
|
# TODO: refactor to avoid the double load
|
|
|
|
config = load_config(DEFAULT_ROOT_PATH, "config.yaml")
|
|
|
|
service_config = load_config_cli(DEFAULT_ROOT_PATH, "config.yaml", SERVICE_NAME)
|
|
|
|
config[SERVICE_NAME] = service_config
|
|
|
|
overrides = service_config["network_overrides"]["constants"][service_config["selected_network"]]
|
2021-02-24 03:54:50 +03:00
|
|
|
updated_constants = DEFAULT_CONSTANTS.replace_str_to_bytes(**overrides)
|
2022-09-13 00:57:31 +03:00
|
|
|
initialize_service_logging(service_name=SERVICE_NAME, config=config)
|
2022-07-13 19:43:48 +03:00
|
|
|
service = create_full_node_service(DEFAULT_ROOT_PATH, config, updated_constants)
|
|
|
|
await service.setup_process_global_state()
|
|
|
|
await service.run()
|
2020-04-04 01:56:45 +03:00
|
|
|
|
2022-07-13 19:43:48 +03:00
|
|
|
return 0
|
2020-04-04 01:56:45 +03:00
|
|
|
|
2022-07-13 19:43:48 +03:00
|
|
|
|
|
|
|
def main() -> int:
|
2020-06-06 01:20:09 +03:00
|
|
|
freeze_support()
|
2022-08-23 06:20:55 +03:00
|
|
|
|
2022-11-06 01:46:40 +03:00
|
|
|
with maybe_manage_task_instrumentation(enable=os.environ.get("CHIA_INSTRUMENT_NODE") is not None):
|
|
|
|
return async_run(async_main())
|
2022-07-13 19:43:48 +03:00
|
|
|
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
sys.exit(main())
|