c++封装odbc类

亦凉 2022-08-06 13:11 264阅读 0赞

**\[cpp\]** [view plain][] [copy][view plain] [print][view plain] [?][view plain]

1.  //DBUtil.h
2.  //By 小E QQ592646022
3.  \#include <windows.h>
4.  \#include <iostream>
5.  \#include <sql.h>
6.  \#include <sqlext.h>
7.  \#include <vector>
8.  \#include <string>
9.  **using****namespace** std; 
10. 
11. **class** ODBC 
12. \{ 
13. **private**: 
14.  SQLHANDLE hEnv; 
15.  SQLHANDLE hDbc; 
16.  SQLHANDLE hStmt; 
17.  SQLRETURN retCode; 
18.  SQLINTEGER retErro; 
19.  SQLINTEGER rowCount; 
20.  SQLSMALLINT colCount; 
21. **bool** bState; 
22. **char**\* pszUName; 
23. **char**\* pszUPassword; 
24. **char**\* pszDSN; 
25. **public**: 
26.  ODBC(); 
27.  ~ODBC(); 
28. **bool** Connect(); 
29. **bool** Close(); 
30. **bool** IsOpen(); 
31. **int** GetRowCount()\{ **return** rowCount;\} 
32. **int** GetColCount()\{ **return** colCount;\} 
33.  vector<string\*> ExecuteQueryVector(**const****char**\* pszSql); 
34. **int** ExecuteQuery(**const****char**\* pszSql); //执行查询
35. **int** ExecuteNonQuery(**const****char**\* pszSql);//执行非查询(更新或删除)
36. **int** ExecuteUpdate(**const****char**\* pszSql); //执行更新
37. \}; 
38. 
39. ODBC::ODBC() 
40. \{ 
41.  bState=**false**; 
42.  rowCount=colCount=0; 
43.  retCode=SQLAllocHandle(SQL\_HANDLE\_ENV,NULL,&hEnv); 
44. **if**((retCode!=SQL\_SUCCESS)&& (retCode != SQL\_SUCCESS\_WITH\_INFO)) 
45.  \{ 
46.  cout<<"Erro AllocHandle"<<retCode<<endl; 
47. **return**; 
48.  \} 
49.  retCode=SQLSetEnvAttr(hEnv,SQL\_ATTR\_ODBC\_VERSION,(SQLPOINTER) SQL\_OV\_ODBC3,SQL\_IS\_INTEGER); 
50. **if**((retCode!=SQL\_SUCCESS)&& (retCode != SQL\_SUCCESS\_WITH\_INFO)) 
51.  \{ 
52.  cout<<"Erro AllocHandle"<<retCode<<endl; 
53.  SQLFreeHandle( SQL\_HANDLE\_DBC, hEnv ); 
54. **return**; 
55.  \} 
56.  retCode=SQLAllocHandle(SQL\_HANDLE\_DBC,hEnv,&hDbc); 
57. **if**((retCode!=SQL\_SUCCESS)&& (retCode != SQL\_SUCCESS\_WITH\_INFO)) 
58.  \{ 
59.  cout<<"Erro AllocHandle"<<retCode<<endl; 
60.  SQLFreeHandle( SQL\_HANDLE\_DBC, hEnv ); 
61. **return**; 
62.  \} 
63.  pszUName="sa"; //用户名
64.  pszUPassword="cty"; //密码
65.  pszDSN="QQ"; //数据源名
66. 
67. 
68. \} 
69. ODBC::~ODBC() 
70. \{ 
71.  Close(); 
72. \} 
73. **bool** ODBC::Connect() 
74. \{ 
75. **if**(bState==**false**) 
76.  \{ 
77.  retCode=SQLConnect(hDbc,(SQLCHAR\*) pszDSN,SQL\_NTS,(SQLCHAR\*) pszUName,SQL\_NTS,(SQLCHAR\*) pszUPassword,SQL\_NTS); 
78. **if**((retCode != SQL\_SUCCESS) && (retCode != SQL\_SUCCESS\_WITH\_INFO)) 
79.  \{ 
80.  cout<<"Erro Connect "<<retCode<<endl; 
81.  SQLFreeHandle( SQL\_HANDLE\_DBC, hDbc ); 
82. **return****false**; 
83.  \} 
84.  retCode=SQLAllocHandle(SQL\_HANDLE\_STMT,hDbc,&hStmt); 
85. **if**((retCode != SQL\_SUCCESS) && (retCode != SQL\_SUCCESS\_WITH\_INFO)) 
86.  \{ 
87.  cout<<"Erro Connect "<<retCode<<endl; 
88.  SQLDisconnect( hDbc ); 
89.  SQLFreeHandle( SQL\_HANDLE\_DBC, hDbc); 
90. **return****false**; 
91.  \} 
92.  \} 
93.  bState=**true**; 
94.  cout<<"success!"<<endl; 
95. **return****true**; 
96. 
97. \} 
98. **int** ODBC::ExecuteQuery(**const****char**\* pszSql) 
99. \{ 
100. 
101. **if**(pszSql==NULL) 
102. **return** 0; 
103.  cout<<"hStmt="<<hStmt<<endl; 
104.  retCode=SQLExecDirect(hStmt,(SQLCHAR\*)pszSql,SQL\_NTS); 
105. **if**((retCode != SQL\_SUCCESS) && (retCode != SQL\_SUCCESS\_WITH\_INFO)) 
106.  \{ 
107.  cout<<"Erro ExecDirect "<<retCode<<endl; 
108. **return** \-1; 
109.  \} 
110. /\* retCode=SQLRowCount(hStmt,&rowCount); //不受select 影响。。
111.  if((retCode != SQL\_SUCCESS) && (retCode != SQL\_SUCCESS\_WITH\_INFO))
112.  \{ 
113.  cout<<"Erro RowCount "<<retCode<<endl;
114.  return -1;
115.  \}\*/
116.  retCode=SQLNumResultCols(hStmt,&colCount); 
117. **if**((retCode != SQL\_SUCCESS) && (retCode != SQL\_SUCCESS\_WITH\_INFO)) 
118.  \{ 
119.  cout<<"Erro ColCount "<<retCode<<endl; 
120. **return** \-1; 
121.  \} 
122.  rowCount=0; 
123. **while**(SQL\_NO\_DATA!=SQLFetch(hStmt)) 
124.  \{ 
125. //cout<<pszBuf<<endl;
126.  rowCount++; 
127.  \} 
128.  SQLCancel(hStmt); 
129. **return** rowCount; 
130. \} 
131. **int** ODBC::ExecuteNonQuery(**const****char**\* pszSql) 
132. \{ 
133.  rowCount=0; 
134. **if**(pszSql==NULL) 
135. **return** 0; 
136. //cout<<"hStmt="<<hStmt<<endl;
137.  retCode=SQLExecDirect(hStmt,(SQLCHAR\*)pszSql,SQL\_NTS); 
138. **if**((retCode != SQL\_SUCCESS) && (retCode != SQL\_SUCCESS\_WITH\_INFO)) 
139.  \{ 
140.  cout<<"Erro ExecDirect "<<retCode<<endl; 
141. **return** \-1; 
142.  \} 
143.  retCode=SQLRowCount(hStmt,&rowCount); 
144. **if**((retCode != SQL\_SUCCESS) && (retCode != SQL\_SUCCESS\_WITH\_INFO)) 
145.  \{ 
146.  cout<<"Erro RowCount "<<retCode<<endl; 
147. **return** \-1; 
148.  \} 
149.  retCode=SQLNumResultCols(hStmt,&colCount); 
150. **if**((retCode != SQL\_SUCCESS) && (retCode != SQL\_SUCCESS\_WITH\_INFO)) 
151.  \{ 
152.  cout<<"Erro ColCount "<<retCode<<endl; 
153. **return** \-1; 
154.  \} 
155. 
156.  SQLCancel(hStmt); 
157. **return** rowCount; 
158. \} 
159. vector<string\* > ODBC::ExecuteQueryVector(**const****char**\* pszSql) 
160. \{ 
161.  vector<string\* > v; 
162. **if**(pszSql==NULL) 
163. **return** 0; 
164.  retCode=SQLExecDirect(hStmt,(SQLCHAR\*)pszSql,SQL\_NTS); 
165. **if**((retCode != SQL\_SUCCESS) && (retCode != SQL\_SUCCESS\_WITH\_INFO)) 
166.  \{ 
167.  cout<<"Erro ExecDirect "<<retCode<<endl; 
168. **return** \-1; 
169.  \} 
170. /\* retCode=SQLRowCount(hStmt,&rowCount); //不受select 影响。。
171.  if((retCode != SQL\_SUCCESS) && (retCode != SQL\_SUCCESS\_WITH\_INFO))
172.  \{ 
173.  cout<<"Erro RowCount "<<retCode<<endl;
174.  return -1;
175.  \}\*/
176.  retCode=SQLNumResultCols(hStmt,&colCount); 
177. **if**((retCode != SQL\_SUCCESS) && (retCode != SQL\_SUCCESS\_WITH\_INFO)) 
178.  \{ 
179.  cout<<"Erro ColCount "<<retCode<<endl; 
180. **return** \-1; 
181.  \} 
182.  rowCount=0; 
183.  SQLINTEGER colLen = 0; 
184.  SQLSMALLINT buf\_len = 0; 
185.  SQLINTEGER colType = 0; 
186. 
187. **while**(**true**) 
188.  \{ 
189. **char** sz\_buf\[256\]; 
190. **char**\* pszBuf; 
191.  SQLINTEGER buflen; 
192.  string\* rowData=**new** string\[colCount+1\]; 
193. **if**(SQLFetch(hStmt)==SQL\_NO\_DATA) 
194.  \{ 
195. **break**; 
196.  \} 
197. **for**(**int** i=1;i<=colCount;i++) 
198.  \{ 
199.  SQLColAttribute(hStmt, i, SQL\_DESC\_NAME, sz\_buf, 256, &buf\_len, 0); 
200.  SQLColAttribute(hStmt, i, SQL\_DESC\_TYPE, 0, 0, 0, &colType); 
201.  SQLColAttribute(hStmt, i, SQL\_DESC\_LENGTH, NULL, 0, 0, &colLen); 
202.  pszBuf=**new****char**\[colLen+1\]; 
203.  pszBuf\[0\]='/0'; 
204.  SQLGetData(hStmt,i,SQL\_C\_CHAR,pszBuf,50,&buflen); 
205.  rowData\[i-1\]=pszBuf; 
206.  \} 
207.  v.push\_back(rowData); 
208.  rowCount++; 
209. //cout<<"rowData\[0\]="<< (rowData\[0\])<<" rowData\[1\]="<<rowData\[1\]<<" rowData\[2\]="<<rowData\[2\]<<endl;;
210.  \} 
211.  SQLCancel(hStmt); 
212. **return** v; 
213. \} 
214. 
215. **bool** ODBC::Close() 
216. \{ 
217. **if**(bState) 
218.  \{ 
219.  SQLDisconnect(hDbc); 
220.  SQLFreeHandle(SQL\_HANDLE\_DBC,hDbc); 
221.  SQLFreeHandle(SQL\_HANDLE\_ENV,hEnv); 
222.  bState=**false**; 
223.  \} 
224. 
225. **return****true**; 
226. \} 
227. **bool** ODBC::IsOpen() 
228. \{ 
229. **return** bState; 
230. \} 
231. 
232. //Main.cpp 测试用 需要自行添加数据源 在odbc里面
233. \#include <iostream>
234. \#include "DBUtil.h"
235. \#include <string.h>
236. 
237. **using****namespace** std; 
238. **int** main() 
239. \{ 
240.  ODBC odbc; 
241.  odbc.Connect(); 
242.  vector<string\* > vv=odbc.ExecuteQueryVector("select \* from users"); 
243.  vector<string\* >::iterator it; 
244. **for**(it=vv.begin();it!=vv.end();it++) 
245.  \{ 
246. **for**(**int** i=0;i<odbc.GetColCount();i++) 
247.  \{ 
248.  cout<<(\*it)\[i\]<<" "; 
249.  \} 
250.  cout<<(\*it)\[6\]<<" "<<endl; 
251.  \} 
252. //odbc.ExecuteNonQuery("insert into users values('1','abcd',34,'男','2010-8-9','')");
253.  cout<<odbc.ExecuteNonQuery("delete from users where ID='10017'"); 
254. **return** 0; 
255. \}

原文地址：http://blog.csdn.net/evilbinary\_root/article/details/6358469

[view plain]: http://blog.csdn.net/evilbinary_root/article/details/6358469#