\n"
+" -o show info for table with Oid \n"
+" -f show info for table with filenode \n"
+" -H host connect to remote host\n"
+" -p port host port to connect to\n"
+" -U username username to connect with\n"
+" -P password password for username\n"
+" (see also $PGPASSWORD and ~/.pgpass)\n"
+);
exit(1);
break;
}
}
}
-/* establish connection with database. */
-PGconn *
-sql_conn(const char *dbName, struct options * my_opts)
+void *
+myalloc(size_t size)
{
- char *pghost,
- *pgport;
- char *pgoptions,
- *pgtty;
- char *pguser,
- *pgpass;
-
- PGconn *conn;
-
- pghost = NULL;
- pgport = NULL;
- pgoptions = NULL; /* special options to start up the backend
- * server */
- pgtty = NULL; /* debugging tty for the backend server */
- pguser = NULL;
- pgpass = NULL;
-
- /* override the NULLs with the user params if passed */
- if (my_opts->remotehost)
+ void *ptr = malloc(size);
+ if (!ptr)
{
- pghost = (char *) malloc(128);
- sscanf(my_opts->_hostname, "%s", pghost);
+ fprintf(stderr, "out of memory");
+ exit(1);
}
+ return ptr;
+}
- if (my_opts->remoteport)
+/*
+ * add_one_elt
+ *
+ * Add one element to a (possibly empty) eary struct.
+ */
+void
+add_one_elt(char *eltname, eary *eary)
+{
+ if (eary->alloc == 0)
{
- pgport = (char *) malloc(6);
- sscanf(my_opts->_port, "%s", pgport);
+ eary->alloc = 8;
+ eary->array = (char **) myalloc(8 * sizeof(char *));
}
-
- if (my_opts->remoteuser)
+ else if (eary->num >= eary->alloc)
{
- pguser = (char *) malloc(128);
- sscanf(my_opts->_username, "%s", pguser);
+ eary->alloc *= 2;
+ eary->array = (char **)
+ realloc(eary->array, eary->alloc * sizeof(char *));
+ if (!eary->array)
+ {
+ fprintf(stderr, "out of memory");
+ exit(1);
+ }
}
- if (my_opts->remotepass)
+ eary->array[eary->num] = strdup(eltname);
+ eary->num++;
+}
+
+/*
+ * get_comma_elts
+ *
+ * Return the elements of an eary as a (freshly allocated) single string, in
+ * single quotes, separated by commas and properly escaped for insertion in an
+ * SQL statement.
+ */
+char *
+get_comma_elts(eary *eary)
+{
+ char *ret,
+ *ptr;
+ int i, length = 0;
+
+ if (eary->num == 0)
+ return "";
+
+ /*
+ * PQescapeString wants 2 * length + 1 bytes of breath space. Add two
+ * chars per element for the single quotes and one for the comma.
+ */
+ for (i = 0; i < eary->num; i++)
+ length += strlen(eary->array[i]);
+
+ ret = (char *) myalloc(length * 2 + 4 * eary->num);
+ ptr = ret;
+
+ for (i = 0; i < eary->num; i++)
{
- pgpass = (char *) malloc(128);
- sscanf(my_opts->_password, "%s", pgpass);
+ if (i != 0)
+ sprintf(ptr++, ",");
+ sprintf(ptr++, "'");
+ ptr += PQescapeString(ptr, eary->array[i], strlen(eary->array[i]));
+ sprintf(ptr++, "'");
}
+ return ret;
+}
+
+/* establish connection with database. */
+PGconn *
+sql_conn(struct options * my_opts)
+{
+ PGconn *conn;
+
/* login */
- conn = PQsetdbLogin(pghost, pgport, pgoptions, pgtty, dbName, pguser, pgpass);
+ conn = PQsetdbLogin(my_opts->hostname,
+ my_opts->port,
+ NULL, /* options */
+ NULL, /* tty */
+ my_opts->dbname,
+ my_opts->username,
+ my_opts->password);
/* deal with errors */
- if (PQstatus(conn) == CONNECTION_BAD)
+ if (PQstatus(conn) != CONNECTION_OK)
{
- fprintf(stderr, "Connection to database '%s' failed.\n", dbName);
+ fprintf(stderr, "%s: connection to database '%s' failed.\n", "oid2name", my_opts->dbname);
fprintf(stderr, "%s", PQerrorMessage(conn));
PQfinish(conn);
exit(1);
}
- /* free data structures: not strictly necessary */
- if (pghost != NULL)
- free(pghost);
- if (pgport != NULL)
- free(pgport);
- if (pguser != NULL)
- free(pguser);
- if (pgpass != NULL)
- free(pgpass);
-
- sql_exec(conn, "SET search_path = public;", 0);
-
/* return the conn if good */
return conn;
}
-/* If the sql_ command has an error, this function looks up the error number and prints it out. */
-void
-sql_exec_error(int error_number)
-{
- fprintf(stderr, "Error number %i.\n", error_number);
- switch (error_number)
- {
- case 3:
- fprintf(stderr, "Error: PGRES_COPY_OUT\n");
- break;
-
- case 4:
- fprintf(stderr, "Error: PGRES_COPY_IN\n");
- break;
-
- case 5:
- fprintf(stderr, "Error: PGRES_BAD_RESPONCE\n");
- break;
-
- case 6:
- fprintf(stderr, "Error: PGRES_NONFATAL_ERROR\n");
- break;
-
- case 7:
- fprintf(stderr, "Error: PGRES_FATAL_ERROR\n");
- break;
- }
-}
-
-/* actual code to make call to the database and print the output data */
+/*
+ * Actual code to make call to the database and print the output data.
+ */
int
-sql_exec(PGconn *conn, const char *todo, int match)
+sql_exec(PGconn *conn, const char *todo, bool quiet)
{
PGresult *res;
- int numbfields;
- int error_number;
- int i,
- len;
+ int nfields;
+ int nrows;
+ int i, j, l;
+ int *length;
+ char *pad;
/* make the call */
res = PQexec(conn, todo);
/* check and deal with errors */
if (!res || PQresultStatus(res) > 2)
{
- error_number = PQresultStatus(res);
- fprintf(stderr, "There was an error in the SQL command:\n%s\n", todo);
- sql_exec_error(error_number);
- fprintf(stderr, "PQerrorMessage = %s\n", PQerrorMessage(conn));
+ fprintf(stderr, "oid2name: query failed: %s\n", PQerrorMessage(conn));
+ fprintf(stderr, "oid2name: query was: %s\n", todo);
PQclear(res);
PQfinish(conn);
}
/* get the number of fields */
- numbfields = PQntuples(res);
+ nrows = PQntuples(res);
+ nfields = PQnfields(res);
- /* if we only expect 1 and there mode than, return -2 */
- if (match == 1 && numbfields > 1)
- return -2;
+ /* for each field, get the needed width */
+ length = (int *) myalloc(sizeof(int) * nfields);
+ for (j = 0; j < nfields; j++)
+ length[j] = strlen(PQfname(res, j));
- /* return -1 if there aren't any returns */
- if (match == 1 && numbfields < 1)
- return -1;
+ for (i = 0; i < nrows; i++)
+ {
+ for (j = 0; j < nfields; j++)
+ {
+ l = strlen(PQgetvalue(res, i, j));
+ if (l > length[j])
+ length[j] = strlen(PQgetvalue(res, i, j));
+ }
+ }
- /* for each row, dump the information */
- for (i = 0; i < numbfields; i++)
+ /* print a header */
+ if (!quiet)
{
- len = strlen(PQgetvalue(res, i, 0));
+ for (j = 0, l = 0; j < nfields; j++)
+ {
+ fprintf(stdout, "%*s", length[j] + 2, PQfname(res, j));
+ l += length[j] + 2;
+ }
+ fprintf(stdout, "\n");
+ pad = (char *) myalloc(l + 1);
+ MemSet(pad, '-', l);
+ pad[l] = '\0';
+ fprintf(stdout, "%s\n", pad);
+ free(pad);
+ }
- fprintf(stdout, "%-6s = %s\n", PQgetvalue(res, i, 0), PQgetvalue(res, i, 1));
+ /* for each row, dump the information */
+ for (i = 0; i < nrows; i++)
+ {
+ for (j = 0; j < nfields; j++)
+ fprintf(stdout, "%*s", length[j] + 2, PQgetvalue(res, i, j));
+ fprintf(stdout, "\n");
}
- /* clean the PGconn once done */
+ /* cleanup */
PQclear(res);
+ free(length);
return 0;
}
-/* dump all databases known by the system table */
+/*
+ * Dump all databases. There are no system objects to worry about.
+ */
void
-sql_exec_dumpdb(PGconn *conn)
+sql_exec_dumpalldbs(PGconn *conn, struct options *opts)
{
char todo[1024];
/* get the oid and database name from the system pg_database table */
- snprintf(todo, 1024, "select oid,datname from pg_database");
+ snprintf(todo, 1024, "SELECT d.oid AS \"Oid\", datname AS \"Database Name\", "
+ "spcname AS \"Tablespace\" FROM pg_database d JOIN pg_tablespace t ON "
+ "(dattablespace = t.oid) ORDER BY 2");
- sql_exec(conn, todo, 0);
+ sql_exec(conn, todo, opts->quiet);
}
-/* display all tables in whatever db we are connected to. don't display the
- system tables by default */
+/*
+ * Dump all tables, indexes and sequences in the current database.
+ */
void
-sql_exec_dumptable(PGconn *conn, int systables)
+sql_exec_dumpalltables(PGconn *conn, struct options *opts)
{
char todo[1024];
-
- /* don't exclude the systables if this is set */
- if (systables == 1)
- snprintf(todo, 1024, "select relfilenode,relname from pg_class order by relname");
- else
- snprintf(todo, 1024, "select relfilenode,relname from pg_class "
- "where relkind not in ('v','s', 'c') and "
- "relname not like 'pg_%%' order by relname");
-
- sql_exec(conn, todo, 0);
+ char *addfields = ",c.oid AS \"Oid\", nspname AS \"Schema\", spcname as \"Tablespace\" ";
+
+ snprintf(todo, 1024,
+ "SELECT relfilenode as \"Filenode\", relname as \"Table Name\" %s "
+ "FROM pg_class c "
+ " LEFT JOIN pg_catalog.pg_namespace n ON n.oid = c.relnamespace "
+ " LEFT JOIN pg_catalog.pg_database d ON d.datname = current_database(),"
+ " pg_catalog.pg_tablespace t "
+ "WHERE relkind IN ('r'%s) AND "
+ " %s"
+ " t.oid = CASE"
+ " WHEN reltablespace <> 0 THEN reltablespace"
+ " WHEN n.nsptablespace <> 0 THEN nsptablespace"
+ " WHEN d.dattablespace <> 0 THEN dattablespace"
+ " END "
+ "ORDER BY relname",
+ opts->extended ? addfields : "",
+ opts->indexes ? ", 'i', 'S', 't'" : "",
+ opts->systables ? "" : "n.nspname NOT IN ('pg_catalog', 'pg_toast', 'information_schema') AND");
+
+ sql_exec(conn, todo, opts->quiet);
}
-/* display the oid for a given tablename for whatever db we are connected
- to. do we want to allow %bar% in the search? Not now. */
+/*
+ * Show oid, relfilenode, name, schema and tablespace for each of the
+ * given objects in the current database.
+ */
void
-sql_exec_searchtable(PGconn *conn, const char *tablename)
+sql_exec_searchtables(PGconn *conn, struct options *opts)
{
- int returnvalue;
- char todo[1024];
-
- /* get the oid and tablename where the name matches tablename */
- snprintf(todo, 1024, "select relfilenode,relname from pg_class where relname = '%s'", tablename);
-
- returnvalue = sql_exec(conn, todo, 1);
-
- /* deal with the return errors */
- if (returnvalue == -1)
- printf("No tables with that name found\n");
-
- if (returnvalue == -2)
- printf("VERY scary: more than one table with that name found!!\n");
+ char *todo;
+ char *qualifiers, *ptr;
+ char *comma_oids, *comma_filenodes, *comma_tables;
+ bool written = false;
+ char *addfields = ",c.oid AS \"Oid\", nspname AS \"Schema\", spcname as \"Tablespace\" ";
+
+ /* get tables qualifiers, whether names, relfilenodes, or OIDs */
+ comma_oids = get_comma_elts(opts->oids);
+ comma_tables = get_comma_elts(opts->tables);
+ comma_filenodes = get_comma_elts(opts->filenodes);
+
+ /* 80 extra chars for SQL expression */
+ qualifiers = (char *) myalloc(strlen(comma_oids) + strlen(comma_tables) +
+ strlen(comma_filenodes) + 80);
+ ptr = qualifiers;
+
+ if (opts->oids->num > 0)
+ {
+ ptr += sprintf(ptr, "c.oid IN (%s)", comma_oids);
+ written = true;
+ }
+ if (opts->filenodes->num > 0)
+ {
+ if (written)
+ ptr += sprintf(ptr, " OR ");
+ ptr += sprintf(ptr, "c.relfilenode IN (%s)", comma_filenodes);
+ written = true;
+ }
+ if (opts->tables->num > 0)
+ {
+ if (written)
+ ptr += sprintf(ptr, " OR ");
+ sprintf(ptr, "c.relname ~~ ANY (ARRAY[%s])", comma_tables);
+ }
+ free(comma_oids);
+ free(comma_tables);
+ free(comma_filenodes);
+
+ /* now build the query */
+ todo = (char *) myalloc(650 + strlen(qualifiers));
+ snprintf(todo, 1024,
+ "SELECT relfilenode as \"Filenode\", relname as \"Table Name\" %s\n"
+ "FROM pg_class c \n"
+ " LEFT JOIN pg_catalog.pg_namespace n ON n.oid = c.relnamespace \n"
+ " LEFT JOIN pg_catalog.pg_database d ON d.datname = current_database(),\n"
+ " pg_catalog.pg_tablespace t \n"
+ "WHERE relkind IN ('r', 'i', 'S', 't') AND \n"
+ " t.oid = CASE\n"
+ " WHEN reltablespace <> 0 THEN reltablespace\n"
+ " WHEN n.nsptablespace <> 0 THEN nsptablespace\n"
+ " WHEN d.dattablespace <> 0 THEN dattablespace\n"
+ " END AND \n"
+ " (%s) \n"
+ "ORDER BY relname\n",
+ opts->extended ? addfields : "",
+ qualifiers);
+
+ free(qualifiers);
+
+ sql_exec(conn, todo, opts->quiet);
}
-/* same as above */
void
-sql_exec_searchoid(PGconn *conn, int oid)
+sql_exec_dumpalltbspc(PGconn *conn, struct options *opts)
{
- int returnvalue;
- char todo[1024];
-
- snprintf(todo, 1024, "select relfilenode,relname from pg_class where oid = %i", oid);
+ char todo[1024];
- returnvalue = sql_exec(conn, todo, 1);
+ snprintf(todo, 1024, "SELECT oid AS \"Oid\", spcname as \"Tablespace Name\"\n"
+ "FROM pg_tablespace");
- if (returnvalue == -1)
- printf("No tables with that oid found\n");
-
- if (returnvalue == -2)
- printf("VERY scary: more than one table with that oid found!!\n");
+ sql_exec(conn, todo, opts->quiet);
}
int
struct options *my_opts;
PGconn *pgconn;
- my_opts = (struct options *) malloc(sizeof(struct options));
+ my_opts = (struct options *) myalloc(sizeof(struct options));
+
+ my_opts->oids = (eary *) myalloc(sizeof(eary));
+ my_opts->tables = (eary *) myalloc(sizeof(eary));
+ my_opts->filenodes = (eary *) myalloc(sizeof(eary));
+
+ my_opts->oids->num = my_opts->oids->alloc = 0;
+ my_opts->tables->num = my_opts->tables->alloc = 0;
+ my_opts->filenodes->num = my_opts->filenodes->alloc = 0;
/* parse the opts */
get_opts(argc, argv, my_opts);
- /* display all the tables in the database */
- if (my_opts->getdatabase & my_opts->gettable)
+ if (my_opts->dbname == NULL)
+ {
+ my_opts->dbname = "template1";
+ my_opts->nodb = true;
+ }
+ pgconn = sql_conn(my_opts);
+
+ /* display only tablespaces */
+ if (my_opts->tablespaces)
{
if (!my_opts->quiet)
- {
- printf("Oid of table %s from database \"%s\":\n", my_opts->_tbname, my_opts->_dbname);
- printf("---------------------------------\n");
- }
- pgconn = sql_conn(my_opts->_dbname, my_opts);
- sql_exec_searchtable(pgconn, my_opts->_tbname);
- PQfinish(pgconn);
+ printf("All tablespaces:\n");
+ sql_exec_dumpalltbspc(pgconn, my_opts);
- exit(1);
+ PQfinish(pgconn);
+ exit(0);
}
- /* search for the tablename of the given OID */
- if (my_opts->getdatabase & my_opts->getoid)
+ /* display the given elements in the database */
+ if (my_opts->oids->num > 0 ||
+ my_opts->tables->num > 0 ||
+ my_opts->filenodes->num > 0)
{
if (!my_opts->quiet)
- {
- printf("Tablename of oid %i from database \"%s\":\n", my_opts->_oid, my_opts->_dbname);
- printf("---------------------------------\n");
- }
- pgconn = sql_conn(my_opts->_dbname, my_opts);
- sql_exec_searchoid(pgconn, my_opts->_oid);
- PQfinish(pgconn);
+ printf("From database \"%s\":\n", my_opts->dbname);
+ sql_exec_searchtables(pgconn, my_opts);
- exit(1);
+ PQfinish(pgconn);
+ exit(0);
}
- /* search for the oid for the given tablename */
- if (my_opts->getdatabase)
+ /* no elements given; dump the given database */
+ if (my_opts->dbname && !my_opts->nodb)
{
if (!my_opts->quiet)
- {
- printf("All tables from database \"%s\":\n", my_opts->_dbname);
- printf("---------------------------------\n");
- }
- pgconn = sql_conn(my_opts->_dbname, my_opts);
- sql_exec_dumptable(pgconn, my_opts->systables);
- PQfinish(pgconn);
+ printf("From database \"%s\":\n", my_opts->dbname);
+ sql_exec_dumpalltables(pgconn, my_opts);
- exit(1);
+ PQfinish(pgconn);
+ exit(0);
}
- /* display all the databases for the server we are connected to.. */
+ /* no database either; dump all databases */
if (!my_opts->quiet)
- {
printf("All databases:\n");
- printf("---------------------------------\n");
- }
- pgconn = sql_conn("template1", my_opts);
- sql_exec_dumpdb(pgconn);
- PQfinish(pgconn);
+ sql_exec_dumpalldbs(pgconn, my_opts);
+ PQfinish(pgconn);
exit(0);
}
projects / postgresql.git / commitdiff