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
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
|
<!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.0 Transitional//EN">
<html><head><meta http-equiv="Content-Type" content="text/html;charset=iso-8859-1">
<title>QuickFIX: /home/omiller/Development/autobuild/quickfix/src/C++/OdbcConnection.h Source File</title>
<link href="quickfix.css" rel="stylesheet" type="text/css">
<table cellspacing="0" cellpadding="0" border="0">
<tr>
<td>
<img src="images/QuickFIX.jpg" align="middle" border=0>
<a href="index.html">Index</a>
<a href="files.html">Source Files</a>
<a href="annotated.html">Annotated Class List</a>
<a href="classes.html">Alphabetical Class List</a>
<a href="hierarchy.html">Class Hierarchy</a>
<a href="inherits.html">Graphical Class Hierarchy</a>
</td>
</tr>
</table>
</head><body bgcolor="#ffffff">
<!-- Generated by Doxygen 1.6.1 -->
<div class="navpath"><a class="el" href="dir_a9003438d6daff57482a26b22b6121a1.html">src</a> » <a class="el" href="dir_348ef16cf184025134029522e7f02883.html">C++</a>
</div>
<div class="contents">
<h1>OdbcConnection.h</h1><a href="_odbc_connection_8h.html">Go to the documentation of this file.</a><div class="fragment"><pre class="fragment"><a name="l00001"></a>00001 <span class="comment">/* -*- C++ -*- */</span>
<a name="l00002"></a>00002
<a name="l00003"></a>00003 <span class="comment">/****************************************************************************</span>
<a name="l00004"></a>00004 <span class="comment">** Copyright (c) quickfixengine.org All rights reserved.</span>
<a name="l00005"></a>00005 <span class="comment">**</span>
<a name="l00006"></a>00006 <span class="comment">** This file is part of the QuickFIX FIX Engine</span>
<a name="l00007"></a>00007 <span class="comment">**</span>
<a name="l00008"></a>00008 <span class="comment">** This file may be distributed under the terms of the quickfixengine.org</span>
<a name="l00009"></a>00009 <span class="comment">** license as defined by quickfixengine.org and appearing in the file</span>
<a name="l00010"></a>00010 <span class="comment">** LICENSE included in the packaging of this file.</span>
<a name="l00011"></a>00011 <span class="comment">**</span>
<a name="l00012"></a>00012 <span class="comment">** This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE</span>
<a name="l00013"></a>00013 <span class="comment">** WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.</span>
<a name="l00014"></a>00014 <span class="comment">**</span>
<a name="l00015"></a>00015 <span class="comment">** See http://www.quickfixengine.org/LICENSE for licensing information.</span>
<a name="l00016"></a>00016 <span class="comment">**</span>
<a name="l00017"></a>00017 <span class="comment">** Contact ask@quickfixengine.org if any conditions of this licensing are</span>
<a name="l00018"></a>00018 <span class="comment">** not clear to you.</span>
<a name="l00019"></a>00019 <span class="comment">**</span>
<a name="l00020"></a>00020 <span class="comment">****************************************************************************/</span>
<a name="l00021"></a>00021
<a name="l00022"></a>00022 <span class="preprocessor">#ifndef HAVE_ODBC</span>
<a name="l00023"></a>00023 <span class="preprocessor"></span><span class="preprocessor">#error OdbcConnection.h included, but HAVE_ODBC not defined</span>
<a name="l00024"></a>00024 <span class="preprocessor"></span><span class="preprocessor">#endif</span>
<a name="l00025"></a>00025 <span class="preprocessor"></span>
<a name="l00026"></a>00026 <span class="preprocessor">#ifdef HAVE_ODBC</span>
<a name="l00027"></a>00027 <span class="preprocessor"></span><span class="preprocessor">#ifndef FIX_ODBCCONNECTION_H</span>
<a name="l00028"></a>00028 <span class="preprocessor"></span><span class="preprocessor">#define FIX_ODBCCONNECTION_H</span>
<a name="l00029"></a>00029 <span class="preprocessor"></span>
<a name="l00030"></a>00030 <span class="preprocessor">#ifdef _MSC_VER</span>
<a name="l00031"></a>00031 <span class="preprocessor"></span><span class="preprocessor">#pragma warning( disable : 4503 4355 4786 4290 )</span>
<a name="l00032"></a>00032 <span class="preprocessor"></span><span class="preprocessor">#pragma comment( lib, "Odbc32" )</span>
<a name="l00033"></a>00033 <span class="preprocessor"></span><span class="preprocessor">#endif</span>
<a name="l00034"></a>00034 <span class="preprocessor"></span>
<a name="l00035"></a>00035 <span class="preprocessor">#include "<a class="code" href="_utility_8h.html">Utility.h</a>"</span>
<a name="l00036"></a>00036 <span class="preprocessor">#include <sql.h></span>
<a name="l00037"></a>00037 <span class="preprocessor">#include <sqlext.h></span>
<a name="l00038"></a>00038 <span class="preprocessor">#include <sqltypes.h></span>
<a name="l00039"></a>00039 <span class="preprocessor">#include <sstream></span>
<a name="l00040"></a>00040 <span class="preprocessor">#include "<a class="code" href="_database_connection_i_d_8h.html">DatabaseConnectionID.h</a>"</span>
<a name="l00041"></a>00041 <span class="preprocessor">#include "<a class="code" href="_database_connection_pool_8h.html">DatabaseConnectionPool.h</a>"</span>
<a name="l00042"></a>00042 <span class="preprocessor">#include "<a class="code" href="_exceptions_8h.html">Exceptions.h</a>"</span>
<a name="l00043"></a>00043 <span class="preprocessor">#include "<a class="code" href="_mutex_8h.html">Mutex.h</a>"</span>
<a name="l00044"></a>00044
<a name="l00045"></a>00045 <span class="keyword">namespace </span>FIX
<a name="l00046"></a>00046 {
<a name="l00047"></a>00047
<a name="l00048"></a>00048 <span class="keyword">inline</span> std::string odbcError( SQLSMALLINT statementType, SQLHANDLE handle )
<a name="l00049"></a>00049 {
<a name="l00050"></a>00050 SQLCHAR state[6];
<a name="l00051"></a>00051 SQLINTEGER error;
<a name="l00052"></a>00052 SQLCHAR text[SQL_MAX_MESSAGE_LENGTH];
<a name="l00053"></a>00053 SQLSMALLINT textLength;
<a name="l00054"></a>00054 RETCODE result = SQLGetDiagRec
<a name="l00055"></a>00055 ( statementType, handle, 1, state, &error, text, <span class="keyword">sizeof</span>(text), &textLength );
<a name="l00056"></a>00056 <span class="keywordflow">if</span>( result == SQL_SUCCESS || result == SQL_SUCCESS_WITH_INFO )
<a name="l00057"></a>00057 <span class="keywordflow">return</span> std::string( (<span class="keywordtype">char</span>*)text );
<a name="l00058"></a>00058 <span class="keywordflow">return</span> <span class="stringliteral">""</span>;
<a name="l00059"></a>00059 }
<a name="l00060"></a>00060
<a name="l00061"></a>00061 <span class="keyword">inline</span> <span class="keywordtype">bool</span> odbcSuccess( RETCODE result )
<a name="l00062"></a>00062 {
<a name="l00063"></a>00063 <span class="keywordflow">return</span> result == SQL_SUCCESS || result == SQL_SUCCESS_WITH_INFO;
<a name="l00064"></a>00064 }
<a name="l00065"></a>00065
<a name="l00066"></a>00066 <span class="keyword">class </span>OdbcQuery
<a name="l00067"></a>00067 {
<a name="l00068"></a>00068 <span class="keyword">public</span>:
<a name="l00069"></a>00069 OdbcQuery( <span class="keyword">const</span> std::string& query )
<a name="l00070"></a>00070 : m_statement( 0 ), m_result( 0 ), m_query( query )
<a name="l00071"></a>00071 {}
<a name="l00072"></a>00072
<a name="l00073"></a>00073 ~OdbcQuery()
<a name="l00074"></a>00074 {
<a name="l00075"></a>00075 close();
<a name="l00076"></a>00076 }
<a name="l00077"></a>00077
<a name="l00078"></a>00078 <span class="keywordtype">void</span> close()
<a name="l00079"></a>00079 {
<a name="l00080"></a>00080 <span class="keywordflow">if</span>( m_statement )
<a name="l00081"></a>00081 {
<a name="l00082"></a>00082 SQLFreeHandle( SQL_HANDLE_STMT, m_statement );
<a name="l00083"></a>00083 m_statement = 0;
<a name="l00084"></a>00084 m_result = 0;
<a name="l00085"></a>00085 }
<a name="l00086"></a>00086 }
<a name="l00087"></a>00087
<a name="l00088"></a>00088 <span class="keywordtype">bool</span> execute( HDBC connection )
<a name="l00089"></a>00089 {
<a name="l00090"></a>00090 <span class="keywordflow">if</span>( m_statement ) SQLFreeHandle( SQL_HANDLE_STMT, m_statement );
<a name="l00091"></a>00091 SQLAllocHandle( SQL_HANDLE_STMT, connection, &m_statement );
<a name="l00092"></a>00092 m_result = SQLExecDirect( m_statement, (SQLCHAR*)m_query.c_str(), m_query.size() );
<a name="l00093"></a>00093 <span class="keywordflow">if</span>( success() || m_result == SQL_NO_DATA )
<a name="l00094"></a>00094 <span class="keywordflow">return</span> <span class="keyword">true</span>;
<a name="l00095"></a>00095 m_reason = odbcError( SQL_HANDLE_STMT, m_statement );
<a name="l00096"></a>00096 SQLFreeHandle( SQL_HANDLE_STMT, m_statement );
<a name="l00097"></a>00097 m_statement = 0;
<a name="l00098"></a>00098 <span class="keywordflow">return</span> success();
<a name="l00099"></a>00099 }
<a name="l00100"></a>00100
<a name="l00101"></a>00101 <span class="keywordtype">bool</span> success()
<a name="l00102"></a>00102 {
<a name="l00103"></a>00103 <span class="keywordflow">return</span> odbcSuccess( m_result );
<a name="l00104"></a>00104 }
<a name="l00105"></a>00105
<a name="l00106"></a>00106 <span class="comment">/*int rows()</span>
<a name="l00107"></a>00107 <span class="comment"> {</span>
<a name="l00108"></a>00108 <span class="comment"> return (int)mysql_num_rows( m_result );</span>
<a name="l00109"></a>00109 <span class="comment"> }*/</span>
<a name="l00110"></a>00110
<a name="l00111"></a>00111 <span class="keyword">const</span> std::string& reason()
<a name="l00112"></a>00112 {
<a name="l00113"></a>00113 <span class="keywordflow">return</span> m_reason;
<a name="l00114"></a>00114 }
<a name="l00115"></a>00115
<a name="l00116"></a>00116 <span class="keywordtype">bool</span> fetch()
<a name="l00117"></a>00117 {
<a name="l00118"></a>00118 <span class="keywordflow">return</span> odbcSuccess( SQLFetch(m_statement) );
<a name="l00119"></a>00119 }
<a name="l00120"></a>00120
<a name="l00121"></a>00121 <span class="comment">/*char* getValue( int row, int column )</span>
<a name="l00122"></a>00122 <span class="comment"> {</span>
<a name="l00123"></a>00123 <span class="comment"> if( m_rows.empty() )</span>
<a name="l00124"></a>00124 <span class="comment"> {</span>
<a name="l00125"></a>00125 <span class="comment"> MYSQL_ROW row = 0;</span>
<a name="l00126"></a>00126 <span class="comment"> while( row = mysql_fetch_row( m_result ) )</span>
<a name="l00127"></a>00127 <span class="comment"> m_rows.push_back(row);</span>
<a name="l00128"></a>00128 <span class="comment"> }</span>
<a name="l00129"></a>00129 <span class="comment"> return m_rows[row][column];</span>
<a name="l00130"></a>00130 <span class="comment"> }*/</span>
<a name="l00131"></a>00131
<a name="l00132"></a>00132 HSTMT statement()
<a name="l00133"></a>00133 {
<a name="l00134"></a>00134 <span class="keywordflow">return</span> m_statement;
<a name="l00135"></a>00135 }
<a name="l00136"></a>00136
<a name="l00137"></a>00137 <span class="keywordtype">void</span> throwException() throw( IOException )
<a name="l00138"></a>00138 {
<a name="l00139"></a>00139 <span class="keywordflow">if</span>( !success() )
<a name="l00140"></a>00140 <span class="keywordflow">throw</span> IOException( <span class="stringliteral">"Query failed ["</span> + m_query + <span class="stringliteral">"] "</span> + reason() );
<a name="l00141"></a>00141 }
<a name="l00142"></a>00142
<a name="l00143"></a>00143 <span class="keyword">private</span>:
<a name="l00144"></a>00144 HSTMT m_statement;
<a name="l00145"></a>00145 RETCODE m_result;
<a name="l00146"></a>00146 std::string m_query;
<a name="l00147"></a>00147 std::string m_reason;
<a name="l00148"></a>00148 };
<a name="l00149"></a>00149
<a name="l00150"></a>00150 <span class="keyword">class </span>OdbcConnection
<a name="l00151"></a>00151 {
<a name="l00152"></a>00152 <span class="keyword">public</span>:
<a name="l00153"></a>00153 OdbcConnection
<a name="l00154"></a>00154 ( <span class="keyword">const</span> DatabaseConnectionID& <span class="keywordtype">id</span> )
<a name="l00155"></a>00155 : m_connection( 0 ), m_environment( 0 ), m_connectionID( id )
<a name="l00156"></a>00156 {
<a name="l00157"></a>00157 connect();
<a name="l00158"></a>00158 }
<a name="l00159"></a>00159
<a name="l00160"></a>00160 OdbcConnection
<a name="l00161"></a>00161 ( <span class="keyword">const</span> std::string& user, <span class="keyword">const</span> std::string& password,
<a name="l00162"></a>00162 <span class="keyword">const</span> std::string& connectionString )
<a name="l00163"></a>00163 : m_connection( 0 ), m_environment( 0 ), m_connectionID( <span class="stringliteral">""</span>, user, password, connectionString, 0 )
<a name="l00164"></a>00164 {
<a name="l00165"></a>00165 connect();
<a name="l00166"></a>00166 }
<a name="l00167"></a>00167
<a name="l00168"></a>00168 ~OdbcConnection()
<a name="l00169"></a>00169 {
<a name="l00170"></a>00170 <span class="keywordflow">if</span>( m_connection )
<a name="l00171"></a>00171 {
<a name="l00172"></a>00172 SQLDisconnect( m_connection );
<a name="l00173"></a>00173 SQLFreeHandle( SQL_HANDLE_DBC, m_connection );
<a name="l00174"></a>00174 SQLFreeHandle( SQL_HANDLE_ENV, m_environment );
<a name="l00175"></a>00175 }
<a name="l00176"></a>00176 }
<a name="l00177"></a>00177
<a name="l00178"></a>00178 <span class="keyword">const</span> DatabaseConnectionID& connectionID()
<a name="l00179"></a>00179 {
<a name="l00180"></a>00180 <span class="keywordflow">return</span> m_connectionID;
<a name="l00181"></a>00181 }
<a name="l00182"></a>00182
<a name="l00183"></a>00183 <span class="keywordtype">bool</span> connected()
<a name="l00184"></a>00184 {
<a name="l00185"></a>00185 Locker locker( m_mutex );
<a name="l00186"></a>00186 <span class="keywordflow">return</span> m_connected;
<a name="l00187"></a>00187 }
<a name="l00188"></a>00188
<a name="l00189"></a>00189 <span class="keywordtype">bool</span> reconnect()
<a name="l00190"></a>00190 {
<a name="l00191"></a>00191 Locker locker( m_mutex );
<a name="l00192"></a>00192 SQLDisconnect( m_connection );
<a name="l00193"></a>00193 SQLFreeHandle( SQL_HANDLE_DBC, m_connection );
<a name="l00194"></a>00194 m_connection = 0;
<a name="l00195"></a>00195 connect();
<a name="l00196"></a>00196 <span class="keywordflow">return</span> <span class="keyword">true</span>;
<a name="l00197"></a>00197 }
<a name="l00198"></a>00198
<a name="l00199"></a>00199 <span class="keywordtype">bool</span> execute( OdbcQuery& pQuery )
<a name="l00200"></a>00200 {
<a name="l00201"></a>00201 Locker locker( m_mutex );
<a name="l00202"></a>00202 <span class="keywordflow">if</span>( !pQuery.execute( m_connection ) )
<a name="l00203"></a>00203 {
<a name="l00204"></a>00204 reconnect();
<a name="l00205"></a>00205 <span class="keywordflow">return</span> pQuery.execute( m_connection );
<a name="l00206"></a>00206 }
<a name="l00207"></a>00207 <span class="keywordflow">return</span> <span class="keyword">true</span>;
<a name="l00208"></a>00208 }
<a name="l00209"></a>00209
<a name="l00210"></a>00210 <span class="keyword">private</span>:
<a name="l00211"></a>00211 <span class="keywordtype">void</span> connect()
<a name="l00212"></a>00212 {
<a name="l00213"></a>00213 m_connected = <span class="keyword">false</span>;
<a name="l00214"></a>00214
<a name="l00215"></a>00215 RETCODE result;
<a name="l00216"></a>00216 <span class="keywordflow">if</span>(!m_environment)
<a name="l00217"></a>00217 {
<a name="l00218"></a>00218 result = SQLAllocHandle( SQL_HANDLE_ENV, SQL_NULL_HANDLE, &m_environment );
<a name="l00219"></a>00219 <span class="keywordflow">if</span>( !odbcSuccess(result) )
<a name="l00220"></a>00220 <span class="keywordflow">throw</span> ConfigError( <span class="stringliteral">"Unable to allocate ODBC environment"</span> );
<a name="l00221"></a>00221
<a name="l00222"></a>00222 result = SQLSetEnvAttr(m_environment, SQL_ATTR_ODBC_VERSION, (<span class="keywordtype">void</span>*)SQL_OV_ODBC3, 0);
<a name="l00223"></a>00223 <span class="keywordflow">if</span>( !odbcSuccess(result) )
<a name="l00224"></a>00224 <span class="keywordflow">throw</span> ConfigError( <span class="stringliteral">"Unable to find ODBC version 3.0"</span> );
<a name="l00225"></a>00225 }
<a name="l00226"></a>00226
<a name="l00227"></a>00227 result = SQLAllocHandle( SQL_HANDLE_DBC, m_environment, &m_connection );
<a name="l00228"></a>00228 <span class="keywordflow">if</span>( !odbcSuccess(result) )
<a name="l00229"></a>00229 <span class="keywordflow">throw</span> ConfigError( <span class="stringliteral">"Unable to allocate ODBC connection"</span> );
<a name="l00230"></a>00230
<a name="l00231"></a>00231 std::stringstream connectionStream;
<a name="l00232"></a>00232 std::string connectionString = m_connectionID.getHost();
<a name="l00233"></a>00233 <span class="keywordflow">if</span>( m_connectionID.getHost().find(<span class="stringliteral">"UID="</span>) == std::string::npos )
<a name="l00234"></a>00234 connectionStream << <span class="stringliteral">"UID="</span> << m_connectionID.getUser() << <span class="stringliteral">";"</span>;
<a name="l00235"></a>00235 <span class="keywordflow">if</span>( m_connectionID.getHost().find(<span class="stringliteral">"PWD="</span>) == std::string::npos )
<a name="l00236"></a>00236 connectionStream << <span class="stringliteral">"PWD="</span> << m_connectionID.getPassword() << <span class="stringliteral">";"</span>;
<a name="l00237"></a>00237 connectionStream << m_connectionID.getHost();
<a name="l00238"></a>00238 connectionString = connectionStream.str();
<a name="l00239"></a>00239
<a name="l00240"></a>00240 SQLCHAR connectionStringOut[255];
<a name="l00241"></a>00241
<a name="l00242"></a>00242 result = SQLDriverConnect(
<a name="l00243"></a>00243 m_connection, NULL,
<a name="l00244"></a>00244 (SQLCHAR*)connectionString.c_str(), SQL_NTS,
<a name="l00245"></a>00245 connectionStringOut, 255,
<a name="l00246"></a>00246 0, SQL_DRIVER_NOPROMPT );
<a name="l00247"></a>00247
<a name="l00248"></a>00248 <span class="keywordflow">if</span>( !odbcSuccess(result) )
<a name="l00249"></a>00249 {
<a name="l00250"></a>00250 std::string error = odbcError( SQL_HANDLE_DBC, m_connection );
<a name="l00251"></a>00251 <span class="keywordflow">throw</span> ConfigError( error );
<a name="l00252"></a>00252 }
<a name="l00253"></a>00253
<a name="l00254"></a>00254 m_connected = <span class="keyword">true</span>;
<a name="l00255"></a>00255 }
<a name="l00256"></a>00256
<a name="l00257"></a>00257 HENV m_environment;
<a name="l00258"></a>00258 HDBC m_connection;
<a name="l00259"></a>00259 <span class="keywordtype">bool</span> m_connected;
<a name="l00260"></a>00260 DatabaseConnectionID m_connectionID;
<a name="l00261"></a>00261 Mutex m_mutex;
<a name="l00262"></a>00262 };
<a name="l00263"></a>00263 }
<a name="l00264"></a>00264
<a name="l00265"></a>00265 <span class="preprocessor">#endif</span>
<a name="l00266"></a>00266 <span class="preprocessor"></span><span class="preprocessor">#endif</span>
</pre></div></div>
<hr><address><small>
Generated on Mon Apr 5 20:59:50 2010 for QuickFIX by <a href="http://www.doxygen.org/index.html">
<img src="doxygen.gif" alt="doxygen" align="middle" border=0 width=110 height=53>
</a> 1.6.1 written by <a href="mailto:dimitri@stack.nl">Dimitri van Heesch</a>,
© 1997-2001</small></address>
</body>
</html>
|
