27.14. 例子程序

Example 27-1. libpq 例子程序 1

/*
 * testlibpq.c
 *
 * 测试 PostgreSQL 前端库 libpq 的 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;



        /*
         * 如果用户在命令行上提供了参数,
         * 那么拿它当作 conninfo 字串;否则缺省设置是 dbname=template1
         * 并且对其它连接使用环境变量或者缺省值。
         * 
         */
        if (argc > 1)
                conninfo = argv[1];
        else
                conninfo = "dbname = template1";


        /* 和数据库建立链接 */
        conn = PQconnectdb(conninfo);


	/*
	* 检查一下与服务器的连接是否成功建立
	*/
	if (PQstatus(conn) != CONNECTION_OK)
	{

                fprintf(stderr, "Connection to database '%s' failed.\n", PQdb(conn));
                fprintf(stderr, "%s", PQerrorMessage(conn));
                exit_nicely(conn);

	}


        /*
         * 我们这里的测试案例涉及使用游标,这种情况下我们必须在一个事务块里面。
         * 我们可以用一个简单的 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);
        }

	/*
	 * 如果不再需要 PGresult 了,我们应该 PQclear,以避免内存泄漏
	 */
	 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;

}

Example 27-2. libpq 例子程序 2

/*
 * testlibpq2.c
 *  测试异步通知接口
 *
 * 运行此程序,然后从另外一个窗口的 psql 里运行 NOTIFY TBL2;
 *
 * 或者,如果你想好玩一点,尝试下面动作∶
 * 用下面的语句填充一个数据库∶
 *
 *   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);
 *
 * 然后做四次∶
 *
 *   INSERT INTO TBL1 values (10);
 *
 */
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h>
#include <sys/time.h>
#include "libpq-fe.h"

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=template1
         * 并且对其它连接使用环境变量或者缺省值。
         * 
         */
        if (argc > 1)
                conninfo = argv[1];
        else
                conninfo = "dbname = template1";


        /* 和数据库建立链接 */
        conn = PQconnectdb(conninfo);


	/*
	* 检查一下与服务器的连接是否成功建立
	*/
	if (PQstatus(conn) != CONNECTION_OK)
	{
                fprintf(stderr, "Connection to database '%s' failed.\n", PQdb(conn));
                fprintf(stderr, "%s", PQerrorMessage(conn));
                exit_nicely(conn);
	}
        /*
         * 发出 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);
        }

        /*
         * 如果不再需要 PGresult 了,我们应该 PQclear,以避免内存泄漏
         */
        PQclear(res);

        /* 收到四次通知之后退出。 */
        nnotifies = 0;
        while (nnotifies < 4)
        {
                /*
                 * 睡眠,直到某些事件发生。我们使用 select(2) 等待输入,Sleep until something happens on the connection.  We use 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++;
                }
        }

        fprintf(stderr, "Done.\n");

        /* 关闭与数据库的连接并清理 */
        PQfinish(conn);

        return 0;

}

Example 27-3. libpq 例子程序 3

/*
 * testlibpq3.c
 *              测试线外参数和二进制I/O。
 *
 * 在运行这个例子之前,用下面的命令填充一个数据库
 * (在 src/test/examples/testlibpq3.sql 里提供):
 *
 * 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
 *
 */
#include <stdio.h>
#include <stdlib.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);
}

int
main(int argc, char **argv)
{
        const char *conninfo;
        PGconn     *conn;
        PGresult   *res;
        const char *paramValues[1];
        int                     i,
                                j;
        int                     i_fnum,
                                t_fnum,
                                b_fnum;

        /*
         * 如果用户在命令行上提供了参数,
         * 那么拿它当作 conninfo 字串;否则缺省设置是 dbname=template1
         * 并且对其它连接使用环境变量或者缺省值。
         *
         */
        if (argc > 1)
                conninfo = argv[1];
        else
                conninfo = "dbname = template1";

        /* 和数据库建立链接 */
        conn = PQconnectdb(conninfo);

        /*
        * 检查一下与服务器的连接是否成功建立
        */
        if (PQstatus(conn) != CONNECTION_OK)
        {
                fprintf(stderr, "Connection to database '%s' failed.\n", PQdb(conn));
                fprintf(stderr, "%s", PQerrorMessage(conn));
                exit_nicely(conn);
        }

        /*
         * 这个程序是用来演示使用线外参数的 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);
        }

        /* 使用 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");
        }

        PQclear(res);

        /* 关闭与数据库的连接并清理 */
        PQfinish(conn);

        return 0;
}