mirror/postgresql
2
0
Fork 0
mirror of https://git.postgresql.org/git/postgresql.git synced 2024-12-09 08:10:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

Cosmetic changes to dblink.

Browse Source
This commit is contained in:
Bruce Momjian 2005-10-08 12:12:29 +00:00
parent 7d937cd408
commit f3576c33f8
1 changed files with 46 additions and 45 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

91
contrib/dblink/dblink.c
View File

@ -60,9 +60,10 @@
typedef struct remoteConn typedef struct remoteConn
{ {
PGconn *con; /* Hold the remote connection */ PGconn *conn; /* Hold the remote connection */
bool remoteTrFlag; /* Indicates whether or not a transaction int autoXactCursors; /* Indicates the number of open cursors,
* on remote database is in progress */ * non-zero means we opened the xact
* ourselves */
} remoteConn; } remoteConn;
/* /*
@ -70,7 +71,7 @@ typedef struct remoteConn
*/ */
static remoteConn *getConnectionByName(const char *name); static remoteConn *getConnectionByName(const char *name);
static HTAB *createConnHash(void); static HTAB *createConnHash(void);
static void createNewConnection(const char *name, remoteConn * con); static void createNewConnection(const char *name, remoteConn *rconn);
static void deleteConnection(const char *name); static void deleteConnection(const char *name);
static char **get_pkey_attnames(Oid relid, int16 *numatts); static char **get_pkey_attnames(Oid relid, int16 *numatts);
static char *get_sql_insert(Oid relid, int2vector *pkattnums, int16 pknumatts, char **src_pkattvals, char **tgt_pkattvals); static char *get_sql_insert(Oid relid, int2vector *pkattnums, int16 pknumatts, char **src_pkattvals, char **tgt_pkattvals);
@ -99,7 +100,7 @@ much like ecpg e.g. a mapping between a name and a PGconn object.
typedef struct remoteConnHashEnt typedef struct remoteConnHashEnt
{ {
char name[NAMEDATALEN]; char name[NAMEDATALEN];
remoteConn *rcon; remoteConn *rconn;
} remoteConnHashEnt; } remoteConnHashEnt;
/* initial number of connection hashes */ /* initial number of connection hashes */
@ -162,10 +163,10 @@ typedef struct remoteConnHashEnt
#define DBLINK_GET_CONN \ #define DBLINK_GET_CONN \
do { \ do { \
char *conname_or_str = GET_STR(PG_GETARG_TEXT_P(0)); \ char *conname_or_str = GET_STR(PG_GETARG_TEXT_P(0)); \
rcon = getConnectionByName(conname_or_str); \ rconn = getConnectionByName(conname_or_str); \
if(rcon) \ if(rconn) \
{ \ { \
conn = rcon->con; \ conn = rconn->conn; \
} \ } \
else \ else \
{ \ { \
@ -197,7 +198,7 @@ dblink_connect(PG_FUNCTION_ARGS)
char *msg; char *msg;
MemoryContext oldcontext; MemoryContext oldcontext;
PGconn *conn = NULL; PGconn *conn = NULL;
remoteConn *rcon = NULL; remoteConn *rconn = NULL;
if (PG_NARGS() == 2) if (PG_NARGS() == 2)
{ {
@ -210,7 +211,7 @@ dblink_connect(PG_FUNCTION_ARGS)
oldcontext = MemoryContextSwitchTo(TopMemoryContext); oldcontext = MemoryContextSwitchTo(TopMemoryContext);
if (connname) if (connname)
rcon = (remoteConn *) palloc(sizeof(remoteConn)); rconn = (remoteConn *) palloc(sizeof(remoteConn));
conn = PQconnectdb(connstr); conn = PQconnectdb(connstr);
MemoryContextSwitchTo(oldcontext); MemoryContextSwitchTo(oldcontext);
@ -219,8 +220,8 @@ dblink_connect(PG_FUNCTION_ARGS)
{ {
msg = pstrdup(PQerrorMessage(conn)); msg = pstrdup(PQerrorMessage(conn));
PQfinish(conn); PQfinish(conn);
if (rcon) if (rconn)
pfree(rcon); pfree(rconn);
ereport(ERROR, ereport(ERROR,
(errcode(ERRCODE_SQLCLIENT_UNABLE_TO_ESTABLISH_SQLCONNECTION), (errcode(ERRCODE_SQLCLIENT_UNABLE_TO_ESTABLISH_SQLCONNECTION),
@ -230,8 +231,8 @@ dblink_connect(PG_FUNCTION_ARGS)
if (connname) if (connname)
{ {
rcon->con = conn; rconn->conn = conn;
createNewConnection(connname, rcon); createNewConnection(connname, rconn);
} }
else else
persistent_conn = conn; persistent_conn = conn;
@ -247,15 +248,15 @@ Datum
dblink_disconnect(PG_FUNCTION_ARGS) dblink_disconnect(PG_FUNCTION_ARGS)
{ {
char *conname = NULL; char *conname = NULL;
remoteConn *rcon = NULL; remoteConn *rconn = NULL;
PGconn *conn = NULL; PGconn *conn = NULL;
if (PG_NARGS() == 1) if (PG_NARGS() == 1)
{ {
conname = GET_STR(PG_GETARG_TEXT_P(0)); conname = GET_STR(PG_GETARG_TEXT_P(0));
rcon = getConnectionByName(conname); rconn = getConnectionByName(conname);
if (rcon) if (rconn)
conn = rcon->con; conn = rconn->conn;
} }
else else
conn = persistent_conn; conn = persistent_conn;
@ -264,10 +265,10 @@ dblink_disconnect(PG_FUNCTION_ARGS)
DBLINK_CONN_NOT_AVAIL; DBLINK_CONN_NOT_AVAIL;
PQfinish(conn); PQfinish(conn);
if (rcon) if (rconn)
{ {
deleteConnection(conname); deleteConnection(conname);
pfree(rcon); pfree(rconn);
} }
else else
persistent_conn = NULL; persistent_conn = NULL;
@ -289,7 +290,7 @@ dblink_open(PG_FUNCTION_ARGS)
char *sql = NULL; char *sql = NULL;
char *conname = NULL; char *conname = NULL;
StringInfo str = makeStringInfo(); StringInfo str = makeStringInfo();
remoteConn *rcon = NULL; remoteConn *rconn = NULL;
bool fail = true; /* default to backward compatible behavior */ bool fail = true; /* default to backward compatible behavior */
if (PG_NARGS() == 2) if (PG_NARGS() == 2)
@ -314,9 +315,9 @@ dblink_open(PG_FUNCTION_ARGS)
conname = GET_STR(PG_GETARG_TEXT_P(0)); conname = GET_STR(PG_GETARG_TEXT_P(0));
curname = GET_STR(PG_GETARG_TEXT_P(1)); curname = GET_STR(PG_GETARG_TEXT_P(1));
sql = GET_STR(PG_GETARG_TEXT_P(2)); sql = GET_STR(PG_GETARG_TEXT_P(2));
rcon = getConnectionByName(conname); rconn = getConnectionByName(conname);
if (rcon) if (rconn)
conn = rcon->con; conn = rconn->conn;
} }
} }
else if (PG_NARGS() == 4) else if (PG_NARGS() == 4)
@ -326,9 +327,9 @@ dblink_open(PG_FUNCTION_ARGS)
curname = GET_STR(PG_GETARG_TEXT_P(1)); curname = GET_STR(PG_GETARG_TEXT_P(1));
sql = GET_STR(PG_GETARG_TEXT_P(2)); sql = GET_STR(PG_GETARG_TEXT_P(2));
fail = PG_GETARG_BOOL(3); fail = PG_GETARG_BOOL(3);
rcon = getConnectionByName(conname); rconn = getConnectionByName(conname);
if (rcon) if (rconn)
conn = rcon->con; conn = rconn->conn;
} }
if (!conn) if (!conn)
@ -370,7 +371,7 @@ dblink_close(PG_FUNCTION_ARGS)
char *conname = NULL; char *conname = NULL;
StringInfo str = makeStringInfo(); StringInfo str = makeStringInfo();
char *msg; char *msg;
remoteConn *rcon = NULL; remoteConn *rconn = NULL;
bool fail = true; /* default to backward compatible behavior */ bool fail = true; /* default to backward compatible behavior */
if (PG_NARGS() == 1) if (PG_NARGS() == 1)
@ -392,9 +393,9 @@ dblink_close(PG_FUNCTION_ARGS)
{ {
conname = GET_STR(PG_GETARG_TEXT_P(0)); conname = GET_STR(PG_GETARG_TEXT_P(0));
curname = GET_STR(PG_GETARG_TEXT_P(1)); curname = GET_STR(PG_GETARG_TEXT_P(1));
rcon = getConnectionByName(conname); rconn = getConnectionByName(conname);
if (rcon) if (rconn)
conn = rcon->con; conn = rconn->conn;
} }
} }
if (PG_NARGS() == 3) if (PG_NARGS() == 3)
@ -403,9 +404,9 @@ dblink_close(PG_FUNCTION_ARGS)
conname = GET_STR(PG_GETARG_TEXT_P(0)); conname = GET_STR(PG_GETARG_TEXT_P(0));
curname = GET_STR(PG_GETARG_TEXT_P(1)); curname = GET_STR(PG_GETARG_TEXT_P(1));
fail = PG_GETARG_BOOL(2); fail = PG_GETARG_BOOL(2);
rcon = getConnectionByName(conname); rconn = getConnectionByName(conname);
if (rcon) if (rconn)
conn = rcon->con; conn = rconn->conn;
} }
if (!conn) if (!conn)
@ -454,7 +455,7 @@ dblink_fetch(PG_FUNCTION_ARGS)
PGresult *res = NULL; PGresult *res = NULL;
MemoryContext oldcontext; MemoryContext oldcontext;
char *conname = NULL; char *conname = NULL;
remoteConn *rcon = NULL; remoteConn *rconn = NULL;
/* stuff done only on the first call of the function */ /* stuff done only on the first call of the function */
if (SRF_IS_FIRSTCALL()) if (SRF_IS_FIRSTCALL())
@ -473,9 +474,9 @@ dblink_fetch(PG_FUNCTION_ARGS)
howmany = PG_GETARG_INT32(2); howmany = PG_GETARG_INT32(2);
fail = PG_GETARG_BOOL(3); fail = PG_GETARG_BOOL(3);
rcon = getConnectionByName(conname); rconn = getConnectionByName(conname);
if (rcon) if (rconn)
conn = rcon->con; conn = rconn->conn;
} }
else if (PG_NARGS() == 3) else if (PG_NARGS() == 3)
{ {
@ -493,9 +494,9 @@ dblink_fetch(PG_FUNCTION_ARGS)
curname = GET_STR(PG_GETARG_TEXT_P(1)); curname = GET_STR(PG_GETARG_TEXT_P(1));
howmany = PG_GETARG_INT32(2); howmany = PG_GETARG_INT32(2);
rcon = getConnectionByName(conname); rconn = getConnectionByName(conname);
if (rcon) if (rconn)
conn = rcon->con; conn = rconn->conn;
} }
} }
else if (PG_NARGS() == 2) else if (PG_NARGS() == 2)
@ -656,7 +657,7 @@ dblink_record(PG_FUNCTION_ARGS)
char *connstr = NULL; char *connstr = NULL;
char *sql = NULL; char *sql = NULL;
char *conname = NULL; char *conname = NULL;
remoteConn *rcon = NULL; remoteConn *rconn = NULL;
bool fail = true; /* default to backward compatible */ bool fail = true; /* default to backward compatible */
/* create a function context for cross-call persistence */ /* create a function context for cross-call persistence */
@ -855,7 +856,7 @@ dblink_exec(PG_FUNCTION_ARGS)
char *connstr = NULL; char *connstr = NULL;
char *sql = NULL; char *sql = NULL;
char *conname = NULL; char *conname = NULL;
remoteConn *rcon = NULL; remoteConn *rconn = NULL;
bool freeconn = false; bool freeconn = false;
bool fail = true; /* default to backward compatible behavior */ bool fail = true; /* default to backward compatible behavior */
@ -2027,7 +2028,7 @@ getConnectionByName(const char *name)
key, HASH_FIND, NULL); key, HASH_FIND, NULL);
if (hentry) if (hentry)
return (hentry->rcon); return (hentry->rconn);
return (NULL); return (NULL);
} }
@ -2063,7 +2064,7 @@ createNewConnection(const char *name, remoteConn *rconn)
(errcode(ERRCODE_DUPLICATE_OBJECT), (errcode(ERRCODE_DUPLICATE_OBJECT),
errmsg("duplicate connection name"))); errmsg("duplicate connection name")));
hentry->rcon = rconn; hentry->rconn = rconn;
strncpy(hentry->name, name, NAMEDATALEN - 1); strncpy(hentry->name, name, NAMEDATALEN - 1);
} }