发个C语言连接Postgresql程序(转)

lizh

　　一直用的是Oracle，今天要在postgresql写个程序，查了半天的pg官方文档件才搞定。
官方的例子有点简单，我把项目中的程序直接发过来供后来者学习和参考。
备注：
1，这个程序功能是使2台服务器的postgresql数据库中的一个表保持一致。
2，加了进程重启功能。
3，postgresql有自动提交回滚功能，害的我在官方api文档找了半天的事务回滚在哪里（高手莫笑，哈哈）
4，程序最好总是在BEGIN和END之间操作数据库，哪怕仅仅是select操作。
5，若遇乱码问题，请在sql语句前加set client_encoding = 'GBK';
/*
*Program Name : updateplicheck.c
*Author : Liu Lin
*Date : 20090701
*/
#include "public.h"
#include "db_manager.h"
#include "libpq-fe.h"
#include <stdio.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <fcntl.h>
static void sig_child(int signo);
static void task_fork();
static void run_task();
int main(void)
{
    int pid;                    //创建守护进程开始
   
    if ((pid = fork()) < 0)
    {
        LOGLINE; LOG("创建进程操作失败");
        exit(1);
    }
   
    if (pid > 0)
        exit(0);
        
    setsid();
    signal(SIGHUP, SIG_IGN);   
    umask(0);                  //创建守护进程结束
   
    signal(SIGCHLD, sig_child);
      
    task_fork();
   
    while (1)
    {
        pause();
    }  
   
}
static void task_fork()
{
    pid_t pid;
   
    if ((pid = fork()) < 0)
    {
        LOGLINE, LOG("创建子进程失败");
        exit(1);
    }
   
    if (pid == 0)  
    {
        signal(SIGCHLD, SIG_DFL);
        run_task();
        exit(0);
    }
}
static void run_task(void)
{
    char conninfo[256] = {0};  
    PGconn *conn;  //连接代理服务器数据库
    PGresult *res;
    PGconn *conn2; //连接本地服务器数据库
    PGresult *res2;
    int status;
    char sql_que[1024] = {0};
    int i;   
    int update_times = 0;
   
    while (update_times < 12*7)//每个进程只负责一周的更新，防止内存泄露
    {
        update_times++;
        memset(conninfo, 0, sizeof(conninfo));
        strcpy(conninfo, "host=192.168.1.110 dbname=ydds user=postgres password=postgres port=5432 connect_timeout=30");  
        conn = PQconnectdb(conninfo);  
        
        if(PQstatus(conn) != CONNECTION_OK)  
        {   
            LOGLINE; LOG("与更新源postgresql数据库连接失败，时间是[%s]", TIME());
            PQfinish(conn);
            sleep(60*60);
            continue;
        }
        else
        {
            LOGLINE; LOG("与更新源postgresql数据库连接成功，时间是[%s]", TIME());
        }  
        
        memset(conninfo, 0, sizeof(conninfo));
        strcpy(conninfo, "host=localhost dbname=ydds user=postgres password=postgres port=5432 connect_timeout=30");  
        conn2 = PQconnectdb(conninfo);  
        
        if(PQstatus(conn2) != CONNECTION_OK)  
        {
            LOGLINE; LOG("与本地postgresql数据库连接失败，时间是[%s]", TIME());
            PQfinish(conn);
            PQfinish(conn2);
            sleep(60*60);
            continue;
        }
        else
        {
            LOGLINE; LOG("与本地postgresql数据库连接成功，时间是[%s]", TIME());
        }  
        
        res2 = PQexec(conn2, "BEGIN");
        
        if(PQresultStatus(res2) != PGRES_COMMAND_OK) //成功完成一个不返回数据的命令
        {
            LOGLINE; LOG("执行BEGIN失败[%s]", PQerrorMessage(conn));
            PQclear(res2);
            PQfinish(conn);
            PQfinish(conn2);
            sleep(60*60);
            continue;
        }
        
        PQclear(res2);
        
        memset(sql_que, 0, sizeof(sql_que));
        sprintf(sql_que, "select adm_no, adm_name, pli_no, pli_name from bank_pli_check");
        res = PQexec(conn, sql_que);
        
        status = PQresultStatus(res);
        
        if (status != PGRES_TUPLES_OK) //成功执行一个返回数据的查询查询
        {
            LOGLINE; LOG("查询表bank_pli_check失败");
            PQclear(res);
            PQfinish(conn);
            PQfinish(conn2);
            sleep(60*60);
            continue;
        }
        
        for (i = 0; i < PQntuples(res); i++)
        {
            memset(sql_que, 0, sizeof(sql_que));
            sprintf(sql_que, "select pli_no from bank_pli_check where pli_no = '%s'", PQgetvalue(res, i, 2));
            res2 = PQexec(conn2, sql_que);//查看该条记录是否已经存在
            status = PQresultStatus(res2);
            
            if (status != PGRES_TUPLES_OK)//注意一个碰巧检索了零条元组的SELECT仍然显示PGRES_TUPLES_OK
            {
                LOGLINE; LOG("查询表bank_pli_check失败");
                PQclear(res);
                PQclear(res2);
                PQfinish(conn);
                PQfinish(conn2);
                sleep(60*60);
                continue;
            }
            
            if (PQntuples(res2) != 0)
            {
                LOGLINE; LOG("记录pli=[%s]已经在bank_pli_check存在", PQgetvalue(res, i, 2));
                PQclear(res2);
            }
            else
            {
                PQclear(res2);
                memset(sql_que, 0, sizeof(sql_que));
                sprintf(sql_que, "insert into bank_pli_check (adm_no, adm_name, pli_no, pli_name) values ('%s', '%s', '%s', '%s')", PQgetvalue(res, i, 0), PQgetvalue(res, i, 1), PQgetvalue(res, i, 2), PQgetvalue(res, i, 3));
                res2 = PQexec(conn2, sql_que);//没有的记录将更新到表里面
                status = PQresultStatus(res2);
               
                if (status == PGRES_COMMAND_OK)
                {
                    LOGLINE; LOG("记录pli=[%s]已经更新到bank_pli_check表里", PQgetvalue(res, i, 2));
                }
                else
                {
                    LOGLINE; LOG("记录pli=[%s]插入bank_pli_check表失败", PQgetvalue(res, i, 2));
                }
                    
            }
            
        }
        
        res2 = PQexec(conn2, "END");
        PQclear(res);
        PQclear(res2);
        PQfinish(conn);
        PQfinish(conn2);
        sleep(60*60);
    }
   
    return;
}
static void sig_child(int signo)
{
        pid_t pid;
        int status;
        
        while ((pid = waitpid(-1, &status, WNOHANG)) > 0)
        {
        LOGLINE; LOG("子进程[%d]]退出， 时间是[%s]", pid, TIME());
        task_fork();
    }
   
    return;
}
static int LOG_INIT(void)
{
    time_t timep;
    struct tm *time_ptr = NULL;
   
    time(&timep);
    time_ptr = localtime(&timep);
   
    memset(g_log_file, 0, sizeof(g_log_file));
    sprintf(g_log_file, "%s%s.log.%d%02d%02d", LOG_PATH, __FILE__, (1900 + time_ptr->tm_year), (1 + time_ptr->tm_mon), (time_ptr->tm_mday));
    memset(g_source_file, 0, sizeof(g_source_file));
    strcpy(g_source_file, __FILE__);
   
    return 0;
}
　　http://bbs2.iyunv.com/thread-1497010-1-1.html