MySQL的C++簡單封裝

/*
*介紹：MySQL的簡單封裝，支持流操作輸入輸出MySQL語句，然而並沒有什麼軟用，大二學生自娛自樂，有不足求指點
*作者：MrEO
*日期：2016.3.26
*/

頭文件 my_sql.h

 1 #ifndef MY_SQL_H
 2 #define MY_SQL_H
 3 
 4 #include <mysql.h>
 5 #include <iostream>
 6 #include <string>
 7 #include <iomanip>
 8 
 9 class My_sql
10 {
11     public:
12         My_sql();                    　//初始化API、初始化結構、設置字元集、連接資料庫 
13         ~My_sql();                     //釋放記憶體、關閉伺服器連接、結束伺服器庫、結束API庫 
14         unsigned row_count();       　 //返回集中的行數 
15         unsigned field_count();        //返回集中的列數 
16         friend std::ostream & operator <<(std::ostream &out, My_sql &sql);
17 　　　　 //通過std::ostream輸出列名、結果集 
18         friend std::istream & operator >>(std::istream &in, My_sql &sql);
19 　　　　 //通過std::istream輸入執行SQL語句，並保存結果集地址至成員變數result 
20 //      friend My_sql & operator >>(const std::string &q, My_sql &sql);
21 　　　　 //通過std::string輸入執行SQL語句 ，並保存結果集地址至成員變數result
22         friend My_sql & operator <<(My_sql &sql, const std::string &q);
23 　　　　 //通過std::string輸入執行SQL語句 ，並保存結果集地址至成員變數result
24 
25     protected:
26         bool query(const std::string &q);   　　    //執行SQL語句，保存完整的結果集地址至成員變數result
27         My_sql & display_field_name(std::ostream &out = std::cout, int setw_n = 16);    //輸出列名 
28         My_sql & display_result(std::ostream &out = std::cout, int setw_n = 16);        //輸出結果集 
29         MYSQL mysql;                      //對象 
30         MYSQL_RES *result = NULL;         //結果集 
31         MYSQL_FIELD *field = NULL;        //列 
32         MYSQL_ROW row;                    //行 
33 };
34 
35 #endif

實現文件 my_sql.cpp

 1 #include "my_sql.h"
 2 
 3 My_sql::My_sql()
 4 {
 5     mysql_library_init( NULL, 0, 0 );                            //初始化MySQL C API庫
 6     mysql_init( &mysql );                                        //獲取或初始化MYSQL結構。
 7     mysql_options( &mysql, MYSQL_SET_CHARSET_NAME, "utf8" );     //設置字元集
 8     mysql_real_connect( &mysql, "localhost", "root", "password", "my_sql", 0, NULL, CLIENT_MULTI_STATEMENTS );    
 9                                                                  //連接資料庫
10 }
11 My_sql::~My_sql()
12 {
13     mysql_free_result( result );      //釋放結果集使用的記憶體
14     mysql_close( &mysql );            //關閉伺服器連接 
15     mysql_server_end();               //最終確定嵌入式伺服器庫
16     mysql_library_end();              //最終確定MySQL C API庫 
17 } 
18 bool My_sql::query(const std::string &q)
19 {
20     result = NULL; 
21     mysql_query( &mysql, q.c_str() );         //執行指定為“以Null終結的字元串”的SQL查詢。 
22     result = mysql_store_result( &mysql );    //檢索完整的結果集至客戶端 
23     if ( NULL == result ) {
24         std::cerr << "Query Error!";
25         return false;
26     }
27     return true;
28 }
29 
30 unsigned My_sql::row_count()
31 {
32     return mysql_num_rows( result );      //返回集中的行數 
33 }
34 unsigned My_sql::field_count()
35 {
36     return mysql_num_fields( result );    //返回集中的列數 
37 }
38 
39 My_sql & My_sql::display_field_name(std::ostream &out, int setw_n)
40 {
41     for ( unsigned int i = 0; i < field_count(); ++i ) {
42         out << std::setw(setw_n) << ( mysql_fetch_field_direct( result, i ) -> name );
43         //給定欄位編號，返回表欄位的類型。輸出列名 
44     }
45     out << std::endl; 
46     return *this;
47 }
48 
49 My_sql & My_sql::display_result(std::ostream &out, int setw_n)
50 {
51     while( row = mysql_fetch_row( result ) )        //從結果集中獲取下一行 
52     {
53         for(int i = 0; i < field_count(); ++i )
54           {
55             if ( NULL == row[i]) {
56                 out << std::setw(setw_n) << "(NULL)";
57             }
58             else {
59                 out << std::setw(setw_n) << row[i];
60             }
61         }
62         out << std::endl; 
63     }
64     return *this;
65 }
66 
67 std::ostream & operator <<(std::ostream &out, My_sql &sql)
68 {
69     sql.display_field_name(out).display_result(out);
70     //輸出列名和結果集 
71     
72     return out;
73 }
74 
75 std::istream & operator >>(std::istream &in,  My_sql &sql)
76 {
77     std::string q;
78     std::getline(in, q);    //輸入整行 
79     if ( !sql.query(q) ) {
80         exit(1);
81     }
82     return in;
83 }
84 
85 //My_sql & operator >>(const std::string &q, My_sql &sql)
86 //{
87 //    if ( !sql.query(q) ) {
88 //        exit(1);
89 //    }
90 //    return sql;
91 //}
92 
93 My_sql & operator <<(My_sql &sql, const std::string &q)
94 {
95     if ( !sql.query(q) ) {
96         exit(1);
97     }
98     return sql;
99 }

調用實例 main.cpp

 1 #include <iostream>
 2 #include "my_sql.h"
 3 
 4 using std::cin;
 5 using std::cout;
 6 
 7 int main(int argc, char* argv[])
 8 {
 9     My_sql sql;
10     sql << "SELECT * FROM my_table";
11     cout << sql;
12     cin >> sql;
13     cout << sql;
14     return 0;
15 }