mirror of
https://git.postgresql.org/git/postgresql.git
synced 2024-12-27 08:39:28 +08:00
190 lines
3.2 KiB
C
190 lines
3.2 KiB
C
/*
|
|
* pginterface.c
|
|
*
|
|
*/
|
|
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <stdarg.h>
|
|
|
|
#include <libpq-fe.h>
|
|
#include "halt.h"
|
|
#include "pginterface.h"
|
|
|
|
#define NUL '\0'
|
|
|
|
/* GLOBAL VARIABLES */
|
|
static PGconn *conn;
|
|
static PGresult *res = NULL;
|
|
|
|
#define ON_ERROR_STOP 0
|
|
#define ON_ERROR_CONTINUE 1
|
|
|
|
static int on_error_state = ON_ERROR_STOP;
|
|
|
|
/* LOCAL VARIABLES */
|
|
static int tuple;
|
|
|
|
/*
|
|
**
|
|
** connectdb - returns PGconn structure
|
|
**
|
|
*/
|
|
PGconn *
|
|
connectdb(char *dbName,
|
|
char *pghost,
|
|
char *pgport,
|
|
char *pgoptions,
|
|
char *pgtty)
|
|
{
|
|
/* make a connection to the database */
|
|
conn = PQsetdb(pghost, pgport, pgoptions, pgtty, dbName);
|
|
if (PQstatus(conn) == CONNECTION_BAD)
|
|
halt("Connection to database '%s' failed.\n%s\n", dbName,
|
|
PQerrorMessage(conn));
|
|
return conn;
|
|
}
|
|
|
|
/*
|
|
**
|
|
** disconnectdb
|
|
**
|
|
*/
|
|
void
|
|
disconnectdb()
|
|
{
|
|
PQfinish(conn);
|
|
}
|
|
|
|
/*
|
|
**
|
|
** doquery - returns PGresult structure
|
|
**
|
|
*/
|
|
PGresult *
|
|
doquery(char *query)
|
|
{
|
|
if (res != NULL)
|
|
PQclear(res);
|
|
|
|
res = PQexec(conn, query);
|
|
|
|
if (on_error_state == ON_ERROR_STOP &&
|
|
(res == NULL ||
|
|
PQresultStatus(res) == PGRES_BAD_RESPONSE ||
|
|
PQresultStatus(res) == PGRES_NONFATAL_ERROR ||
|
|
PQresultStatus(res) == PGRES_FATAL_ERROR))
|
|
{
|
|
if (res != NULL)
|
|
fprintf(stderr, "query error: %s\n", PQcmdStatus(res));
|
|
else
|
|
fprintf(stderr, "connection error: %s\n", PQerrorMessage(conn));
|
|
PQfinish(conn);
|
|
halt("failed request: %s\n", query);
|
|
}
|
|
tuple = 0;
|
|
return res;
|
|
}
|
|
|
|
/*
|
|
**
|
|
** fetch - returns tuple number (starts at 0), or the value END_OF_TUPLES
|
|
** NULL pointers are skipped
|
|
**
|
|
*/
|
|
int
|
|
fetch(void *param,...)
|
|
{
|
|
va_list ap;
|
|
int arg,
|
|
num_fields;
|
|
|
|
num_fields = PQnfields(res);
|
|
|
|
if (tuple >= PQntuples(res))
|
|
return END_OF_TUPLES;
|
|
|
|
va_start(ap, param);
|
|
for (arg = 0; arg < num_fields; arg++)
|
|
{
|
|
if (param != NULL)
|
|
{
|
|
if (PQfsize(res, arg) == -1)
|
|
{
|
|
memcpy(param, PQgetvalue(res, tuple, arg), PQgetlength(res, tuple, arg));
|
|
((char *) param)[PQgetlength(res, tuple, arg)] = NUL;
|
|
}
|
|
else
|
|
memcpy(param, PQgetvalue(res, tuple, arg), PQfsize(res, arg));
|
|
}
|
|
param = va_arg(ap, char *);
|
|
}
|
|
va_end(ap);
|
|
return tuple++;
|
|
}
|
|
|
|
/*
|
|
**
|
|
** fetchwithnulls - returns tuple number (starts at 0),
|
|
** or the value END_OF_TUPLES
|
|
** Returns true or false into null indicator variables
|
|
** NULL pointers are skipped
|
|
*/
|
|
int
|
|
fetchwithnulls(void *param,...)
|
|
{
|
|
va_list ap;
|
|
int arg,
|
|
num_fields;
|
|
|
|
num_fields = PQnfields(res);
|
|
|
|
if (tuple >= PQntuples(res))
|
|
return END_OF_TUPLES;
|
|
|
|
va_start(ap, param);
|
|
for (arg = 0; arg < num_fields; arg++)
|
|
{
|
|
if (param != NULL)
|
|
{
|
|
if (PQfsize(res, arg) == -1)
|
|
{
|
|
memcpy(param, PQgetvalue(res, tuple, arg), PQgetlength(res, tuple, arg));
|
|
((char *) param)[PQgetlength(res, tuple, arg)] = NUL;
|
|
}
|
|
else
|
|
memcpy(param, PQgetvalue(res, tuple, arg), PQfsize(res, arg));
|
|
}
|
|
param = va_arg(ap, char *);
|
|
if (PQgetisnull(res, tuple, arg) != 0)
|
|
*(int *) param = 1;
|
|
else
|
|
*(int *) param = 0;
|
|
param = va_arg(ap, char *);
|
|
}
|
|
va_end(ap);
|
|
return tuple++;
|
|
}
|
|
|
|
/*
|
|
**
|
|
** on_error_stop
|
|
**
|
|
*/
|
|
void
|
|
on_error_stop()
|
|
{
|
|
on_error_state = ON_ERROR_STOP;
|
|
}
|
|
|
|
/*
|
|
**
|
|
** on_error_continue
|
|
**
|
|
*/
|
|
void
|
|
on_error_continue()
|
|
{
|
|
on_error_state = ON_ERROR_CONTINUE;
|
|
}
|