[sheepdog] [PATCH v3 2/2] collie: add a new option --progress to "node recovery" for showing recovery progress

[Hitoshi Mitake]

This patch adds a new option --progress (or -P) to the node recovery
subcommand. With this subcommand, users can show a progress of
recovery process.

Example:
$ sudo collie node recovery --progress
 99.7 % [==============================================>] 7047 / 7068

The denominator (7068 in the above case) indicates a number of entire
object which should be checked. The numerator (7047 in the above case)
indicates a number of objects which is already checked or copied.

Signed-off-by: Hitoshi Mitake <mitake.hitoshi at lab.ntt.co.jp>
---
v3:
 - clean coding style

v2:
 - make this feature as an option of "node recovery", not a new subcommand
 - clean coding style
 -- renaming recovery_progress_unit() -> get_recovery_progress()

 collie/node.c |   91 +++++++++++++++++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 89 insertions(+), 2 deletions(-)

diff --git a/collie/node.c b/collie/node.c
index 4230af5..19b5508 100644
--- a/collie/node.c
+++ b/collie/node.c
@@ -13,6 +13,7 @@
 
 static struct node_cmd_data {
 	bool all_nodes;
+	bool recovery_progress;
 } node_cmd_data;
 
 static void cal_total_vdi_size(uint32_t vid, const char *name, const char *tag,
@@ -120,10 +121,92 @@ static int node_info(int argc, char **argv)
 	return EXIT_SUCCESS;
 }
 
+static int get_recovery_state(struct recovery_state *state)
+{
+	int ret;
+	struct sd_req req;
+
+	sd_init_req(&req, SD_OP_STAT_RECOVERY);
+	req.data_length = sizeof(*state);
+
+	ret = collie_exec_req(sdhost, sdport, &req, state);
+	if (ret < 0) {
+		fprintf(stderr, "Failed to execute request\n");
+		return -1;
+	}
+
+	return 0;
+}
+
+static int node_recovery_progress(void)
+{
+	int result, prev_in_recovery = 0;
+
+	/*
+	 * prev_in_recovery is required for expressing state transition.
+	 * If the variable is 0 and obtained state indicates not in recovery,
+	 * node wasn't doing recovery from first.
+	 */
+
+	/*
+	 * ToDos
+	 *
+	 * 1. Calculate size of actually copied objects.
+	 *    For doing this, not so trivial changes for recovery process are
+	 *    required.
+	 *
+	 * 2. Print remaining physical time.
+	 *    Even if it is not so acculate, the information is helpful for
+	 *    administrators.
+	 */
+
+	do {
+		struct recovery_state rstate;
+
+		result = get_recovery_state(&rstate);
+		if (result < 0)
+			break;
+
+		if (!rstate.in_recovery) {
+			if (prev_in_recovery)
+				/* not an immediate completion */
+				show_progress(rstate.nr_total, rstate.nr_total,
+					true);
+
+			break;
+		}
+
+		prev_in_recovery = rstate.in_recovery;
+
+		switch (rstate.state) {
+		case RW_PREPARE_LIST:
+			printf("\rpreparing a checked object list...");
+			break;
+		case RW_NOTIFY_COMPLETION:
+			printf("\rnotifying a completion of recovery...");
+			break;
+		case RW_RECOVER_OBJ:
+			show_progress(rstate.nr_finished, rstate.nr_total,
+				true);
+			break;
+		default:
+			panic("unknown state of recovery: %d", rstate.state);
+			break;
+		}
+
+		sleep(1);
+	} while (true);
+
+	return result < 0 ? EXIT_SYSFAIL : EXIT_SUCCESS;
+}
+
 static int node_recovery(int argc, char **argv)
 {
 	int i, ret;
 
+	if (node_cmd_data.recovery_progress)
+		return node_recovery_progress();
+
 	if (!raw_output) {
 		printf("Nodes In Recovery:\n");
 		printf("  Id   Host:Port         V-Nodes       Zone\n");
@@ -315,6 +398,9 @@ static int node_parser(int ch, char *opt)
 	case 'A':
 		node_cmd_data.all_nodes = true;
 		break;
+	case 'P':
+		node_cmd_data.recovery_progress = true;
+		break;
 	}
 
 	return 0;
@@ -322,6 +408,7 @@ static int node_parser(int ch, char *opt)
 
 static struct sd_option node_options[] = {
 	{'A', "all", false, "show md information of all the nodes"},
+	{'P', "progress", false, "show progress of recovery in the node"},
 
 	{ 0, NULL, false, NULL },
 };
@@ -333,8 +420,8 @@ static struct subcommand node_cmd[] = {
 	 SUBCMD_FLAG_NEED_NODELIST, node_list},
 	{"info", NULL, "aprh", "show information about each node", NULL,
 	 SUBCMD_FLAG_NEED_NODELIST, node_info},
-	{"recovery", NULL, "aprh", "show nodes in recovery", NULL,
-	 SUBCMD_FLAG_NEED_NODELIST, node_recovery},
+	{"recovery", NULL, "aphP", "show recovery information of nodes", NULL,
+	 SUBCMD_FLAG_NEED_NODELIST, node_recovery, node_options},
 	{"md", "[disks]", "apAh", "See 'collie node md' for more information",
 	 node_md_cmd, SUBCMD_FLAG_NEED_ARG, node_md, node_options},
 	{NULL,},
-- 
1.7.10.4