Update output message of restart command to include container id (#3335)
diff --git a/heron/tools/cli/src/python/restart.py b/heron/tools/cli/src/python/restart.py
index 010c6f8..9b731fa 100644
--- a/heron/tools/cli/src/python/restart.py
+++ b/heron/tools/cli/src/python/restart.py
@@ -66,9 +66,13 @@
Log.debug("Restart Args: %s", cl_args)
container_id = cl_args['container-id']
+ message = "restart topology"
+ if container_id >= 0:
+ message = "restart container " + str(container_id) + " of topology"
+
if cl_args['deploy_mode'] == config.SERVER_MODE:
dict_extra_args = {"container_id": str(container_id)}
- return cli_helper.run_server(command, cl_args, "restart topology", extra_args=dict_extra_args)
+ return cli_helper.run_server(command, cl_args, message, extra_args=dict_extra_args)
else:
list_extra_args = ["--container_id", str(container_id)]
- return cli_helper.run_direct(command, cl_args, "restart topology", extra_args=list_extra_args)
+ return cli_helper.run_direct(command, cl_args, message, extra_args=list_extra_args)