db_bridge/postgresql_db.cpp
author Markus Bröker <mbroeker@largo.dyndns.tv>
Sat, 13 Dec 2008 17:58:09 +0100
changeset 17 b3731a25b9ec
parent 15 5a0ca1f9a2f1
child 18 f3657061ec00
permissions -rw-r--r--
I have improved the connection info and cleaned something up. committer: Markus Bröker <mbroeker@largo.homelinux.org>
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
12
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
     1
/**
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
     2
 * db_bridge/postgresql_db.cpp
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
     3
 * Copyright 2008 (C) Markus Broeker
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
     4
 */
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
     5
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
     6
#include <postgresql_db.h>
15
5a0ca1f9a2f1 db_bridge acts as an interactive sql monitor
Markus Bröker <mbroeker@largo.dyndns.tv>
parents: 14
diff changeset
     7
#include <exception>
12
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
     8
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
     9
PostgreSQL_DB::PostgreSQL_DB (std::string srv, std::string usr, std::string pwd, std::string db)
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    10
{
17
b3731a25b9ec I have improved the connection info and cleaned something up.
Markus Bröker <mbroeker@largo.dyndns.tv>
parents: 15
diff changeset
    11
    server = srv;
b3731a25b9ec I have improved the connection info and cleaned something up.
Markus Bröker <mbroeker@largo.dyndns.tv>
parents: 15
diff changeset
    12
    user = (usr != "") ? usr : getenv ("USER");
b3731a25b9ec I have improved the connection info and cleaned something up.
Markus Bröker <mbroeker@largo.dyndns.tv>
parents: 15
diff changeset
    13
    password = pwd;
b3731a25b9ec I have improved the connection info and cleaned something up.
Markus Bröker <mbroeker@largo.dyndns.tv>
parents: 15
diff changeset
    14
    database = db;
b3731a25b9ec I have improved the connection info and cleaned something up.
Markus Bröker <mbroeker@largo.dyndns.tv>
parents: 15
diff changeset
    15
b3731a25b9ec I have improved the connection info and cleaned something up.
Markus Bröker <mbroeker@largo.dyndns.tv>
parents: 15
diff changeset
    16
    conninfo = "user=" + user + "dbname=" + database;
12
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    17
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    18
    if (server != "" && pwd == "") {
17
b3731a25b9ec I have improved the connection info and cleaned something up.
Markus Bröker <mbroeker@largo.dyndns.tv>
parents: 15
diff changeset
    19
        std::cout << "Passwort von " << user << "@" << server << ": ";
b3731a25b9ec I have improved the connection info and cleaned something up.
Markus Bröker <mbroeker@largo.dyndns.tv>
parents: 15
diff changeset
    20
        std::getline (std::cin, password);
b3731a25b9ec I have improved the connection info and cleaned something up.
Markus Bröker <mbroeker@largo.dyndns.tv>
parents: 15
diff changeset
    21
        conninfo += " host=" + server + "password=" + password;
12
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    22
    }
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    23
}
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    24
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    25
PostgreSQL_DB::~PostgreSQL_DB ()
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    26
{
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    27
    if (conn != NULL) {
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    28
        PQfinish (conn);
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    29
        std::cerr << "Datenbank wird geschlossen" << std::endl;
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    30
    }
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    31
}
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    32
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    33
bool PostgreSQL_DB::connect ()
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    34
{
17
b3731a25b9ec I have improved the connection info and cleaned something up.
Markus Bröker <mbroeker@largo.dyndns.tv>
parents: 15
diff changeset
    35
    std::cerr << "Verbindungsparameter:" << " host=" << ((server != "") ? server : "LOCAL")
b3731a25b9ec I have improved the connection info and cleaned something up.
Markus Bröker <mbroeker@largo.dyndns.tv>
parents: 15
diff changeset
    36
        << " user=" << user << " password=" << password << " database=" << database << std::endl;
b3731a25b9ec I have improved the connection info and cleaned something up.
Markus Bröker <mbroeker@largo.dyndns.tv>
parents: 15
diff changeset
    37
12
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    38
    if ((conn = PQconnectdb (conninfo.c_str ())) == NULL) {
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    39
        std::cerr << server << ": " << "ERROR " << std::endl;
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    40
        return false;
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    41
    }
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    42
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    43
    return true;
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    44
}
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    45
14
862d63715611 the long vector needs a typedef
Markus Bröker <mbroeker@largo.dyndns.tv>
parents: 12
diff changeset
    46
Abstract_DB::DB_RESULT PostgreSQL_DB::query (std::string sql_string)
12
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    47
{
14
862d63715611 the long vector needs a typedef
Markus Bröker <mbroeker@largo.dyndns.tv>
parents: 12
diff changeset
    48
    Abstract_DB::DB_ROW vec;
862d63715611 the long vector needs a typedef
Markus Bröker <mbroeker@largo.dyndns.tv>
parents: 12
diff changeset
    49
    Abstract_DB::DB_RESULT rows;
12
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    50
    PGresult *res;
15
5a0ca1f9a2f1 db_bridge acts as an interactive sql monitor
Markus Bröker <mbroeker@largo.dyndns.tv>
parents: 14
diff changeset
    51
    int i, j;
5a0ca1f9a2f1 db_bridge acts as an interactive sql monitor
Markus Bröker <mbroeker@largo.dyndns.tv>
parents: 14
diff changeset
    52
    int nFields, nRows;
12
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    53
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    54
    /*
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    55
     * send SQL query
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    56
     */
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    57
    res = PQexec (conn, sql_string.c_str ());
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    58
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    59
    nFields = PQnfields (res);
15
5a0ca1f9a2f1 db_bridge acts as an interactive sql monitor
Markus Bröker <mbroeker@largo.dyndns.tv>
parents: 14
diff changeset
    60
    nRows = PQntuples (res);
12
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    61
15
5a0ca1f9a2f1 db_bridge acts as an interactive sql monitor
Markus Bröker <mbroeker@largo.dyndns.tv>
parents: 14
diff changeset
    62
    std::cout << "Die Abfrage hat " << nFields << " Spalte(n) und " << nRows << " Reihe(n)." << std::endl;
12
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    63
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    64
    /*
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    65
     * push everything into a vector< vector<string> >
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    66
     */
15
5a0ca1f9a2f1 db_bridge acts as an interactive sql monitor
Markus Bröker <mbroeker@largo.dyndns.tv>
parents: 14
diff changeset
    67
    for (i = 0; i < nRows; i++) {
12
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    68
        for (j = 0; j < nFields; j++) {
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    69
            vec.push_back (PQgetvalue (res, i, j));
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    70
        }
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    71
        rows.push_back (vec);
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    72
        vec.clear ();
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    73
    }
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    74
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    75
    if (res != NULL) {
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    76
        PQclear (res);
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    77
    }
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    78
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    79
    return rows;
9f0ce4eaa1ce I added a small db_bridge to the repository
Markus Bröker <mbroeker@largo.dyndns.tv>
parents:
diff changeset
    80
}