CLEANUP: map: slightly reorder the add map function

The function uses two distinct code paths for single the key/value pair
and multiple pairs inserted as payload, each with a copy-paste of the
error handling. Let's modify the loop to factor them out.
This commit is contained in:
Willy Tarreau 2021-04-29 16:02:48 +02:00
parent 4000ff0448
commit f7dd0e8796
1 changed files with 24 additions and 25 deletions

View File

@ -804,25 +804,24 @@ static int cli_parse_add_map(char **args, char *payload, struct appctx *appctx,
"This ACL is shared with a map containing samples. "
"You must use the command 'add map' to add values.\n");
}
/* Add value(s). */
err = NULL;
if (!payload) {
ret = map_add_key_value(appctx, args[3], args[4], &err);
if (!ret) {
if (err)
return cli_dynerr(appctx, memprintf(&err, "%s.\n", err));
else
return cli_err(appctx, "Failed to add an entry.\n");
}
}
else {
const char *end = payload + strlen(payload);
while (payload < end) {
char *key, *value;
/* Add value(s). If no payload is used, key and value are read
* from the command line and only one key is set. If a payload
* is passed, one key/value pair is read per line till the end
* of the payload is reached.
*/
err = NULL;
do {
char *key = args[3];
char *value = args[4];
size_t l;
/* key */
if (payload) {
/* key and value passed as payload, one pair per line */
if (!*payload)
break;
key = payload;
l = strcspn(key, " \t");
payload += l;
@ -841,6 +840,7 @@ static int cli_parse_add_map(char **args, char *payload, struct appctx *appctx,
if (*payload)
payload++;
value[l] = 0;
}
ret = map_add_key_value(appctx, key, value, &err);
if (!ret) {
@ -849,8 +849,7 @@ static int cli_parse_add_map(char **args, char *payload, struct appctx *appctx,
else
return cli_err(appctx, "Failed to add a key.\n");
}
}
}
} while (payload && *payload);
/* The add is done, send message. */
appctx->st0 = CLI_ST_PROMPT;