1996-07-09 06:22:35 +00:00
/*-------------------------------------------------------------------------
*
* pg_dump . c - -
* pg_dump is an utility for dumping out a postgres database
* into a script file .
*
* pg_dump will read the system catalogs in a database and
* dump out a script that reproduces
* the schema of the database in terms of
* user - defined types
* user - defined functions
* tables
* indices
* aggregates
* operators
*
* the output script is SQL that is understood by Postgres95
*
* Copyright ( c ) 1994 , Regents of the University of California
*
*
* IDENTIFICATION
1996-07-31 06:09:46 +00:00
* $ Header : / cvsroot / pgsql / src / bin / pg_dump / pg_dump . c , v 1.5 1996 / 07 / 31 06 : 09 : 46 scrappy Exp $
1996-07-12 05:39:39 +00:00
*
1996-07-22 08:37:00 +00:00
* Modifications - 6 / 10 / 96 - dave @ bensoft . com - version 1.13 . dhb
1996-07-12 05:39:39 +00:00
*
* Applied ' insert string ' patch from " Marc G. Fournier " < scrappy @ ki . net >
* Added ' - t table ' option
* Added ' - a ' option
* Added ' - da ' option
1996-07-09 06:22:35 +00:00
*
1996-07-22 08:37:00 +00:00
* Modifications - 6 / 12 / 96 - dave @ bensoft . com - version 1.13 . dhb .2
*
* - Fixed dumpTable output to output lengths for char and varchar types !
* - Added single . quote to twin single quote expansion for ' insert ' string
* mode .
*
1996-07-27 02:29:51 +00:00
* Modifications - 7 / 26 / 96 - asussman @ vidya . com
*
* - Fixed ouput lengths for char and varchar type where the length is variable ( - 1 )
*
1996-07-09 06:22:35 +00:00
* - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
*/
# include <stdlib.h>
# include <stdio.h>
# include <string.h>
# include <sys/param.h> /* for MAXHOSTNAMELEN on most */
# ifdef PORTNAME_sparc_solaris
# include <netdb.h> /* for MAXHOSTNAMELEN on some */
# endif
# include "postgres.h"
1996-07-31 06:09:46 +00:00
# include "access/htup.h"
1996-07-09 06:22:35 +00:00
# include "libpq-fe.h"
# include "pg_dump.h"
extern char * optarg ;
extern int optind , opterr ;
/* global decls */
int g_verbose ; /* verbose flag */
int g_last_builtin_oid ; /* value of the last builtin oid */
FILE * g_fout ; /* the script file */
PGconn * g_conn ; /* the database connection */
1996-07-12 05:39:39 +00:00
int dumpData ; /* dump data using proper insert strings */
int attrNames ; /* put attr names into insert strings */
1996-07-09 06:22:35 +00:00
char g_opaque_type [ 10 ] ; /* name for the opaque type */
/* placeholders for the delimiters for comments */
char g_comment_start [ 10 ] ;
char g_comment_end [ 10 ] ;
static void
usage ( char * progname )
{
1996-07-22 08:37:00 +00:00
fprintf ( stderr , " %s - version 1.13.dhb.2 \n \n " , progname ) ;
1996-07-09 06:22:35 +00:00
fprintf ( stderr , " usage: %s [options] [dbname] \n " , progname ) ;
fprintf ( stderr , " \t -f filename \t \t script output filename \n " ) ;
1996-07-12 05:39:39 +00:00
fprintf ( stderr , " \t -d[a] \t \t dump data as proper insert strings \n " ) ;
fprintf ( stderr , " \t \t \t (if 'a' then attribute names also) \n " ) ;
1996-07-09 06:22:35 +00:00
fprintf ( stderr , " \t -H hostname \t \t server host name \n " ) ;
fprintf ( stderr , " \t -p port \t \t server port number \n " ) ;
fprintf ( stderr , " \t -v \t \t verbose \n " ) ;
fprintf ( stderr , " \t -S \t \t dump out only the schema, no data \n " ) ;
1996-07-12 05:39:39 +00:00
fprintf ( stderr , " \t -a \t \t dump out only the data, no schema \n " ) ;
fprintf ( stderr , " \t -t table \t \t dump for this table only \n " ) ;
1996-07-09 06:22:35 +00:00
fprintf ( stderr , " \n if dbname is not supplied, then the DATABASE environment name is used \n " ) ;
fprintf ( stderr , " \n " ) ;
fprintf ( stderr , " \t pg_dump dumps out postgres databases and produces a script file \n " ) ;
fprintf ( stderr , " \t of SQL commands to regenerate the schema \n " ) ;
fprintf ( stderr , " \t The SQL output is designed for import into Postgres95 \n " ) ;
exit ( 1 ) ;
}
void
exit_nicely ( PGconn * conn )
{
PQfinish ( conn ) ;
exit ( 1 ) ;
}
void
main ( int argc , char * * argv )
{
int c ;
char * progname ;
char * filename ;
char * dbname ;
int schemaOnly ;
1996-07-12 05:39:39 +00:00
int dataOnly ;
1996-07-09 06:22:35 +00:00
char * pghost = NULL ;
char * pgport = NULL ;
1996-07-12 05:39:39 +00:00
char * tablename ;
1996-07-09 06:22:35 +00:00
TableInfo * tblinfo ;
int numTables ;
dbname = NULL ;
filename = NULL ;
1996-07-12 05:39:39 +00:00
tablename = NULL ;
1996-07-09 06:22:35 +00:00
g_verbose = 0 ;
strcpy ( g_comment_start , " -- " ) ;
g_comment_end [ 0 ] = ' \0 ' ;
strcpy ( g_opaque_type , " opaque " ) ;
1996-07-12 05:39:39 +00:00
dataOnly = schemaOnly = dumpData = attrNames = 0 ;
1996-07-09 06:22:35 +00:00
progname = * argv ;
1996-07-12 05:39:39 +00:00
while ( ( c = getopt ( argc , argv , " f:H:p:t:vSDd:a " ) ) ! = EOF ) {
1996-07-09 06:22:35 +00:00
switch ( c ) {
case ' f ' : /* output file name */
filename = optarg ;
break ;
case ' H ' : /* server host */
pghost = optarg ;
break ;
case ' p ' : /* server port */
pgport = optarg ;
break ;
case ' v ' : /* verbose */
g_verbose = 1 ;
break ;
case ' S ' : /* dump schema only */
schemaOnly = 1 ;
break ;
1996-07-12 05:39:39 +00:00
case ' d ' : /* dump data as proper insert strings */
dumpData = 1 ;
attrNames = strstr ( optarg , " a " ) ? 1 : 0 ;
break ;
case ' t ' : /* Dump data for this table only */
tablename = optarg ;
break ;
case ' a ' : /* Dump data only */
dataOnly = 1 ;
break ;
1996-07-09 06:22:35 +00:00
default :
usage ( progname ) ;
break ;
}
}
/* open the output file */
if ( filename = = NULL ) {
g_fout = stdout ;
} else {
g_fout = fopen ( filename , " w " ) ;
if ( g_fout = = NULL ) {
fprintf ( stderr , " %s: could not open output file named %s for writing \n " ,
progname , filename ) ;
exit ( 2 ) ;
}
}
/* find database */
if ( ! ( dbname = argv [ optind ] ) & &
! ( dbname = getenv ( " DATABASE " ) ) ) {
fprintf ( stderr , " %s: no database name specified \n " , progname ) ;
exit ( 2 ) ;
}
g_conn = PQsetdb ( pghost , pgport , NULL , NULL , dbname ) ;
/* check to see that the backend connection was successfully made */
if ( PQstatus ( g_conn ) = = CONNECTION_BAD ) {
fprintf ( stderr , " Connection to database '%s' failed. \n " , dbname ) ;
fprintf ( stderr , " %s " , PQerrorMessage ( g_conn ) ) ;
exit_nicely ( g_conn ) ;
}
g_last_builtin_oid = findLastBuiltinOid ( ) ;
1996-07-12 05:39:39 +00:00
if ( ! dataOnly ) {
1996-07-09 06:22:35 +00:00
if ( g_verbose )
fprintf ( stderr , " %s last builtin oid is %d %s \n " ,
g_comment_start , g_last_builtin_oid , g_comment_end ) ;
1996-07-12 05:39:39 +00:00
tblinfo = dumpSchema ( g_fout , & numTables , tablename ) ;
}
else {
tblinfo = dumpSchema ( NULL , & numTables , tablename ) ;
}
1996-07-09 06:22:35 +00:00
if ( ! schemaOnly ) {
if ( g_verbose ) fprintf ( stderr , " %s dumping out the contents of each table %s \n " ,
g_comment_start , g_comment_end ) ;
1996-07-12 05:39:39 +00:00
dumpClasses ( tblinfo , numTables , g_fout , tablename ) ;
1996-07-09 06:22:35 +00:00
}
fflush ( g_fout ) ;
fclose ( g_fout ) ;
PQfinish ( g_conn ) ;
exit ( 0 ) ;
}
/*
* getTypes :
* read all base types in the system catalogs and return them in the
* TypeInfo * structure
*
* numTypes is set to the number of types read in
*
*/
TypeInfo *
getTypes ( int * numTypes )
{
PGresult * res ;
int ntups ;
int i ;
char query [ MAXQUERYLEN ] ;
TypeInfo * tinfo ;
int i_oid ;
int i_typowner ;
int i_typname ;
int i_typlen ;
int i_typprtlen ;
int i_typinput ;
int i_typoutput ;
int i_typreceive ;
int i_typsend ;
int i_typelem ;
int i_typdelim ;
int i_typdefault ;
int i_typrelid ;
int i_typbyval ;
res = PQexec ( g_conn , " begin " ) ;
if ( ! res | |
PQresultStatus ( res ) ! = PGRES_COMMAND_OK ) {
fprintf ( stderr , " BEGIN command failed \n " ) ;
exit_nicely ( g_conn ) ;
}
PQclear ( res ) ;
/* find all base types */
/* we include even the built-in types
because those may be used as array elements by user - defined types */
/* we filter out the built-in types when
we dump out the types */
sprintf ( query , " SELECT oid, typowner,typname, typlen, typprtlen, typinput, typoutput, typreceive, typsend, typelem, typdelim, typdefault, typrelid,typbyval from pg_type " ) ;
res = PQexec ( g_conn , query ) ;
if ( ! res | |
PQresultStatus ( res ) ! = PGRES_TUPLES_OK ) {
fprintf ( stderr , " getTypes(): SELECT failed " ) ;
exit_nicely ( g_conn ) ;
}
ntups = PQntuples ( res ) ;
tinfo = ( TypeInfo * ) malloc ( ntups * sizeof ( TypeInfo ) ) ;
i_oid = PQfnumber ( res , " oid " ) ;
i_typowner = PQfnumber ( res , " typowner " ) ;
i_typname = PQfnumber ( res , " typname " ) ;
i_typlen = PQfnumber ( res , " typlen " ) ;
i_typprtlen = PQfnumber ( res , " typprtlen " ) ;
i_typinput = PQfnumber ( res , " typinput " ) ;
i_typoutput = PQfnumber ( res , " typoutput " ) ;
i_typreceive = PQfnumber ( res , " typreceive " ) ;
i_typsend = PQfnumber ( res , " typsend " ) ;
i_typelem = PQfnumber ( res , " typelem " ) ;
i_typdelim = PQfnumber ( res , " typdelim " ) ;
i_typdefault = PQfnumber ( res , " typdefault " ) ;
i_typrelid = PQfnumber ( res , " typrelid " ) ;
i_typbyval = PQfnumber ( res , " typbyval " ) ;
for ( i = 0 ; i < ntups ; i + + ) {
tinfo [ i ] . oid = dupstr ( PQgetvalue ( res , i , i_oid ) ) ;
tinfo [ i ] . typowner = dupstr ( PQgetvalue ( res , i , i_typowner ) ) ;
tinfo [ i ] . typname = dupstr ( PQgetvalue ( res , i , i_typname ) ) ;
tinfo [ i ] . typlen = dupstr ( PQgetvalue ( res , i , i_typlen ) ) ;
tinfo [ i ] . typprtlen = dupstr ( PQgetvalue ( res , i , i_typprtlen ) ) ;
tinfo [ i ] . typinput = dupstr ( PQgetvalue ( res , i , i_typinput ) ) ;
tinfo [ i ] . typoutput = dupstr ( PQgetvalue ( res , i , i_typoutput ) ) ;
tinfo [ i ] . typreceive = dupstr ( PQgetvalue ( res , i , i_typreceive ) ) ;
tinfo [ i ] . typsend = dupstr ( PQgetvalue ( res , i , i_typsend ) ) ;
tinfo [ i ] . typelem = dupstr ( PQgetvalue ( res , i , i_typelem ) ) ;
tinfo [ i ] . typdelim = dupstr ( PQgetvalue ( res , i , i_typdelim ) ) ;
tinfo [ i ] . typdefault = dupstr ( PQgetvalue ( res , i , i_typdefault ) ) ;
tinfo [ i ] . typrelid = dupstr ( PQgetvalue ( res , i , i_typrelid ) ) ;
if ( strcmp ( PQgetvalue ( res , i , i_typbyval ) , " f " ) = = 0 )
tinfo [ i ] . passedbyvalue = 0 ;
else
tinfo [ i ] . passedbyvalue = 1 ;
/* check for user-defined array types,
omit system generated ones */
if ( ( strcmp ( tinfo [ i ] . typelem , " 0 " ) ! = 0 ) & &
tinfo [ i ] . typname [ 0 ] ! = ' _ ' )
tinfo [ i ] . isArray = 1 ;
else
tinfo [ i ] . isArray = 0 ;
}
* numTypes = ntups ;
PQclear ( res ) ;
res = PQexec ( g_conn , " end " ) ;
PQclear ( res ) ;
return tinfo ;
}
/*
* getOperators :
* read all operators in the system catalogs and return them in the
* OprInfo * structure
*
* numOprs is set to the number of operators read in
*
*
*/
OprInfo *
getOperators ( int * numOprs )
{
PGresult * res ;
int ntups ;
int i ;
char query [ MAXQUERYLEN ] ;
OprInfo * oprinfo ;
int i_oid ;
int i_oprname ;
int i_oprkind ;
int i_oprcode ;
int i_oprleft ;
int i_oprright ;
int i_oprcom ;
int i_oprnegate ;
int i_oprrest ;
int i_oprjoin ;
int i_oprcanhash ;
int i_oprlsortop ;
int i_oprrsortop ;
/* find all operators, including builtin operators,
filter out system - defined operators at dump - out time */
res = PQexec ( g_conn , " begin " ) ;
if ( ! res | |
PQresultStatus ( res ) ! = PGRES_COMMAND_OK ) {
fprintf ( stderr , " BEGIN command failed \n " ) ;
exit_nicely ( g_conn ) ;
}
PQclear ( res ) ;
sprintf ( query , " SELECT oid, oprname, oprkind, oprcode, oprleft, oprright, oprcom, oprnegate, oprrest, oprjoin, oprcanhash, oprlsortop, oprrsortop from pg_operator " ) ;
res = PQexec ( g_conn , query ) ;
if ( ! res | |
PQresultStatus ( res ) ! = PGRES_TUPLES_OK ) {
fprintf ( stderr , " getOperators(): SELECT failed " ) ;
exit_nicely ( g_conn ) ;
}
ntups = PQntuples ( res ) ;
* numOprs = ntups ;
oprinfo = ( OprInfo * ) malloc ( ntups * sizeof ( OprInfo ) ) ;
i_oid = PQfnumber ( res , " oid " ) ;
i_oprname = PQfnumber ( res , " oprname " ) ;
i_oprkind = PQfnumber ( res , " oprkind " ) ;
i_oprcode = PQfnumber ( res , " oprcode " ) ;
i_oprleft = PQfnumber ( res , " oprleft " ) ;
i_oprright = PQfnumber ( res , " oprright " ) ;
i_oprcom = PQfnumber ( res , " oprcom " ) ;
i_oprnegate = PQfnumber ( res , " oprnegate " ) ;
i_oprrest = PQfnumber ( res , " oprrest " ) ;
i_oprjoin = PQfnumber ( res , " oprjoin " ) ;
i_oprcanhash = PQfnumber ( res , " oprcanhash " ) ;
i_oprlsortop = PQfnumber ( res , " oprlsortop " ) ;
i_oprrsortop = PQfnumber ( res , " oprrsortop " ) ;
for ( i = 0 ; i < ntups ; i + + ) {
oprinfo [ i ] . oid = dupstr ( PQgetvalue ( res , i , i_oid ) ) ;
oprinfo [ i ] . oprname = dupstr ( PQgetvalue ( res , i , i_oprname ) ) ;
oprinfo [ i ] . oprkind = dupstr ( PQgetvalue ( res , i , i_oprkind ) ) ;
oprinfo [ i ] . oprcode = dupstr ( PQgetvalue ( res , i , i_oprcode ) ) ;
oprinfo [ i ] . oprleft = dupstr ( PQgetvalue ( res , i , i_oprleft ) ) ;
oprinfo [ i ] . oprright = dupstr ( PQgetvalue ( res , i , i_oprright ) ) ;
oprinfo [ i ] . oprcom = dupstr ( PQgetvalue ( res , i , i_oprcom ) ) ;
oprinfo [ i ] . oprnegate = dupstr ( PQgetvalue ( res , i , i_oprnegate ) ) ;
oprinfo [ i ] . oprrest = dupstr ( PQgetvalue ( res , i , i_oprrest ) ) ;
oprinfo [ i ] . oprjoin = dupstr ( PQgetvalue ( res , i , i_oprjoin ) ) ;
oprinfo [ i ] . oprcanhash = dupstr ( PQgetvalue ( res , i , i_oprcanhash ) ) ;
oprinfo [ i ] . oprlsortop = dupstr ( PQgetvalue ( res , i , i_oprlsortop ) ) ;
oprinfo [ i ] . oprrsortop = dupstr ( PQgetvalue ( res , i , i_oprrsortop ) ) ;
}
PQclear ( res ) ;
res = PQexec ( g_conn , " end " ) ;
PQclear ( res ) ;
return oprinfo ;
}
/*
* getAggregates :
* read all the user - defined aggregates in the system catalogs and
* return them in the AggInfo * structure
*
* numAggs is set to the number of aggregates read in
*
*
*/
AggInfo *
getAggregates ( int * numAggs )
{
PGresult * res ;
int ntups ;
int i ;
char query [ MAXQUERYLEN ] ;
AggInfo * agginfo ;
int i_oid ;
int i_aggname ;
int i_aggtransfn1 ;
int i_aggtransfn2 ;
int i_aggfinalfn ;
int i_aggtranstype1 ;
int i_aggbasetype ;
int i_aggtranstype2 ;
int i_agginitval1 ;
int i_agginitval2 ;
/* find all user-defined aggregates */
res = PQexec ( g_conn , " begin " ) ;
if ( ! res | |
PQresultStatus ( res ) ! = PGRES_COMMAND_OK ) {
fprintf ( stderr , " BEGIN command failed \n " ) ;
exit_nicely ( g_conn ) ;
}
PQclear ( res ) ;
sprintf ( query ,
" SELECT oid, aggname, aggtransfn1, aggtransfn2, aggfinalfn, aggtranstype1, aggbasetype, aggtranstype2, agginitval1, agginitval2 from pg_aggregate; " ) ;
res = PQexec ( g_conn , query ) ;
if ( ! res | |
PQresultStatus ( res ) ! = PGRES_TUPLES_OK ) {
fprintf ( stderr , " getAggregates(): SELECT failed " ) ;
exit_nicely ( g_conn ) ;
}
ntups = PQntuples ( res ) ;
* numAggs = ntups ;
agginfo = ( AggInfo * ) malloc ( ntups * sizeof ( AggInfo ) ) ;
i_oid = PQfnumber ( res , " oid " ) ;
i_aggname = PQfnumber ( res , " aggname " ) ;
i_aggtransfn1 = PQfnumber ( res , " aggtransfn1 " ) ;
i_aggtransfn2 = PQfnumber ( res , " aggtransfn2 " ) ;
i_aggfinalfn = PQfnumber ( res , " aggfinalfn " ) ;
i_aggtranstype1 = PQfnumber ( res , " aggtranstype1 " ) ;
i_aggbasetype = PQfnumber ( res , " aggbasetype " ) ;
i_aggtranstype2 = PQfnumber ( res , " aggtranstype2 " ) ;
i_agginitval1 = PQfnumber ( res , " agginitval1 " ) ;
i_agginitval2 = PQfnumber ( res , " agginitval2 " ) ;
for ( i = 0 ; i < ntups ; i + + ) {
agginfo [ i ] . oid = dupstr ( PQgetvalue ( res , i , i_oid ) ) ;
agginfo [ i ] . aggname = dupstr ( PQgetvalue ( res , i , i_aggname ) ) ;
agginfo [ i ] . aggtransfn1 = dupstr ( PQgetvalue ( res , i , i_aggtransfn1 ) ) ;
agginfo [ i ] . aggtransfn2 = dupstr ( PQgetvalue ( res , i , i_aggtransfn2 ) ) ;
agginfo [ i ] . aggfinalfn = dupstr ( PQgetvalue ( res , i , i_aggfinalfn ) ) ;
agginfo [ i ] . aggtranstype1 = dupstr ( PQgetvalue ( res , i , i_aggtranstype1 ) ) ;
agginfo [ i ] . aggbasetype = dupstr ( PQgetvalue ( res , i , i_aggbasetype ) ) ;
agginfo [ i ] . aggtranstype2 = dupstr ( PQgetvalue ( res , i , i_aggtranstype2 ) ) ;
agginfo [ i ] . agginitval1 = dupstr ( PQgetvalue ( res , i , i_agginitval1 ) ) ;
agginfo [ i ] . agginitval2 = dupstr ( PQgetvalue ( res , i , i_agginitval2 ) ) ;
}
PQclear ( res ) ;
res = PQexec ( g_conn , " end " ) ;
PQclear ( res ) ;
return agginfo ;
}
/*
* getFuncs :
* read all the user - defined functions in the system catalogs and
* return them in the FuncInfo * structure
*
* numFuncs is set to the number of functions read in
*
*
*/
FuncInfo *
getFuncs ( int * numFuncs )
{
PGresult * res ;
int ntups ;
int i ;
char query [ MAXQUERYLEN ] ;
FuncInfo * finfo ;
int i_oid ;
int i_proname ;
int i_proowner ;
int i_prolang ;
int i_pronargs ;
int i_proargtypes ;
int i_prorettype ;
int i_proretset ;
int i_prosrc ;
int i_probin ;
/* find all user-defined funcs */
res = PQexec ( g_conn , " begin " ) ;
if ( ! res | |
PQresultStatus ( res ) ! = PGRES_COMMAND_OK ) {
fprintf ( stderr , " BEGIN command failed \n " ) ;
exit_nicely ( g_conn ) ;
}
PQclear ( res ) ;
sprintf ( query ,
" SELECT oid, proname, proowner, prolang, pronargs, prorettype, proretset, proargtypes, prosrc, probin from pg_proc where oid > '%d'::oid " ,
g_last_builtin_oid ) ;
res = PQexec ( g_conn , query ) ;
if ( ! res | |
PQresultStatus ( res ) ! = PGRES_TUPLES_OK ) {
fprintf ( stderr , " getFuncs(): SELECT failed " ) ;
exit_nicely ( g_conn ) ;
}
ntups = PQntuples ( res ) ;
* numFuncs = ntups ;
finfo = ( FuncInfo * ) malloc ( ntups * sizeof ( FuncInfo ) ) ;
i_oid = PQfnumber ( res , " oid " ) ;
i_proname = PQfnumber ( res , " proname " ) ;
i_proowner = PQfnumber ( res , " proowner " ) ;
i_prolang = PQfnumber ( res , " prolang " ) ;
i_pronargs = PQfnumber ( res , " pronargs " ) ;
i_proargtypes = PQfnumber ( res , " proargtypes " ) ;
i_prorettype = PQfnumber ( res , " prorettype " ) ;
i_proretset = PQfnumber ( res , " proretset " ) ;
i_prosrc = PQfnumber ( res , " prosrc " ) ;
i_probin = PQfnumber ( res , " probin " ) ;
for ( i = 0 ; i < ntups ; i + + ) {
finfo [ i ] . oid = dupstr ( PQgetvalue ( res , i , i_oid ) ) ;
finfo [ i ] . proname = dupstr ( PQgetvalue ( res , i , i_proname ) ) ;
finfo [ i ] . proowner = dupstr ( PQgetvalue ( res , i , i_proowner ) ) ;
finfo [ i ] . prosrc = checkForQuote ( PQgetvalue ( res , i , i_prosrc ) ) ;
finfo [ i ] . probin = dupstr ( PQgetvalue ( res , i , i_probin ) ) ;
finfo [ i ] . prorettype = dupstr ( PQgetvalue ( res , i , i_prorettype ) ) ;
finfo [ i ] . retset = ( strcmp ( PQgetvalue ( res , i , i_proretset ) , " t " ) = = 0 ) ;
finfo [ i ] . nargs = atoi ( PQgetvalue ( res , i , i_pronargs ) ) ;
finfo [ i ] . lang = ( atoi ( PQgetvalue ( res , i , i_prolang ) ) = = C_PROLANG_OID ) ;
parseArgTypes ( finfo [ i ] . argtypes , PQgetvalue ( res , i , i_proargtypes ) ) ;
finfo [ i ] . dumped = 0 ;
}
PQclear ( res ) ;
res = PQexec ( g_conn , " end " ) ;
PQclear ( res ) ;
return finfo ;
}
/*
* getTables
* read all the user - defined tables ( no indices , no catalogs )
* in the system catalogs return them in the TableInfo * structure
*
* numTables is set to the number of tables read in
*
*
*/
TableInfo *
getTables ( int * numTables )
{
PGresult * res ;
int ntups ;
int i ;
char query [ MAXQUERYLEN ] ;
TableInfo * tblinfo ;
int i_oid ;
int i_relname ;
int i_relarch ;
/* find all the user-defined tables (no indices and no catalogs),
ordering by oid is important so that we always process the parent
tables before the child tables when traversing the tblinfo *
we ignore tables that start with Xinv */
res = PQexec ( g_conn , " begin " ) ;
if ( ! res | |
PQresultStatus ( res ) ! = PGRES_COMMAND_OK ) {
fprintf ( stderr , " BEGIN command failed \n " ) ;
exit_nicely ( g_conn ) ;
}
PQclear ( res ) ;
sprintf ( query ,
" SELECT oid, relname, relarch from pg_class where relkind = 'r' and relname !~ '^pg_' and relname !~ '^Xinv' order by oid; " ) ;
res = PQexec ( g_conn , query ) ;
if ( ! res | |
PQresultStatus ( res ) ! = PGRES_TUPLES_OK ) {
fprintf ( stderr , " getTables(): SELECT failed " ) ;
exit_nicely ( g_conn ) ;
}
ntups = PQntuples ( res ) ;
* numTables = ntups ;
tblinfo = ( TableInfo * ) malloc ( ntups * sizeof ( TableInfo ) ) ;
i_oid = PQfnumber ( res , " oid " ) ;
i_relname = PQfnumber ( res , " relname " ) ;
i_relarch = PQfnumber ( res , " relarch " ) ;
for ( i = 0 ; i < ntups ; i + + ) {
tblinfo [ i ] . oid = dupstr ( PQgetvalue ( res , i , i_oid ) ) ;
tblinfo [ i ] . relname = dupstr ( PQgetvalue ( res , i , i_relname ) ) ;
tblinfo [ i ] . relarch = dupstr ( PQgetvalue ( res , i , i_relarch ) ) ;
}
PQclear ( res ) ;
res = PQexec ( g_conn , " end " ) ;
PQclear ( res ) ;
return tblinfo ;
}
/*
* getInherits
* read all the inheritance information
* from the system catalogs return them in the InhInfo * structure
*
* numInherits is set to the number of tables read in
*
*
*/
InhInfo *
getInherits ( int * numInherits )
{
PGresult * res ;
int ntups ;
int i ;
char query [ MAXQUERYLEN ] ;
InhInfo * inhinfo ;
int i_inhrel ;
int i_inhparent ;
/* find all the inheritance information */
res = PQexec ( g_conn , " begin " ) ;
if ( ! res | |
PQresultStatus ( res ) ! = PGRES_COMMAND_OK ) {
fprintf ( stderr , " BEGIN command failed \n " ) ;
exit_nicely ( g_conn ) ;
}
PQclear ( res ) ;
sprintf ( query , " SELECT inhrel, inhparent from pg_inherits " ) ;
res = PQexec ( g_conn , query ) ;
if ( ! res | |
PQresultStatus ( res ) ! = PGRES_TUPLES_OK ) {
fprintf ( stderr , " getInherits(): SELECT failed " ) ;
exit_nicely ( g_conn ) ;
}
ntups = PQntuples ( res ) ;
* numInherits = ntups ;
inhinfo = ( InhInfo * ) malloc ( ntups * sizeof ( InhInfo ) ) ;
i_inhrel = PQfnumber ( res , " inhrel " ) ;
i_inhparent = PQfnumber ( res , " inhparent " ) ;
for ( i = 0 ; i < ntups ; i + + ) {
inhinfo [ i ] . inhrel = dupstr ( PQgetvalue ( res , i , i_inhrel ) ) ;
inhinfo [ i ] . inhparent = dupstr ( PQgetvalue ( res , i , i_inhparent ) ) ;
}
PQclear ( res ) ;
res = PQexec ( g_conn , " end " ) ;
PQclear ( res ) ;
return inhinfo ;
}
/*
* getTableAttrs -
* for each table in tblinfo , read its attributes types and names
*
* this is implemented in a very inefficient way right now , looping
* through the tblinfo and doing a join per table to find the attrs and their
* types
*
* modifies tblinfo
*/
void
getTableAttrs ( TableInfo * tblinfo , int numTables )
{
int i , j ;
char q [ MAXQUERYLEN ] ;
int i_attname ;
int i_typname ;
1996-07-22 08:37:00 +00:00
int i_attlen ;
1996-07-09 06:22:35 +00:00
PGresult * res ;
int ntups ;
for ( i = 0 ; i < numTables ; i + + ) {
/* skip archive tables */
if ( isArchiveName ( tblinfo [ i ] . relname ) )
continue ;
/* find all the user attributes and their types*/
/* we must read the attribute names in attribute number order! */
/* because we will use the attnum to index into the attnames array
later */
if ( g_verbose )
fprintf ( stderr , " %s finding the attrs and types for table: %s %s \n " ,
g_comment_start ,
tblinfo [ i ] . relname ,
g_comment_end ) ;
1996-07-22 08:37:00 +00:00
sprintf ( q , " SELECT a.attnum, a.attname, t.typname, a.attlen from pg_attribute a, pg_type t where a.attrelid = '%s'::oid and a.atttypid = t.oid and a.attnum > 0 order by attnum " , tblinfo [ i ] . oid ) ;
1996-07-09 06:22:35 +00:00
res = PQexec ( g_conn , q ) ;
if ( ! res | |
PQresultStatus ( res ) ! = PGRES_TUPLES_OK ) {
fprintf ( stderr , " getTableAttrs(): SELECT failed " ) ;
exit_nicely ( g_conn ) ;
}
ntups = PQntuples ( res ) ;
i_attname = PQfnumber ( res , " attname " ) ;
i_typname = PQfnumber ( res , " typname " ) ;
1996-07-22 08:37:00 +00:00
i_attlen = PQfnumber ( res , " attlen " ) ;
1996-07-09 06:22:35 +00:00
tblinfo [ i ] . numatts = ntups ;
tblinfo [ i ] . attnames = ( char * * ) malloc ( ntups * sizeof ( char * ) ) ;
tblinfo [ i ] . typnames = ( char * * ) malloc ( ntups * sizeof ( char * ) ) ;
1996-07-22 08:37:00 +00:00
tblinfo [ i ] . attlen = ( int * ) malloc ( ntups * sizeof ( int ) ) ;
1996-07-09 06:22:35 +00:00
tblinfo [ i ] . inhAttrs = ( int * ) malloc ( ntups * sizeof ( int ) ) ;
tblinfo [ i ] . parentRels = NULL ;
tblinfo [ i ] . numParents = 0 ;
for ( j = 0 ; j < ntups ; j + + ) {
tblinfo [ i ] . attnames [ j ] = dupstr ( PQgetvalue ( res , j , i_attname ) ) ;
tblinfo [ i ] . typnames [ j ] = dupstr ( PQgetvalue ( res , j , i_typname ) ) ;
1996-07-22 08:37:00 +00:00
tblinfo [ i ] . attlen [ j ] = atoi ( PQgetvalue ( res , j , i_attlen ) ) ;
if ( tblinfo [ i ] . attlen [ j ] > 0 )
tblinfo [ i ] . attlen [ j ] = tblinfo [ i ] . attlen [ j ] - 4 ;
1996-07-09 06:22:35 +00:00
tblinfo [ i ] . inhAttrs [ j ] = 0 ; /* this flag is set in flagInhAttrs()*/
}
PQclear ( res ) ;
}
}
/*
* getIndices
* read all the user - defined indices information
* from the system catalogs return them in the InhInfo * structure
*
* numIndices is set to the number of indices read in
*
*
*/
IndInfo *
getIndices ( int * numIndices )
{
int i ;
char query [ MAXQUERYLEN ] ;
PGresult * res ;
int ntups ;
IndInfo * indinfo ;
int i_indexrelname ;
int i_indrelname ;
int i_indamname ;
int i_indproc ;
int i_indkey ;
int i_indclassname ;
/* find all the user-defined indices.
We do not handle partial indices .
We also assume that only single key indices
skip ' Xinx * ' - indices on inversion objects
this is a 5 - way join ! !
*/
res = PQexec ( g_conn , " begin " ) ;
if ( ! res | |
PQresultStatus ( res ) ! = PGRES_COMMAND_OK ) {
fprintf ( stderr , " BEGIN command failed \n " ) ;
exit_nicely ( g_conn ) ;
}
PQclear ( res ) ;
sprintf ( query ,
" SELECT t1.relname as indexrelname, t2.relname as indrelname, i.indproc, i.indkey[0], o.opcname as indclassname, a.amname as indamname from pg_index i, pg_class t1, pg_class t2, pg_opclass o, pg_am a where t1.oid = i.indexrelid and t2.oid = i.indrelid and o.oid = i.indclass[0] and t1.relam = a.oid and i.indexrelid > '%d'::oid and t2.relname !~ '^pg_' and t1.relname !~ '^Xinx' ; " ,
g_last_builtin_oid ) ;
res = PQexec ( g_conn , query ) ;
if ( ! res | |
PQresultStatus ( res ) ! = PGRES_TUPLES_OK ) {
fprintf ( stderr , " getIndices(): SELECT failed " ) ;
exit_nicely ( g_conn ) ;
}
ntups = PQntuples ( res ) ;
* numIndices = ntups ;
indinfo = ( IndInfo * ) malloc ( ntups * sizeof ( IndInfo ) ) ;
i_indexrelname = PQfnumber ( res , " indexrelname " ) ;
i_indrelname = PQfnumber ( res , " indrelname " ) ;
i_indamname = PQfnumber ( res , " indamname " ) ;
i_indproc = PQfnumber ( res , " indproc " ) ;
i_indkey = PQfnumber ( res , " indkey " ) ;
i_indclassname = PQfnumber ( res , " indclassname " ) ;
for ( i = 0 ; i < ntups ; i + + ) {
indinfo [ i ] . indexrelname = dupstr ( PQgetvalue ( res , i , i_indexrelname ) ) ;
indinfo [ i ] . indrelname = dupstr ( PQgetvalue ( res , i , i_indrelname ) ) ;
indinfo [ i ] . indamname = dupstr ( PQgetvalue ( res , i , i_indamname ) ) ;
indinfo [ i ] . indproc = dupstr ( PQgetvalue ( res , i , i_indproc ) ) ;
indinfo [ i ] . indkey = dupstr ( PQgetvalue ( res , i , i_indkey ) ) ;
indinfo [ i ] . indclassname = dupstr ( PQgetvalue ( res , i , i_indclassname ) ) ;
}
PQclear ( res ) ;
res = PQexec ( g_conn , " end " ) ;
return indinfo ;
}
/*
* dumpTypes
* writes out to fout the queries to recreate all the user - defined types
*
*/
void
dumpTypes ( FILE * fout , FuncInfo * finfo , int numFuncs ,
TypeInfo * tinfo , int numTypes )
{
int i ;
char q [ MAXQUERYLEN ] ;
int funcInd ;
for ( i = 0 ; i < numTypes ; i + + ) {
/* skip all the builtin types */
if ( atoi ( tinfo [ i ] . oid ) < g_last_builtin_oid )
continue ;
/* skip relation types */
if ( atoi ( tinfo [ i ] . typrelid ) ! = 0 )
continue ;
/* skip all array types that start w/ underscore */
if ( ( tinfo [ i ] . typname [ 0 ] = = ' _ ' ) & &
( strcmp ( tinfo [ i ] . typinput , " array_in " ) = = 0 ) )
continue ;
/* before we create a type, we need to create the input and
output functions for it , if they haven ' t been created already */
funcInd = findFuncByName ( finfo , numFuncs , tinfo [ i ] . typinput ) ;
if ( funcInd ! = - 1 )
dumpOneFunc ( fout , finfo , funcInd , tinfo , numTypes ) ;
funcInd = findFuncByName ( finfo , numFuncs , tinfo [ i ] . typoutput ) ;
if ( funcInd ! = - 1 )
dumpOneFunc ( fout , finfo , funcInd , tinfo , numTypes ) ;
sprintf ( q ,
" CREATE TYPE %s ( internallength = %s, externallength = %s, input = %s, output = %s, send = %s, receive = %s, default = '%s' " ,
tinfo [ i ] . typname ,
tinfo [ i ] . typlen ,
tinfo [ i ] . typprtlen ,
tinfo [ i ] . typinput ,
tinfo [ i ] . typoutput ,
tinfo [ i ] . typsend ,
tinfo [ i ] . typreceive ,
tinfo [ i ] . typdefault ) ;
if ( tinfo [ i ] . isArray ) {
char * elemType ;
elemType = findTypeByOid ( tinfo , numTypes , tinfo [ i ] . typelem ) ;
sprintf ( q , " %s, element = %s, delimiter = '%s' " ,
q , elemType , tinfo [ i ] . typdelim ) ;
}
if ( tinfo [ i ] . passedbyvalue )
strcat ( q , " ,passedbyvalue); \n " ) ;
else
strcat ( q , " ); \n " ) ;
fputs ( q , fout ) ;
}
}
/*
* dumpFuncs
* writes out to fout the queries to recreate all the user - defined functions
*
*/
void
dumpFuncs ( FILE * fout , FuncInfo * finfo , int numFuncs ,
TypeInfo * tinfo , int numTypes )
{
int i ;
for ( i = 0 ; i < numFuncs ; i + + ) {
dumpOneFunc ( fout , finfo , i , tinfo , numTypes ) ;
}
}
/*
* dumpOneFunc :
* dump out only one function , the index of which is given in the third
* argument
*
*/
void
dumpOneFunc ( FILE * fout , FuncInfo * finfo , int i ,
TypeInfo * tinfo , int numTypes )
{
char q [ MAXQUERYLEN ] ;
int j ;
if ( finfo [ i ] . dumped )
return ;
else
finfo [ i ] . dumped = 1 ;
sprintf ( q , " CREATE FUNCTION %s ( " , finfo [ i ] . proname ) ;
for ( j = 0 ; j < finfo [ i ] . nargs ; j + + ) {
char * typname ;
typname = findTypeByOid ( tinfo , numTypes , finfo [ i ] . argtypes [ j ] ) ;
sprintf ( q , " %s%s%s " ,
q ,
( j > 0 ) ? " , " : " " ,
typname ) ;
}
sprintf ( q , " %s ) RETURNS %s%s AS '%s' LANGUAGE '%s'; \n " ,
q ,
finfo [ i ] . retset ? " SETOF " : " " ,
findTypeByOid ( tinfo , numTypes , finfo [ i ] . prorettype ) ,
( finfo [ i ] . lang ) ? finfo [ i ] . probin : finfo [ i ] . prosrc ,
( finfo [ i ] . lang ) ? " C " : " SQL " ) ;
fputs ( q , fout ) ;
}
/*
* dumpOprs
* writes out to fout the queries to recreate all the user - defined operators
*
*/
void
dumpOprs ( FILE * fout , OprInfo * oprinfo , int numOperators ,
TypeInfo * tinfo , int numTypes )
{
int i ;
char q [ MAXQUERYLEN ] ;
char leftarg [ MAXQUERYLEN ] ;
char rightarg [ MAXQUERYLEN ] ;
char commutator [ MAXQUERYLEN ] ;
char negator [ MAXQUERYLEN ] ;
char restrict [ MAXQUERYLEN ] ;
char join [ MAXQUERYLEN ] ;
char sortop [ MAXQUERYLEN ] ;
for ( i = 0 ; i < numOperators ; i + + ) {
/* skip all the builtin oids */
if ( atoi ( oprinfo [ i ] . oid ) < g_last_builtin_oid )
continue ;
/* some operator are invalid because they were the result
of user defining operators before commutators exist */
if ( strcmp ( oprinfo [ i ] . oprcode , " - " ) = = 0 )
continue ;
leftarg [ 0 ] = ' \0 ' ;
rightarg [ 0 ] = ' \0 ' ;
/* right unary means there's a left arg
and left unary means there ' s a right arg */
if ( strcmp ( oprinfo [ i ] . oprkind , " r " ) = = 0 | |
strcmp ( oprinfo [ i ] . oprkind , " b " ) = = 0 ) {
sprintf ( leftarg , " , LEFTARG = %s " ,
findTypeByOid ( tinfo , numTypes , oprinfo [ i ] . oprleft ) ) ;
}
if ( strcmp ( oprinfo [ i ] . oprkind , " l " ) = = 0 | |
strcmp ( oprinfo [ i ] . oprkind , " b " ) = = 0 ) {
sprintf ( rightarg , " , RIGHTARG = %s " ,
findTypeByOid ( tinfo , numTypes , oprinfo [ i ] . oprright ) ) ;
}
if ( strcmp ( oprinfo [ i ] . oprcom , " 0 " ) = = 0 )
commutator [ 0 ] = ' \0 ' ;
else
sprintf ( commutator , " , COMMUTATOR = %s " ,
findOprByOid ( oprinfo , numOperators , oprinfo [ i ] . oprcom ) ) ;
if ( strcmp ( oprinfo [ i ] . oprnegate , " 0 " ) = = 0 )
negator [ 0 ] = ' \0 ' ;
else
sprintf ( negator , " , NEGATOR = %s " ,
findOprByOid ( oprinfo , numOperators , oprinfo [ i ] . oprnegate ) ) ;
if ( strcmp ( oprinfo [ i ] . oprrest , " - " ) = = 0 )
restrict [ 0 ] = ' \0 ' ;
else
sprintf ( restrict , " , RESTRICT = %s " , oprinfo [ i ] . oprrest ) ;
if ( strcmp ( oprinfo [ i ] . oprjoin , " - " ) = = 0 )
join [ 0 ] = ' \0 ' ;
else
sprintf ( join , " , JOIN = %s " , oprinfo [ i ] . oprjoin ) ;
if ( strcmp ( oprinfo [ i ] . oprlsortop , " 0 " ) = = 0 )
sortop [ 0 ] = ' \0 ' ;
else
{
sprintf ( sortop , " , SORT = %s " ,
findOprByOid ( oprinfo , numOperators ,
oprinfo [ i ] . oprlsortop ) ) ;
if ( strcmp ( oprinfo [ i ] . oprrsortop , " 0 " ) ! = 0 )
sprintf ( sortop , " %s , %s " , sortop ,
findOprByOid ( oprinfo , numOperators ,
oprinfo [ i ] . oprlsortop ) ) ;
}
sprintf ( q ,
" CREATE OPERATOR %s (PROCEDURE = %s %s %s %s %s %s %s %s %s); \n " ,
oprinfo [ i ] . oprname ,
oprinfo [ i ] . oprcode ,
leftarg ,
rightarg ,
commutator ,
negator ,
restrict ,
( strcmp ( oprinfo [ i ] . oprcanhash , " t " ) ) ? " , HASHES " : " " ,
join ,
sortop ) ;
fputs ( q , fout ) ;
}
}
/*
* dumpAggs
* writes out to fout the queries to create all the user - defined aggregates
*
*/
void
dumpAggs ( FILE * fout , AggInfo * agginfo , int numAggs ,
TypeInfo * tinfo , int numTypes )
{
int i ;
char q [ MAXQUERYLEN ] ;
char sfunc1 [ MAXQUERYLEN ] ;
char sfunc2 [ MAXQUERYLEN ] ;
char finalfunc [ MAXQUERYLEN ] ;
char comma1 [ 2 ] , comma2 [ 2 ] ;
for ( i = 0 ; i < numAggs ; i + + ) {
/* skip all the builtin oids */
if ( atoi ( agginfo [ i ] . oid ) < g_last_builtin_oid )
continue ;
if ( strcmp ( agginfo [ i ] . aggtransfn1 , " - " ) = = 0 )
sfunc1 [ 0 ] = ' \0 ' ;
else {
sprintf ( sfunc1 ,
" SFUNC1 = %s, BASETYPE = %s, STYPE1 = %s " ,
agginfo [ i ] . aggtransfn1 ,
findTypeByOid ( tinfo , numTypes , agginfo [ i ] . aggbasetype ) ,
findTypeByOid ( tinfo , numTypes , agginfo [ i ] . aggtranstype1 ) ) ;
if ( agginfo [ i ] . agginitval1 )
sprintf ( sfunc1 , " %s ,INITCOND1 = '%s' " ,
sfunc1 , agginfo [ i ] . agginitval1 ) ;
}
if ( strcmp ( agginfo [ i ] . aggtransfn2 , " - " ) = = 0 )
sfunc2 [ 0 ] = ' \0 ' ;
else {
sprintf ( sfunc2 ,
" SFUNC2 = %s, STYPE2 = %s " ,
agginfo [ i ] . aggtransfn2 ,
findTypeByOid ( tinfo , numTypes , agginfo [ i ] . aggtranstype2 ) ) ;
if ( agginfo [ i ] . agginitval2 )
sprintf ( sfunc2 , " %s ,INITCOND2 = '%s' " ,
sfunc2 , agginfo [ i ] . agginitval2 ) ;
}
if ( strcmp ( agginfo [ i ] . aggfinalfn , " - " ) = = 0 )
finalfunc [ 0 ] = ' \0 ' ;
else {
sprintf ( finalfunc , " FINALFUNC = %s " , agginfo [ i ] . aggfinalfn ) ;
}
if ( sfunc1 [ 0 ] ! = ' \0 ' & & sfunc2 [ 0 ] ! = ' \0 ' ) {
comma1 [ 0 ] = ' , ' ; comma1 [ 1 ] = ' \0 ' ;
} else
comma1 [ 0 ] = ' \0 ' ;
if ( finalfunc [ 0 ] ! = ' \0 ' & & ( sfunc1 [ 0 ] ! = ' \0 ' | | sfunc2 [ 0 ] ! = ' \0 ' ) ) {
comma2 [ 0 ] = ' , ' ; comma2 [ 1 ] = ' \0 ' ;
} else
comma2 [ 0 ] = ' \0 ' ;
sprintf ( q , " CREATE AGGREGATE %s ( %s %s %s %s %s ); \n " ,
agginfo [ i ] . aggname ,
sfunc1 ,
comma1 ,
sfunc2 ,
comma2 ,
finalfunc ) ;
fputs ( q , fout ) ;
}
}
/*
* dumpTables :
* write out to fout all the user - define tables
*/
1996-07-12 05:39:39 +00:00
void dumpTables ( FILE * fout , TableInfo * tblinfo , int numTables ,
1996-07-09 06:22:35 +00:00
InhInfo * inhinfo , int numInherits ,
1996-07-12 05:39:39 +00:00
TypeInfo * tinfo , int numTypes , char * tablename )
1996-07-09 06:22:35 +00:00
{
int i , j , k ;
char q [ MAXQUERYLEN ] ;
char * * parentRels ; /* list of names of parent relations */
int numParents ;
int actual_atts ; /* number of attrs in this CREATE statment */
char * archiveMode ;
for ( i = 0 ; i < numTables ; i + + ) {
1996-07-12 05:39:39 +00:00
if ( ! tablename | | ( ! strcmp ( tblinfo [ i ] . relname , tablename ) ) ) {
/* skip archive names*/
if ( isArchiveName ( tblinfo [ i ] . relname ) )
continue ;
parentRels = tblinfo [ i ] . parentRels ;
numParents = tblinfo [ i ] . numParents ;
sprintf ( q , " CREATE TABLE %s ( " , tblinfo [ i ] . relname ) ;
actual_atts = 0 ;
for ( j = 0 ; j < tblinfo [ i ] . numatts ; j + + ) {
if ( tblinfo [ i ] . inhAttrs [ j ] = = 0 ) {
1996-07-22 08:37:00 +00:00
/* Show lengths on bpchar and varchar */
if ( ! strcmp ( tblinfo [ i ] . typnames [ j ] , " bpchar " ) ) {
1996-07-27 02:29:51 +00:00
sprintf ( q , " %s%s%s char " ,
1996-07-22 08:37:00 +00:00
q ,
( actual_atts > 0 ) ? " , " : " " ,
1996-07-27 02:29:51 +00:00
tblinfo [ i ] . attnames [ j ] ) ;
/* stored length can be -1 (variable) */
if ( tblinfo [ i ] . attlen [ j ] > 0 )
sprintf ( q , " %s(%d) " ,
q ,
tblinfo [ i ] . attlen [ j ] ) ;
1996-07-22 08:37:00 +00:00
actual_atts + + ;
}
else if ( ! strcmp ( tblinfo [ i ] . typnames [ j ] , " varchar " ) ) {
1996-07-27 02:29:51 +00:00
sprintf ( q , " %s%s%s %s " ,
1996-07-22 08:37:00 +00:00
q ,
( actual_atts > 0 ) ? " , " : " " ,
tblinfo [ i ] . attnames [ j ] ,
1996-07-27 02:29:51 +00:00
tblinfo [ i ] . typnames [ j ] ) ;
/* stored length can be -1 (variable) */
if ( tblinfo [ i ] . attlen [ j ] > 0 )
sprintf ( q , " %s(%d) " ,
q ,
tblinfo [ i ] . attlen [ j ] ) ;
1996-07-22 08:37:00 +00:00
actual_atts + + ;
}
else {
sprintf ( q , " %s%s%s %s " ,
q ,
( actual_atts > 0 ) ? " , " : " " ,
tblinfo [ i ] . attnames [ j ] ,
tblinfo [ i ] . typnames [ j ] ) ;
actual_atts + + ;
}
1996-07-12 05:39:39 +00:00
}
1996-07-09 06:22:35 +00:00
}
1996-07-12 05:39:39 +00:00
strcat ( q , " ) " ) ;
1996-07-09 06:22:35 +00:00
1996-07-12 05:39:39 +00:00
if ( numParents > 0 ) {
sprintf ( q , " %s inherits ( " , q ) ;
for ( k = 0 ; k < numParents ; k + + ) {
1996-07-09 06:22:35 +00:00
sprintf ( q , " %s%s%s " ,
1996-07-12 05:39:39 +00:00
q ,
( k > 0 ) ? " , " : " " ,
parentRels [ k ] ) ;
}
strcat ( q , " ) " ) ;
1996-07-09 06:22:35 +00:00
}
1996-07-12 05:39:39 +00:00
switch ( tblinfo [ i ] . relarch [ 0 ] ) {
case ' n ' :
archiveMode = " none " ;
break ;
case ' h ' :
archiveMode = " heavy " ;
break ;
case ' l ' :
archiveMode = " light " ;
break ;
default :
fprintf ( stderr , " unknown archive mode \n " ) ;
archiveMode = " none " ;
break ;
}
1996-07-09 06:22:35 +00:00
1996-07-12 05:39:39 +00:00
sprintf ( q , " %s archive = %s; \n " ,
q ,
archiveMode ) ;
fputs ( q , fout ) ;
}
1996-07-09 06:22:35 +00:00
}
}
/*
* dumpIndices :
* write out to fout all the user - define indices
*/
void
dumpIndices ( FILE * fout , IndInfo * indinfo , int numIndices ,
1996-07-12 05:39:39 +00:00
TableInfo * tblinfo , int numTables , char * tablename )
1996-07-09 06:22:35 +00:00
{
int i ;
int tableInd ;
char * attname ; /* the name of the indexed attribute */
char * funcname ; /* the name of the function to comput the index key from*/
int indkey ;
char q [ MAXQUERYLEN ] ;
PGresult * res ;
for ( i = 0 ; i < numIndices ; i + + ) {
tableInd = findTableByName ( tblinfo , numTables ,
indinfo [ i ] . indrelname ) ;
indkey = atoi ( indinfo [ i ] . indkey ) - 1 ;
1996-07-31 06:09:46 +00:00
if ( indkey = = ObjectIdAttributeNumber - 1 )
attname = " oid " ;
else
attname = tblinfo [ tableInd ] . attnames [ indkey ] ;
1996-07-09 06:22:35 +00:00
if ( strcmp ( indinfo [ i ] . indproc , " 0 " ) = = 0 ) {
funcname = NULL ;
} else {
/* the funcname is an oid which we use to
find the name of the pg_proc . We need to do this
because getFuncs ( ) only reads in the user - defined funcs
not all the funcs . We might not find what we want
by looking in FuncInfo * */
sprintf ( q ,
" SELECT proname from pg_proc where pg_proc.oid = '%s'::oid " ,
indinfo [ i ] . indproc ) ;
res = PQexec ( g_conn , q ) ;
funcname = dupstr ( PQgetvalue ( res , 0 ,
PQfnumber ( res , " proname " ) ) ) ;
PQclear ( res ) ;
}
1996-07-12 05:39:39 +00:00
if ( ! tablename | | ( ! strcmp ( indinfo [ i ] . indrelname , tablename ) ) ) {
sprintf ( q , " CREATE INDEX %s on %s using %s ( " ,
indinfo [ i ] . indexrelname ,
indinfo [ i ] . indrelname ,
indinfo [ i ] . indamname ) ;
if ( funcname ) {
sprintf ( q , " %s %s(%s) %s); \n " ,
q , funcname , attname , indinfo [ i ] . indclassname ) ;
free ( funcname ) ;
} else
sprintf ( q , " %s %s %s); \n " ,
q , attname , indinfo [ i ] . indclassname ) ;
fputs ( q , fout ) ;
}
1996-07-09 06:22:35 +00:00
}
}
/*
* DumpClasses -
* dump the contents of all the classes .
*/
void
1996-07-12 05:39:39 +00:00
dumpClasses ( TableInfo * tblinfo , int numTables , FILE * fout , char * onlytable )
1996-07-09 06:22:35 +00:00
{
char query [ 255 ] ;
# define COPYBUFSIZ 8192
char copybuf [ COPYBUFSIZ ] ;
1996-07-22 08:37:00 +00:00
char expandbuf [ COPYBUFSIZ ] ;
char * expsrc , * expdest ;
1996-07-12 05:39:39 +00:00
char q [ MAXQUERYLEN ] ;
1996-07-09 06:22:35 +00:00
PGresult * res ;
1996-07-12 05:39:39 +00:00
int i , j ;
int actual_atts ; /* number of attrs in this a table */
1996-07-09 06:22:35 +00:00
int ret ;
1996-07-12 05:39:39 +00:00
int field ;
int tuple ;
1996-07-09 06:22:35 +00:00
int copydone ;
for ( i = 0 ; i < numTables ; i + + ) {
char * classname = tblinfo [ i ] . relname ;
1996-07-12 05:39:39 +00:00
if ( ! onlytable | | ( ! strcmp ( classname , onlytable ) ) ) {
/* skip archive names*/
if ( isArchiveName ( classname ) )
continue ;
if ( ! dumpData ) {
fprintf ( fout , " COPY %s from stdin; \n " , classname ) ;
sprintf ( query , " COPY %s to stdout; \n " , classname ) ;
res = PQexec ( g_conn , query ) ;
if ( ! res | |
PQresultStatus ( res ) ! = PGRES_COPY_OUT ) {
fprintf ( stderr , " dumpClasses(): COPY to stdout failed " ) ;
exit_nicely ( g_conn ) ;
}
copydone = 0 ;
while ( ! copydone ) {
ret = PQgetline ( res - > conn , copybuf , COPYBUFSIZ ) ;
1996-07-09 06:22:35 +00:00
1996-07-12 05:39:39 +00:00
if ( copybuf [ 0 ] = = ' . ' & & copybuf [ 1 ] = = ' \0 ' ) {
copydone = true ; /* don't print this... */
} else {
fputs ( copybuf , stdout ) ;
switch ( ret ) {
case EOF :
copydone = true ;
/*FALLTHROUGH*/
case 0 :
fputc ( ' \n ' , stdout ) ;
break ;
case 1 :
break ;
}
}
}
fprintf ( fout , " . \n " ) ;
PQclear ( res ) ;
PQendcopy ( res - > conn ) ;
} else {
sprintf ( query , " select * from %s; \n " , classname ) ;
res = PQexec ( g_conn , query ) ;
if ( ! res | |
PQresultStatus ( res ) ! = PGRES_TUPLES_OK ) {
fprintf ( stderr , " dumpClasses(): command failed " ) ;
exit_nicely ( g_conn ) ;
}
tuple = 0 ;
while ( tuple < PQntuples ( res ) ) {
fprintf ( fout , " insert into %s " , classname ) ;
if ( attrNames ) {
actual_atts = 0 ;
sprintf ( q , " ( " ) ;
for ( j = 0 ; j < tblinfo [ i ] . numatts ; j + + ) {
if ( tblinfo [ i ] . inhAttrs [ j ] = = 0 ) {
sprintf ( q , " %s%s%s " ,
q ,
( actual_atts > 0 ) ? " , " : " " ,
tblinfo [ i ] . attnames [ j ] ) ;
actual_atts + + ;
}
}
sprintf ( q , " %s%s " , q , " ) " ) ;
fprintf ( fout , q ) ;
}
fprintf ( fout , " values ( " ) ;
field = 0 ;
do {
switch ( PQftype ( res , field ) ) {
case 21 : case 22 : case 23 : /* int types */
case 810 : case 910 : /* oldint types */
case 1005 : case 1006 : case 1007 : /* _int types */
case 700 : case 701 : /* float types */
case 1021 : case 1022 : /* _float types */
fprintf ( fout , " %s " , PQgetvalue ( res , tuple , field ) ) ;
break ;
default :
1996-07-22 08:37:00 +00:00
/* Before outputing string value, expand all
single quotes to twin single quotes -
dhb - 6 / 11 / 96 */
expsrc = PQgetvalue ( res , tuple , field ) ;
expdest = expandbuf ;
while ( * expsrc ) {
* expdest + + = * expsrc ;
if ( * expsrc = = ( char ) 0x27 ) /*sing. quote*/
* expdest + + = * expsrc ;
expsrc + + ;
}
* expdest = * expsrc ; /* null term. */
fprintf ( fout , " '%s' " , expandbuf ) ;
1996-07-12 05:39:39 +00:00
break ;
}
field + + ;
if ( field ! = PQnfields ( res ) )
fprintf ( fout , " , " ) ;
} while ( field < PQnfields ( res ) ) ;
fprintf ( fout , " ); \n " ) ;
tuple + + ;
}
PQclear ( res ) ;
1996-07-09 06:22:35 +00:00
}
1996-07-12 05:39:39 +00:00
}
}
1996-07-09 06:22:35 +00:00
}
/*
* dumpTuples - -
* prints out the tuples in ASCII representation . The output is a valid
* input to COPY FROM stdin .
*
* We only need to do this for POSTGRES 4.2 databases since the
* COPY TO statment doesn ' t escape newlines properly . It ' s been fixed
* in Postgres95 .
*
* the attrmap passed in tells how to map the attributes copied in to the
* attributes copied out
*/
void
dumpTuples ( PGresult * res , FILE * fout , int * attrmap )
{
int j , k ;
int m , n ;
char * * outVals = NULL ; /* values to copy out */
n = PQntuples ( res ) ;
m = PQnfields ( res ) ;
if ( m > 0 ) {
/*
* Print out the tuples but only print tuples with at least
* 1 field .
*/
outVals = ( char * * ) malloc ( m * sizeof ( char * ) ) ;
for ( j = 0 ; j < n ; j + + ) {
for ( k = 0 ; k < m ; k + + ) {
outVals [ attrmap [ k ] ] = PQgetvalue ( res , j , k ) ;
}
for ( k = 0 ; k < m ; k + + ) {
char * pval = outVals [ k ] ;
if ( k ! = 0 )
fputc ( ' \t ' , fout ) ; /* delimiter for attribute */
if ( pval ) {
while ( * pval ! = ' \0 ' ) {
/* escape tabs, newlines and backslashes */
if ( * pval = = ' \t ' | | * pval = = ' \n ' | | * pval = = ' \\ ' )
fputc ( ' \\ ' , fout ) ;
fputc ( * pval , fout ) ;
pval + + ;
}
}
}
fputc ( ' \n ' , fout ) ; /* delimiter for a tuple */
}
free ( outVals ) ;
}
}
/*
* findLastBuiltInOid -
* find the last built in oid
* we do this by looking up the oid of ' template1 ' in pg_database ,
* this is probably not foolproof but comes close
*/
int
findLastBuiltinOid ( )
{
PGresult * res ;
int ntups ;
int last_oid ;
res = PQexec ( g_conn ,
" SELECT oid from pg_database where datname = 'template1'; " ) ;
if ( res = = NULL | |
PQresultStatus ( res ) ! = PGRES_TUPLES_OK ) {
fprintf ( stderr , " pg_dump error in finding the template1 database " ) ;
exit_nicely ( g_conn ) ;
}
ntups = PQntuples ( res ) ;
if ( ntups ! = 1 ) {
fprintf ( stderr , " pg_dump: couldn't find the template1 database. You are really hosed \n Giving up \n " ) ;
exit_nicely ( g_conn ) ;
}
last_oid = atoi ( PQgetvalue ( res , 0 , PQfnumber ( res , " oid " ) ) ) ;
PQclear ( res ) ;
return last_oid ;
}
/*
* checkForQuote :
* checks a string for quote characters and quotes them
*/
char *
checkForQuote ( char * s )
{
char * r ;
char c ;
char * result ;
int j = 0 ;
r = malloc ( strlen ( s ) * 3 + 1 ) ; /* definitely long enough */
while ( ( c = * s ) ! = ' \0 ' ) {
if ( c = = ' \' ' ) {
r [ j + + ] = ' \' ' ; /* quote the single quotes */
}
r [ j + + ] = c ;
s + + ;
}
r [ j ] = ' \0 ' ;
result = dupstr ( r ) ;
free ( r ) ;
return result ;
}