From 57e6db706e81fd2609fa385677e6ae72471822fe Mon Sep 17 00:00:00 2001 From: Michael Paquier Date: Wed, 3 Mar 2021 10:14:21 +0900 Subject: [PATCH] Add --tablespace option to reindexdb This option provides REINDEX (TABLESPACE) for reindexdb, applying the tablespace value given by the caller to all the REINDEX queries generated. While on it, this commit adds some tests for REINDEX TABLESPACE, with and without CONCURRENTLY, when run on toast indexes and tables. Such operations are not allowed, and toast relation names are not stable enough to be part of the main regression test suite (even if using a PL function with a TRY/CATCH logic, as CONCURRENTLY could not be tested). Author: Michael Paquier Reviewed-by: Mark Dilger, Daniel Gustafsson Discussion: https://postgr.es/m/YDiaDMnzLICqeukl@paquier.xyz --- doc/src/sgml/ref/reindexdb.sgml | 10 +++ src/bin/scripts/reindexdb.c | 109 ++++++++++++++++++++--------- src/bin/scripts/t/090_reindexdb.pl | 81 ++++++++++++++++++++- 3 files changed, 163 insertions(+), 37 deletions(-) diff --git a/doc/src/sgml/ref/reindexdb.sgml b/doc/src/sgml/ref/reindexdb.sgml index a3b0f7ce3190b..80a7f84886bed 100644 --- a/doc/src/sgml/ref/reindexdb.sgml +++ b/doc/src/sgml/ref/reindexdb.sgml @@ -237,6 +237,16 @@ PostgreSQL documentation + + + + + Specifies the tablespace where indexes are rebuilt. (This name is + processed as a double-quoted identifier.) + + + + diff --git a/src/bin/scripts/reindexdb.c b/src/bin/scripts/reindexdb.c index 9f072ac49aea9..cf281762431c5 100644 --- a/src/bin/scripts/reindexdb.c +++ b/src/bin/scripts/reindexdb.c @@ -40,14 +40,15 @@ static void reindex_one_database(const ConnParams *cparams, ReindexType type, SimpleStringList *user_list, const char *progname, bool echo, bool verbose, bool concurrently, - int concurrentCons); + int concurrentCons, const char *tablespace); static void reindex_all_databases(ConnParams *cparams, const char *progname, bool echo, bool quiet, bool verbose, bool concurrently, - int concurrentCons); + int concurrentCons, const char *tablespace); static void run_reindex_command(PGconn *conn, ReindexType type, const char *name, bool echo, bool verbose, - bool concurrently, bool async); + bool concurrently, bool async, + const char *tablespace); static void help(const char *progname); @@ -72,6 +73,7 @@ main(int argc, char *argv[]) {"verbose", no_argument, NULL, 'v'}, {"concurrently", no_argument, NULL, 1}, {"maintenance-db", required_argument, NULL, 2}, + {"tablespace", required_argument, NULL, 3}, {NULL, 0, NULL, 0} }; @@ -84,6 +86,7 @@ main(int argc, char *argv[]) const char *host = NULL; const char *port = NULL; const char *username = NULL; + const char *tablespace = NULL; enum trivalue prompt_password = TRI_DEFAULT; ConnParams cparams; bool syscatalog = false; @@ -164,6 +167,9 @@ main(int argc, char *argv[]) case 2: maintenance_db = pg_strdup(optarg); break; + case 3: + tablespace = pg_strdup(optarg); + break; default: fprintf(stderr, _("Try \"%s --help\" for more information.\n"), progname); exit(1); @@ -228,7 +234,7 @@ main(int argc, char *argv[]) cparams.dbname = maintenance_db; reindex_all_databases(&cparams, progname, echo, quiet, verbose, - concurrently, concurrentCons); + concurrently, concurrentCons, tablespace); } else if (syscatalog) { @@ -268,7 +274,7 @@ main(int argc, char *argv[]) reindex_one_database(&cparams, REINDEX_SYSTEM, NULL, progname, echo, verbose, - concurrently, 1); + concurrently, 1, tablespace); } else { @@ -298,17 +304,17 @@ main(int argc, char *argv[]) if (schemas.head != NULL) reindex_one_database(&cparams, REINDEX_SCHEMA, &schemas, progname, echo, verbose, - concurrently, concurrentCons); + concurrently, concurrentCons, tablespace); if (indexes.head != NULL) reindex_one_database(&cparams, REINDEX_INDEX, &indexes, progname, echo, verbose, - concurrently, 1); + concurrently, 1, tablespace); if (tables.head != NULL) reindex_one_database(&cparams, REINDEX_TABLE, &tables, progname, echo, verbose, - concurrently, concurrentCons); + concurrently, concurrentCons, tablespace); /* * reindex database only if neither index nor table nor schema is @@ -317,7 +323,7 @@ main(int argc, char *argv[]) if (indexes.head == NULL && tables.head == NULL && schemas.head == NULL) reindex_one_database(&cparams, REINDEX_DATABASE, NULL, progname, echo, verbose, - concurrently, concurrentCons); + concurrently, concurrentCons, tablespace); } exit(0); @@ -327,7 +333,8 @@ static void reindex_one_database(const ConnParams *cparams, ReindexType type, SimpleStringList *user_list, const char *progname, bool echo, - bool verbose, bool concurrently, int concurrentCons) + bool verbose, bool concurrently, int concurrentCons, + const char *tablespace) { PGconn *conn; SimpleStringListCell *cell; @@ -348,6 +355,14 @@ reindex_one_database(const ConnParams *cparams, ReindexType type, exit(1); } + if (tablespace && PQserverVersion(conn) < 140000) + { + PQfinish(conn); + pg_log_error("cannot use the \"%s\" option on server versions older than PostgreSQL %s", + "tablespace", "14"); + exit(1); + } + if (!parallel) { switch (process_type) @@ -386,7 +401,8 @@ reindex_one_database(const ConnParams *cparams, ReindexType type, pg_log_warning("cannot reindex system catalogs concurrently, skipping all"); else run_reindex_command(conn, REINDEX_SYSTEM, PQdb(conn), echo, - verbose, concurrently, false); + verbose, concurrently, false, + tablespace); /* Build a list of relations from the database */ process_list = get_parallel_object_list(conn, process_type, @@ -468,7 +484,7 @@ reindex_one_database(const ConnParams *cparams, ReindexType type, ParallelSlotSetHandler(free_slot, TableCommandResultHandler, NULL); run_reindex_command(free_slot->connection, process_type, objname, - echo, verbose, concurrently, true); + echo, verbose, concurrently, true, tablespace); cell = cell->next; } while (cell != NULL); @@ -492,8 +508,12 @@ reindex_one_database(const ConnParams *cparams, ReindexType type, static void run_reindex_command(PGconn *conn, ReindexType type, const char *name, - bool echo, bool verbose, bool concurrently, bool async) + bool echo, bool verbose, bool concurrently, bool async, + const char *tablespace) { + const char *paren = "("; + const char *comma = ", "; + const char *sep = paren; PQExpBufferData sql; bool status; @@ -505,7 +525,19 @@ run_reindex_command(PGconn *conn, ReindexType type, const char *name, appendPQExpBufferStr(&sql, "REINDEX "); if (verbose) - appendPQExpBufferStr(&sql, "(VERBOSE) "); + { + appendPQExpBuffer(&sql, "%sVERBOSE", sep); + sep = comma; + } + + if (tablespace) + { + appendPQExpBuffer(&sql, "%sTABLESPACE %s", sep, fmtId(tablespace)); + sep = comma; + } + + if (sep != paren) + appendPQExpBufferStr(&sql, ") "); /* object type */ switch (type) @@ -527,6 +559,11 @@ run_reindex_command(PGconn *conn, ReindexType type, const char *name, break; } + /* + * Parenthesized grammar is only supported for CONCURRENTLY since + * PostgreSQL 14. Since 12, CONCURRENTLY can be specified after the + * object type. + */ if (concurrently) appendPQExpBufferStr(&sql, "CONCURRENTLY "); @@ -716,7 +753,8 @@ get_parallel_object_list(PGconn *conn, ReindexType type, static void reindex_all_databases(ConnParams *cparams, const char *progname, bool echo, bool quiet, bool verbose, - bool concurrently, int concurrentCons) + bool concurrently, int concurrentCons, + const char *tablespace) { PGconn *conn; PGresult *result; @@ -740,7 +778,7 @@ reindex_all_databases(ConnParams *cparams, reindex_one_database(cparams, REINDEX_DATABASE, NULL, progname, echo, verbose, concurrently, - concurrentCons); + concurrentCons, tablespace); } PQclear(result); @@ -753,26 +791,27 @@ help(const char *progname) printf(_("Usage:\n")); printf(_(" %s [OPTION]... [DBNAME]\n"), progname); printf(_("\nOptions:\n")); - printf(_(" -a, --all reindex all databases\n")); - printf(_(" --concurrently reindex concurrently\n")); - printf(_(" -d, --dbname=DBNAME database to reindex\n")); - printf(_(" -e, --echo show the commands being sent to the server\n")); - printf(_(" -i, --index=INDEX recreate specific index(es) only\n")); - printf(_(" -j, --jobs=NUM use this many concurrent connections to reindex\n")); - printf(_(" -q, --quiet don't write any messages\n")); - printf(_(" -s, --system reindex system catalogs\n")); - printf(_(" -S, --schema=SCHEMA reindex specific schema(s) only\n")); - printf(_(" -t, --table=TABLE reindex specific table(s) only\n")); - printf(_(" -v, --verbose write a lot of output\n")); - printf(_(" -V, --version output version information, then exit\n")); - printf(_(" -?, --help show this help, then exit\n")); + printf(_(" -a, --all reindex all databases\n")); + printf(_(" --concurrently reindex concurrently\n")); + printf(_(" -d, --dbname=DBNAME database to reindex\n")); + printf(_(" -e, --echo show the commands being sent to the server\n")); + printf(_(" -i, --index=INDEX recreate specific index(es) only\n")); + printf(_(" -j, --jobs=NUM use this many concurrent connections to reindex\n")); + printf(_(" -q, --quiet don't write any messages\n")); + printf(_(" -s, --system reindex system catalogs\n")); + printf(_(" -S, --schema=SCHEMA reindex specific schema(s) only\n")); + printf(_(" -t, --table=TABLE reindex specific table(s) only\n")); + printf(_(" --tablespace=TABLESPACE tablespace where indexes are rebuilt\n")); + printf(_(" -v, --verbose write a lot of output\n")); + printf(_(" -V, --version output version information, then exit\n")); + printf(_(" -?, --help show this help, then exit\n")); printf(_("\nConnection options:\n")); - printf(_(" -h, --host=HOSTNAME database server host or socket directory\n")); - printf(_(" -p, --port=PORT database server port\n")); - printf(_(" -U, --username=USERNAME user name to connect as\n")); - printf(_(" -w, --no-password never prompt for password\n")); - printf(_(" -W, --password force password prompt\n")); - printf(_(" --maintenance-db=DBNAME alternate maintenance database\n")); + printf(_(" -h, --host=HOSTNAME database server host or socket directory\n")); + printf(_(" -p, --port=PORT database server port\n")); + printf(_(" -U, --username=USERNAME user name to connect as\n")); + printf(_(" -w, --no-password never prompt for password\n")); + printf(_(" -W, --password force password prompt\n")); + printf(_(" --maintenance-db=DBNAME alternate maintenance database\n")); printf(_("\nRead the description of the SQL command REINDEX for details.\n")); printf(_("\nReport bugs to <%s>.\n"), PACKAGE_BUGREPORT); printf(_("%s home page: <%s>\n"), PACKAGE_NAME, PACKAGE_URL); diff --git a/src/bin/scripts/t/090_reindexdb.pl b/src/bin/scripts/t/090_reindexdb.pl index d02cca7a8d7fb..159b637230335 100644 --- a/src/bin/scripts/t/090_reindexdb.pl +++ b/src/bin/scripts/t/090_reindexdb.pl @@ -3,7 +3,7 @@ use PostgresNode; use TestLib; -use Test::More tests => 44; +use Test::More tests => 58; program_help_ok('reindexdb'); program_version_ok('reindexdb'); @@ -15,17 +15,38 @@ $ENV{PGOPTIONS} = '--client-min-messages=WARNING'; +# Create a tablespace for testing. +my $tbspace_path = $node->basedir . '/regress_reindex_tbspace'; +mkdir $tbspace_path or die "cannot create directory $tbspace_path"; +$tbspace_path = TestLib::perl2host($tbspace_path); +my $tbspace_name = 'reindex_tbspace'; +$node->safe_psql('postgres', + "CREATE TABLESPACE $tbspace_name LOCATION '$tbspace_path';"); + $node->issues_sql_like( [ 'reindexdb', 'postgres' ], qr/statement: REINDEX DATABASE postgres;/, 'SQL REINDEX run'); +# Use text as data type to get a toast table. $node->safe_psql('postgres', - 'CREATE TABLE test1 (a int); CREATE INDEX test1x ON test1 (a);'); + 'CREATE TABLE test1 (a text); CREATE INDEX test1x ON test1 (a);'); +# Collect toast table and index names of this relation, for later use. +my $toast_table = $node->safe_psql('postgres', + "SELECT reltoastrelid::regclass FROM pg_class WHERE oid = 'test1'::regclass;" +); +my $toast_index = $node->safe_psql('postgres', + "SELECT indexrelid::regclass FROM pg_index WHERE indrelid = '$toast_table'::regclass;" +); + $node->issues_sql_like( [ 'reindexdb', '-t', 'test1', 'postgres' ], qr/statement: REINDEX TABLE public\.test1;/, 'reindex specific table'); +$node->issues_sql_like( + [ 'reindexdb', '-t', 'test1', '--tablespace', $tbspace_name, 'postgres' ], + qr/statement: REINDEX \(TABLESPACE $tbspace_name\) TABLE public\.test1;/, + 'reindex specific table on tablespace'); $node->issues_sql_like( [ 'reindexdb', '-i', 'test1x', 'postgres' ], qr/statement: REINDEX INDEX public\.test1x;/, @@ -42,6 +63,13 @@ [ 'reindexdb', '-v', '-t', 'test1', 'postgres' ], qr/statement: REINDEX \(VERBOSE\) TABLE public\.test1;/, 'reindex with verbose output'); +$node->issues_sql_like( + [ + 'reindexdb', '-v', '-t', 'test1', + '--tablespace', $tbspace_name, 'postgres' + ], + qr/statement: REINDEX \(VERBOSE, TABLESPACE $tbspace_name\) TABLE public\.test1;/, + 'reindex with verbose output and tablespace'); # the same with --concurrently $node->issues_sql_like( @@ -67,6 +95,55 @@ [ 'reindexdb', '--concurrently', '-v', '-t', 'test1', 'postgres' ], qr/statement: REINDEX \(VERBOSE\) TABLE CONCURRENTLY public\.test1;/, 'reindex with verbose output concurrently'); +$node->issues_sql_like( + [ + 'reindexdb', '--concurrently', '-v', '-t', + 'test1', '--tablespace', $tbspace_name, 'postgres' + ], + qr/statement: REINDEX \(VERBOSE, TABLESPACE $tbspace_name\) TABLE CONCURRENTLY public\.test1;/, + 'reindex concurrently with verbose output and tablespace'); + +# REINDEX TABLESPACE on toast indexes and tables fails. This is not +# part of the main regression test suite as these have unpredictable +# names, and CONCURRENTLY cannot be used in transaction blocks, preventing +# the use of TRY/CATCH blocks in a custom function to filter error +# messages. +$node->command_checks_all( + [ + 'reindexdb', '-t', $toast_table, '--tablespace', + $tbspace_name, 'postgres' + ], + 1, + [], + [qr/cannot move system relation/], + 'reindex toast table with tablespace'); +$node->command_checks_all( + [ + 'reindexdb', '--concurrently', '-t', $toast_table, + '--tablespace', $tbspace_name, 'postgres' + ], + 1, + [], + [qr/cannot move system relation/], + 'reindex toast table concurrently with tablespace'); +$node->command_checks_all( + [ + 'reindexdb', '-i', $toast_index, '--tablespace', + $tbspace_name, 'postgres' + ], + 1, + [], + [qr/cannot move system relation/], + 'reindex toast index with tablespace'); +$node->command_checks_all( + [ + 'reindexdb', '--concurrently', '-i', $toast_index, + '--tablespace', $tbspace_name, 'postgres' + ], + 1, + [], + [qr/cannot move system relation/], + 'reindex toast index concurrently with tablespace'); # connection strings $node->command_ok([qw(reindexdb --echo --table=pg_am dbname=template1)],