PostgreSQL 8.1 中文文晻紭PostgreSQL 中国制作挤
Prev	Fast Backward	Chapter 28. libpq - C 	Fast Forward	Next

28.16. 例子刻序

Example 28-1. libpq 例子刻序 1

/*
 * testlibpq.c
 *
 *      Test C 勬寙暷 libpq几PostgreSQL 莿剿
 */
#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=postgres 镜且使用倷喛屽联傭者所有棋它粮樣疚数
     * 綒使用葘蕠謺嚰
     */
    if (argc > 1)
        conninfo = argv[1];
    else
        conninfo = "dbname = postgres";

    /* Make a connection to the database */
    conn = PQconnectdb(conninfo);

    /* 氺鹃后剿粮樣可攻槝恋 */
    if (PQstatus(conn) != CONNECTION_OK)
    {
        fprintf(stderr, "Connection to database failed: %s",
                PQerrorMessage(conn));
        exit_nicely(conn);
    }

    /*
     * 我们暷锯试蕰例涉殑游岅暷使用几这蕦候我们屫须使用事务
     * 我们以勓葐惊事情櫯在覀  "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 PGresult几以屲免内樻泄路
     */
    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 28-2. libpq 例子刻序 2

/*
 * testlibpq2.c
 *  锯试异緲蜆只樣
 *
 * 运行標刻序几葌后樣另外覀槃暷 psql 里运行
 *  NOTIFY TBL2;
 * 重四樜几謱晿刻序退况
 *
 * 傭者几蠕果你想好玩覀曘几用下面命令填垮数嗇嚸
 * 紭在 src/test/examples/testlibpq2.sql 里提狗挤
 *
 *   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"

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=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);
    }

    /*
     * 櫟况 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) 暼橗叔入几
         * 晢是揖以用 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 28-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
 *
 * tuple 0: got
 *  i = (4 bytes) 2
 *  t = (8 bytes) 'ho there'
 *  b = (5 bytes) \004\003\002\001\000
 */
#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);
}

/*
 * 这簢数橋訃鹃褔樶果几这蟹樶果是浸橒制蕵几樣上面暷
 * 椎释里面槝槝暷岉中讝葒况罉暷嚰我们勓这簢数晼嚼攫况罉
 * 是因位 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;

    /*
     * 蠕果用倱在命令行上提狗了疚数几
     * 那脴拿它晫作 conninfo 字槥紓欛则葘蕠设置是 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);
    }

    /*
     * 这刻序是用罉演蕟使用线外疚数暷 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;
}

Prev	Home	Next
制作 libpq 刻序	Up	橌皆象