summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--cli/src/cli-cmd-misc.c10
-rw-r--r--cli/src/cli-cmd-volume.c50
-rw-r--r--cli/src/cli-cmd.h5
-rw-r--r--cli/src/cli.h4
-rw-r--r--cli/src/input.c22
5 files changed, 69 insertions, 22 deletions
diff --git a/cli/src/cli-cmd-misc.c b/cli/src/cli-cmd-misc.c
index 306a7628f..7833456b7 100644
--- a/cli/src/cli-cmd-misc.c
+++ b/cli/src/cli-cmd-misc.c
@@ -61,8 +61,10 @@ cli_cmd_display_help (struct cli_state *state, struct cli_cmd_word *in_word,
for (cmd = cli_probe_cmds; cmd->pattern; cmd++)
cli_out ("%s - %s", cmd->pattern, cmd->desc);
- for (cmd = cli_misc_cmds; cmd->pattern; cmd++)
- cli_out ("%s - %s", cmd->pattern, cmd->desc);
+ for (cmd = cli_misc_cmds; cmd->pattern; cmd++) {
+ if (cmd->cbk)
+ cli_out ("%s - %s", cmd->pattern, cmd->desc);
+ }
if (!state->rl_enabled)
exit (0);
@@ -79,6 +81,10 @@ struct cli_cmd cli_misc_cmds[] = {
cli_cmd_display_help,
"display command options"},
+ { "mode script",
+ NULL,
+ "assign gluster execution mode"},
+
{ NULL, NULL, NULL }
};
diff --git a/cli/src/cli-cmd-volume.c b/cli/src/cli-cmd-volume.c
index ce8310557..01fff9bf1 100644
--- a/cli/src/cli-cmd-volume.c
+++ b/cli/src/cli-cmd-volume.c
@@ -236,6 +236,24 @@ out:
return ret;
}
+gf_answer_t
+cli_cmd_get_confirmation (struct cli_state *state, const char *question)
+{
+ char answer = '\0';
+ char flush = '\0';
+
+ if (GLUSTER_MODE_SCRIPT == state->mode)
+ return GF_ANSWER_YES;
+ printf ("%s (y/n) ", question);
+ answer = getchar ();
+ flush = answer;
+ while ('\n' != flush)
+ flush = getchar ();
+ if ('y' != answer) {
+ return GF_ANSWER_NO;
+ }
+ return GF_ANSWER_YES;
+}
int
cli_cmd_volume_stop_cbk (struct cli_state *state, struct cli_cmd_word *word,
@@ -246,8 +264,10 @@ cli_cmd_volume_stop_cbk (struct cli_state *state, struct cli_cmd_word *word,
call_frame_t *frame = NULL;
int flags = 0;
gf1_cli_stop_vol_req req = {0,};
- char answer;
- char flush;
+ gf_answer_t answer = GF_ANSWER_NO;
+
+ const char *question = "Stopping volume will make its data inaccessible. "
+ "Do you want to Continue?";
frame = create_frame (THIS, THIS->ctx->pool);
if (!frame)
@@ -272,13 +292,9 @@ cli_cmd_volume_stop_cbk (struct cli_state *state, struct cli_cmd_word *word,
}
}
- printf ("Stopping volume will make its data inaccessible. "
- "Do you want to Continue? (y/n) ");
- answer = getchar ();
- flush = answer;
- while ('\n' != flush)
- flush = getchar ();
- if ('y' != answer) {
+ answer = cli_cmd_get_confirmation (state, question);
+
+ if (GF_ANSWER_NO == answer) {
ret = 0;
goto out;
}
@@ -473,8 +489,10 @@ cli_cmd_volume_remove_brick_cbk (struct cli_state *state,
rpc_clnt_procedure_t *proc = NULL;
call_frame_t *frame = NULL;
dict_t *options = NULL;
- char answer;
- char flush;
+ gf_answer_t answer = GF_ANSWER_NO;
+
+ const char *question = "Removing brick(s) can result in data loss. "
+ "Do you want to Continue?";
frame = create_frame (THIS, THIS->ctx->pool);
if (!frame)
@@ -487,13 +505,9 @@ cli_cmd_volume_remove_brick_cbk (struct cli_state *state,
goto out;
}
- printf ("Removing brick(s) can result in data loss. "
- "Do you want to Continue? (y/n) ");
- answer = getchar ();
- flush = answer;
- while ('\n' != flush)
- flush = getchar ();
- if ('y' != answer) {
+ answer = cli_cmd_get_confirmation (state, question);
+
+ if (GF_ANSWER_NO == answer) {
ret = 0;
goto out;
}
diff --git a/cli/src/cli-cmd.h b/cli/src/cli-cmd.h
index b7cd7aadc..7dda509dd 100644
--- a/cli/src/cli-cmd.h
+++ b/cli/src/cli-cmd.h
@@ -27,6 +27,11 @@
#include "cli.h"
+typedef enum {
+ GF_ANSWER_YES = 1,
+ GF_ANSWER_NO = 2
+} gf_answer_t;
+
struct cli_cmd {
const char *pattern;
cli_cmd_cbk_t *cbk;
diff --git a/cli/src/cli.h b/cli/src/cli.h
index fdf53421f..eaa638c29 100644
--- a/cli/src/cli.h
+++ b/cli/src/cli.h
@@ -39,6 +39,9 @@ enum argp_option_keys {
ARGP_PORT_KEY = 'p',
};
+enum gluster_mode {
+ GLUSTER_MODE_SCRIPT = 1
+};
struct cli_state;
struct cli_cmd_word;
struct cli_cmd_tree;
@@ -96,6 +99,7 @@ struct cli_state {
char **matchesp;
int remote_port;
+ int mode;
};
struct cli_local {
diff --git a/cli/src/input.c b/cli/src/input.c
index e762fbd13..2c67b354d 100644
--- a/cli/src/input.c
+++ b/cli/src/input.c
@@ -41,7 +41,11 @@ cli_batch (void *d)
state = d;
- ret = cli_cmd_process (state, state->argc, state->argv);
+ if (state->mode == GLUSTER_MODE_SCRIPT)
+ ret = cli_cmd_process (state, state->argc - 2, state->argv + 2);
+ else
+ ret = cli_cmd_process (state, state->argc, state->argv);
+
gf_log ("", GF_LOG_NORMAL, "Exiting with: %d", ret);
exit (ret);
@@ -82,8 +86,22 @@ int
cli_input_init (struct cli_state *state)
{
int ret = 0;
+ gf_boolean_t is_batch = _gf_false;
+
+ if (1 < state->argc) {
+ if (!strcmp ("mode", state->argv[0]) &&
+ !strcmp ("script", state->argv[1])) {
+ state->mode = GLUSTER_MODE_SCRIPT;
+ if (2 < state->argc)
+ is_batch = _gf_true;
+ } else {
+ is_batch = _gf_true;
+ }
+ } else if (1 == state->argc) {
+ is_batch = _gf_true;
+ }
- if (state->argc) {
+ if (is_batch) {
ret = pthread_create (&state->input, NULL, cli_batch, state);
return ret;
}