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
|
/*
* FALCON - The Falcon Programming Language.
* FILE: pgsql_mod.h
*
* Pgsql driver main module interface
* -------------------------------------------------------------------
* Author: Jeremy Cowgar, Stanislas Marquis
* Begin: Sun Dec 23 21:36:20 2007
*
* -------------------------------------------------------------------
* (C) Copyright 2007: the FALCON developers (see list in AUTHORS file)
*
* See LICENSE file for licensing details.
*/
#ifndef PGSQL_MOD_H
#define PGSQL_MOD_H
#include <libpq-fe.h>
#include <falcon/dbi_common.h>
#include <falcon/srv/dbi_service.h>
namespace Falcon
{
class DBIHandlePgSQL;
int32 dbi_pgsqlQuestionMarksToDollars( const String& input, String& output );
class PgSQLHandlerRef: public DBIRefCounter<PGconn*>
{
public:
PgSQLHandlerRef( PGconn* conn ):
DBIRefCounter<PGconn*>( conn )
{}
~PgSQLHandlerRef() {
PQfinish( handle() );
}
};
class DBIRecordsetPgSQL
:
public DBIRecordset
{
protected:
int64 m_row;
int64 m_rowCount;
int m_columnCount;
PGresult* m_res;
PgSQLHandlerRef* m_pConn;
public:
DBIRecordsetPgSQL( DBIHandlePgSQL* dbh, PGresult* res );
virtual ~DBIRecordsetPgSQL();
virtual bool fetchRow();
virtual int64 getRowIndex();
virtual int64 getRowCount();
virtual int getColumnCount();
virtual bool getColumnName( int nCol, String& name );
virtual bool getColumnValue( int nCol, Item& value );
virtual bool discard( int64 ncount );
virtual void close();
};
class DBIStatementPgSQL
:
public DBIStatement
{
protected:
uint32 m_nParams;
String m_execString;
String m_name;
PgSQLHandlerRef* m_pConn;
void getExecString( uint32 nParams, const String& name );
public:
DBIStatementPgSQL( DBIHandlePgSQL* dbh );
virtual ~DBIStatementPgSQL();
virtual DBIRecordset* execute( ItemArray* params = 0 );
virtual void reset();
virtual void close();
void init( const String& query, const String& name="happy_falcon" );
};
class DBIHandlePgSQL
:
public DBIHandle
{
protected:
PGconn* m_conn;
bool m_bInTrans;
DBISettingParams m_settings;
PgSQLHandlerRef* m_pConn;
public:
DBIHandlePgSQL( PGconn* = 0 );
virtual ~DBIHandlePgSQL();
PGconn* getConn() { return m_conn; }
virtual void options( const String& params );
virtual const DBISettingParams* options() const;
virtual void begin();
virtual void commit();
virtual void rollback();
virtual DBIRecordset* query( const String &sql, ItemArray* params = 0 );
virtual DBIStatement* prepare( const String &query );
virtual DBIStatement* prepareNamed( const String &name, const String& query );
virtual int64 getLastInsertedId( const String& name = "" );
virtual void selectLimited( const String& query, int64 nBegin, int64 nCount, String& result );
virtual void close();
static void throwError( const char* file, int line, PGresult* res );
PGresult* internal_exec( const String& sql, int64& affectedRows );
PgSQLHandlerRef* getConnRef() const { return m_pConn; }
};
class DBIServicePgSQL
:
public DBIService
{
public:
DBIServicePgSQL()
:
DBIService( "DBI_pgsql" )
{}
virtual void init();
virtual DBIHandle* connect( const String& parameters );
virtual CoreObject* makeInstance( VMachine* vm, DBIHandle* dbh );
};
} // !namespace Falcon
extern Falcon::DBIServicePgSQL thePgSQLService;
#endif /* PGSQL_MOD_H */
|
