-
Notifications
You must be signed in to change notification settings - Fork 3
/
SQLDriverPGSQL.C
127 lines (106 loc) · 3.56 KB
/
SQLDriverPGSQL.C
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
/*
* Copyright (c) 2005-2011, Guillaume Gimenez <guillaume@blackmilk.fr>
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are met:
* * Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* * Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
* * Neither the name of G.Gimenez nor the names of its contributors may
* be used to endorse or promote products derived from this software
* without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL G.GIMENEZ BE LIABLE FOR ANY DIRECT,
* INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
* THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*
* Authors:
* * Guillaume Gimenez <guillaume@blackmilk.fr>
*
*/
#include "raii.H"
#include "SQLDriverPGSQL.H"
#include "ResultSetImplPGSQL.H"
namespace raii {
namespace sql {
extern "C" {
bool isPoolable() {
return true;
}
raii::ptr<raii::sql::SQLDriver> getDriver( const raii::String& user, const raii::String& password,
const raii::String& host, int port,
const raii::String& path, const raii::String& query) {
return new raii::sql::SQLDriverPGSQL(host,query,user,password);
}
}
SQLDriverPGSQL::SQLDriverPGSQL(const String& host, const String& dbname, const String& user, const String& passwd)
: SQLDriver(),
autoCommit(true)
{
conn = PQsetdbLogin(host.c_str(),NULL,NULL,NULL,dbname.c_str(),user.c_str(),passwd.c_str());
if ( !conn ) {
throw PostgreSQLException("connexion impossible");
}
if ( PQstatus((PGconn*)conn) != CONNECTION_OK ) {
throw PostgreSQLException(PQerrorMessage((PGconn*)conn));
}
}
SQLDriverPGSQL::~SQLDriverPGSQL() {
if ( !isAutoCommitOn() )
rollback();
if ( conn )
PQfinish((PGconn*)conn);
}
ptr<ResultSetImpl> SQLDriverPGSQL::query(const String& q) {
return new ResultSetImplPGSQL(*this,q);
}
bool SQLDriverPGSQL::isAutoCommitOn() {
return autoCommit;
}
void SQLDriverPGSQL::autoCommitOn() {
if ( !autoCommit ) {
query("ROLLBACK");
autoCommit=true;
}
}
void SQLDriverPGSQL::autoCommitOff() {
if ( autoCommit ) {
query("BEGIN");
autoCommit=false;
}
}
void SQLDriverPGSQL::commit() {
if ( !autoCommit ) {
query("COMMIT");
query("BEGIN");
}
}
void SQLDriverPGSQL::rollback() {
if ( !autoCommit ) {
query("ROLLBACK");
query("BEGIN");
}
}
String SQLDriverPGSQL::sqlize(const String& str) {
StringStream res;
for ( size_t i = 0 ; i < str.size() ; i++ ) {
res << str[i];
if ( str[i] == '\'' )
res << "'";
if ( str[i] == '\\' )
res << "\\";
}
return res.str();
}
}
}