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
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
|
//////////////////////////////////////////////////////////////////////////
//
// pgAgent - PostgreSQL Tools
//
// Copyright (C) 2002 - 2024, The pgAdmin Development Team
// This software is released under the PostgreSQL Licence
//
// connection.h - database connection
//
//////////////////////////////////////////////////////////////////////////
#ifndef CONNECTION_H
#define CONNECTION_H
#include <libpq-fe.h>
class DBresult;
class CONNinfo
{
public:
bool Set(const std::string& connStr);
const std::string Get(const std::string &dbName="") const;
const std::string& GetError() const { return m_error; }
static const std::string Parse(
const std::string& connStr, std::string *error,
std::string *dbName, bool forLogging=false
);
operator bool() const { return m_connStr.empty(); }
private:
std::string m_connStr;
std::string m_dbName;
std::string m_error;
};
class DBconn
{
protected:
DBconn(const std::string& connStr);
~DBconn();
public:
static DBconn *Get(const std::string &connStr="", const std::string &db="");
static DBconn *InitConnection(const std::string &connectString);
static void ClearConnections(bool allIncludingPrimary = false);
std::string qtDbString(const std::string &value);
bool BackendMinimumVersion(int major, int minor);
std::string GetLastError();
operator bool() const { return m_conn != NULL; }
DBresult *Execute(const std::string &query);
std::string ExecuteScalar(const std::string &query);
int ExecuteVoid(const std::string &query);
void Return();
const std::string &DebugConnectionStr() const;
bool LastCommandOk()
{
return IsCommandOk((ExecStatusType)m_lastResult);
}
bool IsCommandOk(ExecStatusType ret)
{
switch (ret)
{
case PGRES_COMMAND_OK:
case PGRES_TUPLES_OK:
case PGRES_COPY_OUT:
case PGRES_COPY_IN:
#if (PG_VERSION_NUM >= 90100)
case PGRES_COPY_BOTH:
#endif
return true;
default:
return false;
};
}
void SetLastResult(int res)
{
m_lastResult = res;
}
int GetLastResult()
{
return m_lastResult;
}
private:
bool Connect(const std::string &connectString);
int m_minorVersion,
m_majorVersion;
protected:
static CONNinfo ms_basicConnInfo;
static DBconn *ms_primaryConn;
std::string m_lastError;
std::string m_connStr;
PGconn *m_conn;
DBconn *m_next;
DBconn *m_prev;
bool m_remoteDatabase;
bool m_inUse;
int m_lastResult;
friend class DBresult;
};
class DBresult
{
protected:
DBresult(DBconn *conn, const std::string &query);
public:
~DBresult();
std::string GetString(int col) const;
std::string GetString(const std::string &colname) const;
bool IsValid() const
{
return m_result != NULL;
}
bool HasData() const
{
return m_currentRow < m_maxRows;
}
void MoveNext()
{
if (m_currentRow < m_maxRows) m_currentRow++;
}
long RowsAffected() const
{
return atol(PQcmdTuples(m_result));
}
protected:
PGresult *m_result;
int m_currentRow, m_maxRows;
friend class DBconn;
};
class DBresultPtr
{
public:
DBresultPtr(DBresult* in_ptr)
: m_ptr(in_ptr)
{}
~DBresultPtr()
{
if (m_ptr) {
delete m_ptr;
m_ptr = NULL;
}
}
DBresultPtr& operator=(DBresult *other)
{
if (m_ptr) {
delete m_ptr;
}
m_ptr = other;
return *this;
}
const DBresult& operator*() const
{
return (*(const DBresult *)m_ptr);
}
const DBresult* operator->() const
{
return (const DBresult*)(m_ptr);
}
DBresult& operator*()
{
return (*(DBresult *)m_ptr);
}
DBresult* operator->()
{
return (DBresult *)m_ptr;
}
operator void*() const
{
return (DBresult *)m_ptr;
}
operator bool() const { return (m_ptr != NULL); }
protected:
DBresult* m_ptr;
};
#endif // CONNECTION_H
|
