Skip to content

Commit 968f073

Browse files
authored
fix: use named artifacts (#69)
Closes #70
1 parent 094352d commit 968f073

File tree

7 files changed

+28
-28
lines changed

7 files changed

+28
-28
lines changed

src/forkmon/forkmon_launcher.star

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -36,20 +36,20 @@ def launch_forkmon(
3636
template_and_data_by_rel_dest_filepath = {}
3737
template_and_data_by_rel_dest_filepath[FORKMON_CONFIG_FILENAME] = template_and_data
3838

39-
config_files_artifact_uuid = plan.render_templates(template_and_data_by_rel_dest_filepath)
39+
config_files_artifact_name = plan.render_templates(template_and_data_by_rel_dest_filepath, "forkmon-config")
4040

41-
config = get_config(config_files_artifact_uuid)
41+
config = get_config(config_files_artifact_name)
4242

4343
plan.add_service(SERVICE_ID, config)
4444

4545

46-
def get_config(config_files_artifact_uuid):
46+
def get_config(config_files_artifact_name):
4747
config_file_path = shared_utils.path_join(FORKMON_CONFIG_MOUNT_DIRPATH_ON_SERVICE, FORKMON_CONFIG_FILENAME)
4848
return ServiceConfig(
4949
image = IMAGE_NAME,
5050
ports = USED_PORTS,
5151
files = {
52-
FORKMON_CONFIG_MOUNT_DIRPATH_ON_SERVICE: config_files_artifact_uuid,
52+
FORKMON_CONFIG_MOUNT_DIRPATH_ON_SERVICE: config_files_artifact_name,
5353
},
5454
cmd = ["--config-path", config_file_path]
5555
)

src/grafana/grafana_launcher.star

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -44,21 +44,21 @@ def get_grafana_config_dir_artifact_uuid(plan, datasource_config_template, dashb
4444
template_and_data_by_rel_dest_filepath[DATASOURCE_CONFIG_REL_FILEPATH] = datasource_template_and_data
4545
template_and_data_by_rel_dest_filepath[DASHBOARD_PROVIDERS_CONFIG_REL_FILEPATH] = dashboard_providers_template_and_data
4646

47-
grafana_config_artifacts_uuid = plan.render_templates(template_and_data_by_rel_dest_filepath)
47+
grafana_config_artifacts_name = plan.render_templates(template_and_data_by_rel_dest_filepath, name="grafana-config")
4848

49-
grafana_dashboards_artifacts_uuid = plan.upload_files(static_files.GRAFANA_DASHBOARDS_CONFIG_DIRPATH)
49+
grafana_dashboards_artifacts_name = plan.upload_files(static_files.GRAFANA_DASHBOARDS_CONFIG_DIRPATH, name="grafana-dashboards")
5050

51-
return grafana_config_artifacts_uuid, grafana_dashboards_artifacts_uuid
51+
return grafana_config_artifacts_name, grafana_dashboards_artifacts_name
5252

5353

54-
def get_config(grafana_config_artifacts_uuid, grafana_dashboards_artifacts_uuid):
54+
def get_config(grafana_config_artifacts_name, grafana_dashboards_artifacts_name):
5555
return ServiceConfig(
5656
image = IMAGE_NAME,
5757
ports = USED_PORTS,
5858
env_vars = {CONFIG_DIRPATH_ENV_VAR: GRAFANA_CONFIG_DIRPATH_ON_SERVICE},
5959
files = {
60-
GRAFANA_CONFIG_DIRPATH_ON_SERVICE: grafana_config_artifacts_uuid,
61-
GRAFANA_DASHBOARDS_DIRPATH_ON_SERVICE: grafana_dashboards_artifacts_uuid
60+
GRAFANA_CONFIG_DIRPATH_ON_SERVICE: grafana_config_artifacts_name,
61+
GRAFANA_DASHBOARDS_DIRPATH_ON_SERVICE: grafana_dashboards_artifacts_name
6262
}
6363
)
6464

src/participant_network/participant_network.star

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -76,12 +76,12 @@ def launch_participant_network(plan, participants, network_params, global_log_le
7676

7777
plan.print("Uploading GETH prefunded keys")
7878

79-
geth_prefunded_keys_artifact_id = plan.upload_files(static_files.GETH_PREFUNDED_KEYS_DIRPATH)
79+
geth_prefunded_keys_artifact_name = plan.upload_files(static_files.GETH_PREFUNDED_KEYS_DIRPATH, name="geth-prefunded-keys")
8080

8181
plan.print("Uploaded GETH files succesfully, launching EL participants")
8282

8383
el_launchers = {
84-
package_io.EL_CLIENT_TYPE.geth : {"launcher": geth.new_geth_launcher(network_params.network_id, el_genesis_data, geth_prefunded_keys_artifact_id, genesis_constants.PRE_FUNDED_ACCOUNTS), "launch_method": geth.launch},
84+
package_io.EL_CLIENT_TYPE.geth : {"launcher": geth.new_geth_launcher(network_params.network_id, el_genesis_data, geth_prefunded_keys_artifact_name, genesis_constants.PRE_FUNDED_ACCOUNTS), "launch_method": geth.launch},
8585
package_io.EL_CLIENT_TYPE.besu : {"launcher": besu.new_besu_launcher(network_params.network_id, el_genesis_data), "launch_method": besu.launch},
8686
package_io.EL_CLIENT_TYPE.erigon : {"launcher": erigon.new_erigon_launcher(network_params.network_id, el_genesis_data), "launch_method": erigon.launch},
8787
package_io.EL_CLIENT_TYPE.nethermind : {"launcher": nethermind.new_nethermind_launcher(el_genesis_data), "launch_method": nethermind.launch},

src/participant_network/prelaunch_data_generator/cl_genesis/cl_genesis_data_generator.star

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -56,13 +56,13 @@ def generate_cl_genesis_data(
5656
template_and_data_by_rel_dest_filepath[MNEMONICS_YML_FILENAME] = genesis_generation_mnemonics_template_and_data
5757
template_and_data_by_rel_dest_filepath[GENESIS_CONFIG_YML_FILENAME] = genesis_generation_config_template_and_data
5858

59-
genesis_generation_config_artifact_uuid = plan.render_templates(template_and_data_by_rel_dest_filepath)
59+
genesis_generation_config_artifact_name = plan.render_templates(template_and_data_by_rel_dest_filepath, "genesis-generation-config-cl")
6060

6161
# TODO(old) Make this the actual data generator - comment copied from the original module
6262
launcher_service_id = prelaunch_data_generator_launcher.launch_prelaunch_data_generator(
6363
plan,
6464
{
65-
CONFIG_DIRPATH_ON_GENERATOR: genesis_generation_config_artifact_uuid,
65+
CONFIG_DIRPATH_ON_GENERATOR: genesis_generation_config_artifact_name,
6666
EL_GENESIS_DIRPATH_ON_GENERATOR: el_genesis_data.files_artifact_uuid,
6767
},
6868
)
@@ -135,7 +135,7 @@ def generate_cl_genesis_data(
135135
genesis_generation_result = plan.exec(struct(service_id=launcher_service_id, command=cl_genesis_generation_cmd))
136136
plan.assert(genesis_generation_result["code"], "==", SUCCESSFUL_EXEC_CMD_EXIT_CODE)
137137

138-
cl_genesis_data_artifact_uuid = plan.store_service_files(launcher_service_id, OUTPUT_DIRPATH_ON_GENERATOR)
138+
cl_genesis_data_artifact_name = plan.store_service_files(launcher_service_id, OUTPUT_DIRPATH_ON_GENERATOR, name = "cl-genesis-data")
139139

140140
jwt_secret_rel_filepath = shared_utils.path_join(
141141
shared_utils.path_base(OUTPUT_DIRPATH_ON_GENERATOR),
@@ -150,7 +150,7 @@ def generate_cl_genesis_data(
150150
GENESIS_STATE_FILENAME,
151151
)
152152
result = cl_genesis_data.new_cl_genesis_data(
153-
cl_genesis_data_artifact_uuid,
153+
cl_genesis_data_artifact_name,
154154
jwt_secret_rel_filepath,
155155
genesis_config_rel_filepath,
156156
genesis_ssz_rel_filepath,

src/participant_network/prelaunch_data_generator/cl_validator_keystores/cl_validator_keystore_generator.star

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -72,12 +72,12 @@ def generate_cl_validator_keystores(
7272
# Store outputs into files artifacts
7373
keystore_files = []
7474
for idx, output_dirpath in enumerate(all_output_dirpaths):
75-
artifact_uuid = plan.store_service_files(service_id, output_dirpath)
75+
artifact_name = plan.store_service_files(service_id, output_dirpath, name = "validator-keystore-" + str(idx))
7676

7777
# This is necessary because the way Kurtosis currently implements artifact-storing is
7878
base_dirname_in_artifact = shared_utils.path_base(output_dirpath)
7979
to_add = keystore_files_module.new_keystore_files(
80-
artifact_uuid,
80+
artifact_name,
8181
shared_utils.path_join(base_dirname_in_artifact, RAW_KEYS_DIRNAME),
8282
shared_utils.path_join(base_dirname_in_artifact, RAW_SECRETS_DIRNAME),
8383
shared_utils.path_join(base_dirname_in_artifact, NIMBUS_KEYS_DIRNAME),
@@ -100,10 +100,10 @@ def generate_cl_validator_keystores(
100100
write_prysm_password_file_cmd_result = plan.exec(struct(service_id=service_id, command=write_prysm_password_file_cmd))
101101
plan.assert(write_prysm_password_file_cmd_result["code"], "==", SUCCESSFUL_EXEC_CMD_EXIT_CODE)
102102

103-
prysm_password_artifact_uuid = plan.store_service_files(service_id, PRYSM_PASSWORD_FILEPATH_ON_GENERATOR)
103+
prysm_password_artifact_name = plan.store_service_files(service_id, PRYSM_PASSWORD_FILEPATH_ON_GENERATOR, name = "prysm-password")
104104

105105
result = keystores_result.new_generate_keystores_result(
106-
prysm_password_artifact_uuid,
106+
prysm_password_artifact_name,
107107
shared_utils.path_base(PRYSM_PASSWORD_FILEPATH_ON_GENERATOR),
108108
keystore_files,
109109
)

src/participant_network/prelaunch_data_generator/el_genesis/el_genesis_data_generator.star

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -48,14 +48,14 @@ def generate_el_genesis_data(
4848
template_and_data_by_rel_dest_filepath = {}
4949
template_and_data_by_rel_dest_filepath[GENESIS_CONFIG_FILENAME] = genesis_config_file_template_and_data
5050

51-
genesis_generation_config_artifact_uuid = plan.render_templates(template_and_data_by_rel_dest_filepath)
51+
genesis_generation_config_artifact_name = plan.render_templates(template_and_data_by_rel_dest_filepath, name="genesis-generation-config-el")
5252

5353

5454
# TODO(old) Make this the actual data generator - comment copied from the original module
5555
launcher_service_id = prelaunch_data_generator_launcher.launch_prelaunch_data_generator(
5656
plan,
5757
{
58-
CONFIG_DIRPATH_ON_GENERATOR: genesis_generation_config_artifact_uuid,
58+
CONFIG_DIRPATH_ON_GENERATOR: genesis_generation_config_artifact_name,
5959
},
6060
)
6161

@@ -118,10 +118,10 @@ def generate_el_genesis_data(
118118
jwt_secret_generation_cmd_result = plan.exec(struct(service_id=launcher_service_id, command=jwt_secret_generation_cmd))
119119
plan.assert(jwt_secret_generation_cmd_result["code"], "==", SUCCESSFUL_EXEC_CMD_EXIT_CODE)
120120

121-
elGenesisDataArtifactUuid = plan.store_service_files(launcher_service_id, OUTPUT_DIRPATH_ON_GENERATOR)
121+
el_genesis_data_artifact_name = plan.store_service_files(launcher_service_id, OUTPUT_DIRPATH_ON_GENERATOR, name = "el-genesis-data")
122122

123123
result = el_genesis.new_el_genesis_data(
124-
elGenesisDataArtifactUuid,
124+
el_genesis_data_artifact_name,
125125
shared_utils.path_join(shared_utils.path_base(OUTPUT_DIRPATH_ON_GENERATOR), JWT_SECRET_FILENAME),
126126
genesis_filename_to_relative_filepath_in_artifact[GETH_GENESIS_FILENAME],
127127
genesis_filename_to_relative_filepath_in_artifact[ERIGON_GENESIS_FILENAME],

src/prometheus/prometheus_launcher.star

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -25,9 +25,9 @@ def launch_prometheus(plan, config_template, cl_client_contexts):
2525
template_and_data_by_rel_dest_filepath = {}
2626
template_and_data_by_rel_dest_filepath[CONFIG_FILENAME] = template_and_data
2727

28-
config_files_artifact_uuid = plan.render_templates(template_and_data_by_rel_dest_filepath)
28+
config_files_artifact_name = plan.render_templates(template_and_data_by_rel_dest_filepath, "prometheus-config")
2929

30-
config = get_config(config_files_artifact_uuid)
30+
config = get_config(config_files_artifact_name)
3131
prometheus_service = plan.add_service(SERVICE_ID, config)
3232

3333
private_ip_address = prometheus_service.ip_address
@@ -36,13 +36,13 @@ def launch_prometheus(plan, config_template, cl_client_contexts):
3636
return "http://{0}:{1}".format(private_ip_address, prometheus_service_http_port)
3737

3838

39-
def get_config(config_files_artifact_uuid):
39+
def get_config(config_files_artifact_name):
4040
config_file_path = shared_utils.path_join(CONFIG_DIR_MOUNTPOINT_ON_PROMETHEUS, shared_utils.path_base(CONFIG_FILENAME))
4141
return ServiceConfig(
4242
image = IMAGE_NAME,
4343
ports = USED_PORTS,
4444
files = {
45-
CONFIG_DIR_MOUNTPOINT_ON_PROMETHEUS: config_files_artifact_uuid
45+
CONFIG_DIR_MOUNTPOINT_ON_PROMETHEUS: config_files_artifact_name
4646
},
4747
cmd = [
4848
# You can check all the cli flags starting the container and going to the flags section

0 commit comments

Comments
 (0)