database/postgres
database/postgres
1
0
Fork 0
mirror of https://github.com/postgres/postgres.git synced 2025-04-22 23:02:54 +03:00
Code

pgbench: accept unambiguous builtin prefixes for -b

Browse Source 
This makes it easier to use "-b se" instead of typing the full "-b
select-only".

Author: Fabien Coelho
Reviewed-by: Michaël Paquier
This commit is contained in:
Alvaro Herrera 2016-03-03 19:37:13 -03:00
parent 2c83f435a3
commit d561f1caec
2 changed files with 20 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
doc/src/sgml/ref/pgbench.sgml
View File

@ -269,6 +269,7 @@ pgbench <optional> <replaceable>options</> </optional> <replaceable>dbname</>
Add the specified builtin script to the list of executed scripts.
Available builtin scripts are: <literal>tpcb-like</>,
<literal>simple-update</> and <literal>select-only</>.
Unambiguous prefixes of builtin names are accepted.
With special name <literal>list</>, show the list of builtin scripts
and exit immediately.
</para>

22
src/bin/pgbench/pgbench.c
View File

@ -2746,22 +2746,36 @@ listAvailableScripts(void)
fprintf(stderr, "\n");
}
/* return builtin script "name" if unambiguous */
static char *
findBuiltin(const char *name, char **desc)
{
int i;
int i,
found = 0,
len = strlen(name);
char *commands = NULL;
for (i = 0; i < N_BUILTIN; i++)
{
if (strncmp(builtin_script[i].name, name,
strlen(builtin_script[i].name)) == 0)
if (strncmp(builtin_script[i].name, name, len) == 0)
{
*desc = builtin_script[i].desc;
return builtin_script[i].commands;
commands = builtin_script[i].commands;
found++;
}
}
/* ok, unambiguous result */
if (found == 1)
return commands;
/* error cases */
if (found == 0)
fprintf(stderr, "no builtin script found for name \"%s\"\n", name);
else /* found > 1 */
fprintf(stderr,
"ambiguous builtin name: %d builtin scripts found for prefix \"%s\"\n", found, name);
listAvailableScripts();
exit(1);
}