Примеры программ
Пример 1. Первый образец программы, использующей libpq
/* Проверка C-версии libpq, клиентской библиотеки QHB. */
#include <stdio.h>
#include <stdlib.h>
#include "libpq-fe.h"
static void
exit_nicely(PGconn *conn)
{
PQfinish(conn);
exit(1);
}
int
main(int argc, char **argv)
{
const char *conninfo;
PGconn *conn;
PGresult *res;
int nFields;
int i,
j;
/*
* Если пользователь передает параметр в командной строке, использовать ее как
* строку информации о подключении (conninfo); в противном случае по умолчанию
* установить dbname=qhb и использовать для всех остальных параметров подключения
* переменные среды или значения по умолчанию.
*/
if (argc > 1)
conninfo = argv[1];
else
conninfo = "dbname = qhb";
/* Установить соединение с базой данных */
conn = PQconnectdb(conninfo);
/* Проверить, что соединение с сервером было успешно установлено */
if (PQstatus(conn) != CONNECTION_OK)
{
fprintf(stderr, "%s", PQerrorMessage(conn));
exit_nicely(conn);
}
/*
* Установить постоянно защищенный путь поиска, чтобы неблагонамеренные
* пользователи не могли перехватить управление.
*/
res = PQexec(conn,
"SELECT pg_catalog.set_config('search_path', '', false)");
if (PQresultStatus(res) != PGRES_TUPLES_OK)
{
fprintf(stderr, "SET failed: %s", PQerrorMessage(conn));
PQclear(res);
exit_nicely(conn);
}
/*
* Когда объект PGresult больше не нужен, функция PQclear обязательно должна
* его очистить во избежание утечек памяти
*/
PQclear(res);
/*
* В нашем тестовом случае используется курсор, и для этого нам нужно находиться
* внутри блока транзакции. Мы могли бы обработать его целиком с помощью одной
* функции PQexec() в команде «select * from pg_database», но это было бы
* слишком просто для хорошего примера.
*/
/* Начать блок транзакции */
res = PQexec(conn, "BEGIN");
if (PQresultStatus(res) != PGRES_COMMAND_OK)
{
fprintf(stderr, "BEGIN command failed: %s", PQerrorMessage(conn));
PQclear(res);
exit_nicely(conn);
}
PQclear(res);
/*
* Извлечь строки из pg_database, системного каталога баз данных
*/
res = PQexec(conn, "DECLARE myportal CURSOR FOR select * from pg_database");
if (PQresultStatus(res) != PGRES_COMMAND_OK)
{
fprintf(stderr, "DECLARE CURSOR failed: %s", PQerrorMessage(conn));
PQclear(res);
exit_nicely(conn);
}
PQclear(res);
res = PQexec(conn, "FETCH ALL in myportal");
if (PQresultStatus(res) != PGRES_TUPLES_OK)
{
fprintf(stderr, "FETCH ALL failed: %s", PQerrorMessage(conn));
PQclear(res);
exit_nicely(conn);
}
/* сначала вывести имена атрибутов */
nFields = PQnfields(res);
for (i = 0; i < nFields; i++)
printf("%-15s", PQfname(res, i));
printf("\n\n");
/* затем вывести строки */
for (i = 0; i < PQntuples(res); i++)
{
for (j = 0; j < nFields; j++)
printf("%-15s", PQgetvalue(res, i, j));
printf("\n");
}
PQclear(res);
/* закрыть портал ... мы не утруждаем себя проверкой на ошибки ... */
res = PQexec(conn, "CLOSE myportal");
PQclear(res);
/* завершить транзакцию */
res = PQexec(conn, "END");
PQclear(res);
/* закрыть соединение с базой данных и провести очистку */
PQfinish(conn);
return 0;
}
Пример 2. Второй образец программы, использующей libpq
/*
* Проверка интерфейса асинхронного уведомления
*
* Запустить эту программу, затем из psql в отдельном окне выполнить
* NOTIFY TBL2;
* Повторить четыре раза, чтобы эта программа завершилась.
*
* Или, если вы хотите проявить изобретательность, попробуйте это:
* заполнить базу данных с помощью следующих команд:
*
* CREATE SCHEMA TESTLIBPQ2;
* SET search_path = TESTLIBPQ2;
* CREATE TABLE TBL1 (i int4);
* CREATE TABLE TBL2 (i int4);
* CREATE RULE r1 AS ON INSERT TO TBL1 DO
* (INSERT INTO TBL2 VALUES (new.i); NOTIFY TBL2);
*
* Запустить эту программу, затем из psql четыре раза выполнить следующее:
*
* INSERT INTO TESTLIBPQ2.TBL1 VALUES (10);
*/
#ifdef WIN32
#include <windows.h>
#endif
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h>
#include <sys/time.h>
#include <sys/types.h>
#ifdef HAVE_SYS_SELECT_H
#include <sys/select.h>
#endif
#include "libpq-fe.h"
static void
exit_nicely(PGconn *conn)
{
PQfinish(conn);
exit(1);
}
int
main(int argc, char **argv)
{
const char *conninfo;
PGconn *conn;
PGresult *res;
PGnotify *notify;
int nnotifies;
/*
* Если пользователь передает параметр в командной строке, использовать ее как
* строку информации о подключении (conninfo); в противном случае по умолчанию
* установить dbname=qhb и использовать для всех остальных параметров подключения
* переменные среды или значения по умолчанию.
*/
if (argc > 1)
conninfo = argv[1];
else
conninfo = "dbname = qhb";
/* Установить соединение с базой данных */
conn = PQconnectdb(conninfo);
/* Проверить, что соединение с сервером было успешно установлено */
if (PQstatus(conn) != CONNECTION_OK)
{
fprintf(stderr, "%s", PQerrorMessage(conn));
exit_nicely(conn);
}
/*
* Установить постоянно защищенный путь поиска, чтобы неблагонамеренные
* пользователи не могли перехватить управление.
*/
res = PQexec(conn,
"SELECT pg_catalog.set_config('search_path', '', false)");
if (PQresultStatus(res) != PGRES_TUPLES_OK)
{
fprintf(stderr, "SET failed: %s", PQerrorMessage(conn));
PQclear(res);
exit_nicely(conn);
}
/*
* Когда объект PGresult больше не нужен, функция PQclear обязательно должна
* его очистить во избежание утечек памяти
*/
PQclear(res);
/*
* Выполнить команду LISTEN, чтобы включить уведомления из NOTIFY правила.
*/
res = PQexec(conn, "LISTEN TBL2");
if (PQresultStatus(res) != PGRES_COMMAND_OK)
{
fprintf(stderr, "LISTEN command failed: %s", PQerrorMessage(conn));
PQclear(res);
exit_nicely(conn);
}
PQclear(res);
/* Выйти после получения четырех уведомлений. */
nnotifies = 0;
while (nnotifies < 4)
{
/*
* Находиться в режиме ожидания, пока в соединении что-нибудь не произойдет.
* Мы используем для ожидания ввода select(2), но также можно использовать
* poll() или подобные ей средства.
*/
int sock;
fd_set input_mask;
sock = PQsocket(conn);
if (sock < 0)
break; /* не должно происходить */
FD_ZERO(&input_mask);
FD_SET(sock, &input_mask);
if (select(sock + 1, &input_mask, NULL, NULL, NULL) < 0)
{
fprintf(stderr, "select() failed: %s\n", strerror(errno));
exit_nicely(conn);
}
/* Теперь проверить, есть ли входные данные */
PQconsumeInput(conn);
while ((notify = PQnotifies(conn)) != NULL)
{
fprintf(stderr,
"ASYNC NOTIFY of '%s' received from backend PID %d\n",
notify->relname, notify->be_pid);
PQfreemem(notify);
nnotifies++;
PQconsumeInput(conn);
}
}
fprintf(stderr, "Done.\n");
/* закрыть соединение с базой данных и провести очистку */
PQfinish(conn);
return 0;
}
Пример 3. Третий образец программы, использующей libpq
/*
* Проверка вынесенных параметров и двоичного ввода/вывода.
*
* Прежде чем выполнить это, заполните базу данных с помощью следующих команд:
*
* CREATE SCHEMA testlibpq3;
* SET search_path = testlibpq3;
* SET standard_conforming_strings = ON;
* CREATE TABLE test1 (i int4, t text, b bytea);
* INSERT INTO test1 values (1, 'joe''s place', '\000\001\002\003\004');
* INSERT INTO test1 values (2, 'ho there', '\004\003\002\001\000');
*
* Ожидаемый вывод:
*
* tuple 0: got
* i = (4 bytes) 1
* t = (11 bytes) 'joe's place'
* b = (5 bytes) \000\001\002\003\004
*
* tuple 0: got
* i = (4 bytes) 2
* t = (8 bytes) 'ho there'
* b = (5 bytes) \004\003\002\001\000
*/
#ifdef WIN32
#include <windows.h>
#endif
#include <stdio.h>
#include <stdlib.h>
#include <stdint.h>
#include <string.h>
#include <sys/types.h>
#include "libpq-fe.h"
/* для ntohl/htonl */
#include <netinet/in.h>
#include <arpa/inet.h>
static void
exit_nicely(PGconn *conn)
{
PQfinish(conn);
exit(1);
}
/*
* Эта функция выводит результат запроса в двоичном формате, извлеченный из
* таблицы, определенной как описано в комментарии выше. Мы разделяем его, потому
* что функция main() использует ее дважды.
*/
static void
show_binary_results(PGresult *res)
{
int i,
j;
int i_fnum,
t_fnum,
b_fnum;
/* Использовать PQfnumber во избежание предположений о порядке полей в результате */
i_fnum = PQfnumber(res, "i");
t_fnum = PQfnumber(res, "t");
b_fnum = PQfnumber(res, "b");
for (i = 0; i < PQntuples(res); i++)
{
char *iptr;
char *tptr;
char *bptr;
int blen;
int ival;
/* Получить поле значений (мы игнорируем возможность, что они равны NULL!) */
iptr = PQgetvalue(res, i, i_fnum);
tptr = PQgetvalue(res, i, t_fnum);
bptr = PQgetvalue(res, i, b_fnum);
/*
* Двоичное представление INT4 находится в сетевом порядке байтов, который
* лучше привести к локальному порядку байтов.
*/
ival = ntohl(*((uint32_t *) iptr));
/*
* Двоичное представление TEXT является, собственно, текстом, и поскольку
* libpq любезно добавила к нему нулевой байт, он прекрасно сработает
* как строка на C.
*
* Двоичное представление BYTEA является кучкой байтов, которые могут включать
* в себя значения NULL, поэтому нам придется обращать внимание на длину поля.
*/
blen = PQgetlength(res, i, b_fnum);
printf("tuple %d: got\n", i);
printf(" i = (%d bytes) %d\n",
PQgetlength(res, i, i_fnum), ival);
printf(" t = (%d bytes) '%s'\n",
PQgetlength(res, i, t_fnum), tptr);
printf(" b = (%d bytes) ", blen);
for (j = 0; j < blen; j++)
printf("\\%03o", bptr[j]);
printf("\n\n");
}
}
int
main(int argc, char **argv)
{
const char *conninfo;
PGconn *conn;
PGresult *res;
const char *paramValues[1];
int paramLengths[1];
int paramFormats[1];
uint32_t binaryIntVal;
/*
* Если пользователь передает параметр в командной строке, использовать ее как
* строку информации о подключении (conninfo); в противном случае по умолчанию
* установить dbname=qhb и использовать для всех остальных параметров подключения
* переменные среды или значения по умолчанию
*/
if (argc > 1)
conninfo = argv[1];
else
conninfo = "dbname = qhb";
/* Установить соединение с базой данных */
conn = PQconnectdb(conninfo);
/* Проверить, что соединение с сервером было успешно установлено */
if (PQstatus(conn) != CONNECTION_OK)
{
fprintf(stderr, "%s", PQerrorMessage(conn));
exit_nicely(conn);
}
/*
* Установить постоянно защищенный путь поиска, чтобы неблагонамеренные
* пользователи не могли перехватить управление.
*/
res = PQexec(conn, "SET search_path = testlibpq3");
if (PQresultStatus(res) != PGRES_COMMAND_OK)
{
fprintf(stderr, "SET failed: %s", PQerrorMessage(conn));
PQclear(res);
exit_nicely(conn);
}
PQclear(res);
/*
* Цель этой программы состоит в том, чтобы проиллюстрировать использование функции
* PQexecParams() с внешними параметрами, а также двоичную передачу данных.
*
* В этом первом примере параметры передаются в виде текста, но результаты
* возвращаются в двоичном формате. Благодаря внешним параметрам мы можем
* избежать трудоемкой возни с заключением в кавычки и экранированием, несмотря
* на то, что данные являются текстом. Обратите внимание, что нам не нужно
* делать ничего особенного со знаком кавычки в значении параметра.
*/
/* Вот наше значение внешнего параметра */
paramValues[0] = "joe's place";
res = PQexecParams(conn,
"SELECT * FROM test1 WHERE t = $1",
1, /* один параметр */
NULL, /* позволить серверу определить тип параметра */
paramValues,
NULL, /* длина параметра не нужна, поскольку это текст */
NULL, /* применить по умолчанию ко все текстовым параметрам */
1); /* попросить двоичные результаты */
if (PQresultStatus(res) != PGRES_TUPLES_OK)
{
fprintf(stderr, "SELECT failed: %s", PQerrorMessage(conn));
PQclear(res);
exit_nicely(conn);
}
show_binary_results(res);
PQclear(res);
/*
* Во втором примере мы передаем целочисленный параметр в двоичной форме
* и снова получаем результаты в двоичной форме.
*
* Хотя мы и говорим PQexecParams, что позволяем серверу определить тип
* параметра, в действительности мы вынуждаем его принять решение, преобразуя
* символ параметра в тексте запроса. Это надежная мера безопасности при
* передаче двоичных параметров.
*/
/* Преобразовать целочисленное значение «2» в сетевой порядок байтов */
binaryIntVal = htonl((uint32_t) 2);
/* Установить значения массивов параметров для PQexecParams */
paramValues[0] = (char *) &binaryIntVal;
paramLengths[0] = sizeof(binaryIntVal);
paramFormats[0] = 1; /* двоичный */
res = PQexecParams(conn,
"SELECT * FROM test1 WHERE i = $1::int4",
1, /* один параметр */
NULL, /* позволить серверу определить тип параметра */
paramValues,
paramLengths,
paramFormats,
1); /* попросить двоичные результаты */
if (PQresultStatus(res) != PGRES_TUPLES_OK)
{
fprintf(stderr, "SELECT failed: %s", PQerrorMessage(conn));
PQclear(res);
exit_nicely(conn);
}
show_binary_results(res);
PQclear(res);
/* закрыть соединение с базой данных и провести очистку */
PQfinish(conn);
return 0;
}