MySQL的C++简单封装

/*
*介绍：MySQL的简单封装，支持流操作输入输出MySQL语句，然而并没有什么软用，大二学生自娱自乐，有不足求指点
*作者：MrEO
*日期：2016.3.26
*/

头文件 my_sql.h

 #ifndef MY_SQL_H
 #define MY_SQL_H

 #include <mysql.h>
 #include <iostream>
 #include <string>
 #include <iomanip>

 class My_sql
 {
     public:
         My_sql();                    　//初始化API、初始化结构、设置字符集、连接数据库
         ~My_sql();                     //释放内存、关闭服务器连接、结束服务器库、结束API库
         unsigned row_count();       　 //返回集中的行数
         unsigned field_count();        //返回集中的列数
         friend std::ostream & operator <<(std::ostream &out, My_sql &sql);
 　　　　 //通过std::ostream输出列名、结果集
         friend std::istream & operator >>(std::istream &in, My_sql &sql);
 　　　　 //通过std::istream输入执行SQL语句，并保存结果集地址至成员变量result
 //      friend My_sql & operator >>(const std::string &q, My_sql &sql);
 　　　　 //通过std::string输入执行SQL语句 ，并保存结果集地址至成员变量result
         friend My_sql & operator <<(My_sql &sql, const std::string &q);
 　　　　 //通过std::string输入执行SQL语句 ，并保存结果集地址至成员变量result

     protected:
         bool query(const std::string &q);   　　    //执行SQL语句，保存完整的结果集地址至成员变量result
         My_sql & display_field_name(std::ostream &out = std::cout, int setw_n = );    //输出列名
         My_sql & display_result(std::ostream &out = std::cout, int setw_n = );        //输出结果集
         MYSQL mysql;                      //对象
         MYSQL_RES *result = NULL;         //结果集
         MYSQL_FIELD *field = NULL;        //列
         MYSQL_ROW row;                    //行
 };

 #endif

实现文件 my_sql.cpp

 #include "my_sql.h"

 My_sql::My_sql()
 {
     mysql_library_init( NULL, ,  );                            //初始化MySQL C API库
     mysql_init( &mysql );                                        //获取或初始化MYSQL结构。
     mysql_options( &mysql, MYSQL_SET_CHARSET_NAME, "utf8" );     //设置字符集
     mysql_real_connect( &mysql, "localhost", "root", "password", "my_sql", , NULL, CLIENT_MULTI_STATEMENTS );
                                                                  //连接数据库
 }
 My_sql::~My_sql()
 {
     mysql_free_result( result );      //释放结果集使用的内存
     mysql_close( &mysql );            //关闭服务器连接
     mysql_server_end();               //最终确定嵌入式服务器库
     mysql_library_end();              //最终确定MySQL C API库
 }
 bool My_sql::query(const std::string &q)
 {
     result = NULL;
     mysql_query( &mysql, q.c_str() );         //执行指定为“以Null终结的字符串”的SQL查询。
     result = mysql_store_result( &mysql );    //检索完整的结果集至客户端
     if ( NULL == result ) {
         std::cerr << "Query Error!";
         return false;
     }
     return true;
 }

 unsigned My_sql::row_count()
 {
     return mysql_num_rows( result );      //返回集中的行数
 }
 unsigned My_sql::field_count()
 {
     return mysql_num_fields( result );    //返回集中的列数
 }

 My_sql & My_sql::display_field_name(std::ostream &out, int setw_n)
 {
     for ( unsigned int i = ; i < field_count(); ++i ) {
         out << std::setw(setw_n) << ( mysql_fetch_field_direct( result, i ) -> name );
         //给定字段编号，返回表字段的类型。输出列名
     }
     out << std::endl;
     return *this;
 }

 My_sql & My_sql::display_result(std::ostream &out, int setw_n)
 {
     while( row = mysql_fetch_row( result ) )        //从结果集中获取下一行
     {
         for(int i = ; i < field_count(); ++i )
           {
             if ( NULL == row[i]) {
                 out << std::setw(setw_n) << "(NULL)";
             }
             else {
                 out << std::setw(setw_n) << row[i];
             }
         }
         out << std::endl;
     }
     return *this;
 }

 std::ostream & operator <<(std::ostream &out, My_sql &sql)
 {
     sql.display_field_name(out).display_result(out);
     //输出列名和结果集 

     return out;
 }

 std::istream & operator >>(std::istream &in,  My_sql &sql)
 {
     std::string q;
     std::getline(in, q);    //输入整行
     if ( !sql.query(q) ) {
         exit();
     }
     return in;
 }

 //My_sql & operator >>(const std::string &q, My_sql &sql)
 //{
 //    if ( !sql.query(q) ) {
 //        exit(1);
 //    }
 //    return sql;
 //}

 My_sql & operator <<(My_sql &sql, const std::string &q)
 {
     if ( !sql.query(q) ) {
         exit();
     }
     return sql;
 }

调用实例 main.cpp

 #include <iostream>
 #include "my_sql.h"

 using std::cin;
 using std::cout;

 int main(int argc, char* argv[])
 {
     My_sql sql;
     sql << "SELECT * FROM my_table";
     cout << sql;
     cin >> sql;
     cout << sql;
     return ;
 }