/*
 * src/test/examples/testlibpq.c
 *
 *
 * testlibpq.c
 *
 * 测试 libpq(PostgreSQL 前端库) 的 C 版本。
 */
#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;
 /*
 * 如果用户在命令行上提供了一个参数,将它用作连接信息串。
 * 否则默认用设置 dbname=postgres 并且为所有其他链接参数使用环境变量或默认值。
 */
 if (argc > 1)
 conninfo = argv[1];
 else
 conninfo = "dbname = postgres";
 /* 建立到数据库的一个连接 */
 conn = PQconnectdb(conninfo);
 /* 检查看后端连接是否成功建立 */
 if (PQstatus(conn) != CONNECTION_OK)
 {
 fprintf(stderr, "Connection to database failed: %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);
 /*
 * 我们的测试案例这里涉及使用一个游标,对它我们必须用在一个事务块内。
 * 我们可以在一个单一的 "select * from pg_database" 的 PQexec() 中做整个事情,
 * 但是作为一个好的例子它太琐碎。
 */
 /* 开始一个事务块 */
 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;
}

/*
 * src/test/examples/testlibpq2.c
 *
 *
 * testlibpq2.c
 * 测试异步通知接口
 *
 * 开始这个程序,然后在另一个窗口的 psql 中做
 * NOTIFY TBL2;
 * 重复四次来让这个程序退出。
 *
 * 或者,如果你想要得到奇妙的事情,尝试:
 * 用下列命令填充一个数据库
 * (在 src/test/examples/testlibpq2.sql 中提供)
 *
 * 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>
#include "libpq-fe.h"
#ifdef HAVE_SYS_SELECT_H
#include <sys/select.h>
#endif
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;
 /*
 * 用过用户在命令行上提供了一个参数,将它用作连接信息串。
 * 否则默认用设置 dbname=postgres 并且为所有其他链接参数使用环境变量或默认值。
 */
 if (argc > 1)
 conninfo = argv[1];
 else
 conninfo = "dbname = postgres";
 /* 建立一个到数据库的连接 */
 conn = PQconnectdb(conninfo);
 /* 检查后端连接是否成功建立 */
 if (PQstatus(conn) != CONNECTION_OK)
 {
 fprintf(stderr, "Connection to database failed: %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;
}

/*
 * src/test/examples/testlibpq3.c
 *
 *
 * testlibpq3.c
 * 测试线外参数和二进制 I/O。
 *
 * 在运行之前,使用下列命令填充一个数据库(在 src/test/examples/testlibpq3.sql 中提供)
 *
 * CREATE SCHEMA testlibpq3;
 * SET search_path = testlibpq3;
 * 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"
/* for 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;
 /* 得到域值(我们忽略它们为空值的可能性!) */
 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 的二进制表示是一堆字节,其中可能包含嵌入的空值,因此我们必须注意域长度。
 */
 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;
 /*
 * 如果用户在命令行上提供了一个参数,将它用作连接信息串。
 * 否则默认用设置 dbname=postgres 并且为所有其他链接参数使用环境变量或默认值。
 */
 if (argc > 1)
 conninfo = argv[1];
 else
 conninfo = "dbname = postgres";
 /* 建立一个到数据库的连接 */
 conn = PQconnectdb(conninfo);
 /* 检查看后端连接是否成功被建立 */
 if (PQstatus(conn) != CONNECTION_OK)
 {
 fprintf(stderr, "Connection to database failed: %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; /* binary */
 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;
}