linux c++ mysql_linux下C++访问mysql数据库（2）

在这个程序中，比上一次的要完善一些，实现了通过C++对数据表进行添加，修改，插入，删除的功能。

头文件——myDB.h：

#ifndef myDB_class

#define myDB_class

#include 

#include 

#include 

class myDB

{

public:

myDB();

~myDB();

int initDB(std::string host, std::string user, std::string password, std::string db_name);

void Run();

private:

int exeSQL();

int insertSQL();

int deleteSQL();

int updateSQL();

MYSQL *connection;

MYSQL_RES *result;

MYSQL_ROW row;

};

#endif

myDB.cpp

#include "myDB.h"

#include 

#include 

using namespace std;

myDB::myDB()

{

connection = mysql_init(NULL); // 初始化数据库的连接变量

if(connection == NULL)

{

std::cout <

exit(1);// exit(1)表示发生错误后退出程序， exit(0)表示正常退出

}

}

myDB::~myDB()

{

if(connection != NULL)

{

mysql_close(connection); // 关闭数据库连接

}

}

int myDB::initDB(std::string host, std::string user, std::string password, std::string db_name)

{

char value = 1;

mysql_options(connection_,MYSQL_OPT_RECONNECT,(char*)&value);//mysql自动重连

connection = mysql_real_connect(connection, host.c_str(), user.c_str(), password.c_str(), db_name.c_str(), 0, NULL, 0); // 建立数据库连接

if(connection == NULL)

{

std::cout <

exit(1);// exit(1)表示发生错误后退出程序， exit(0)表示正常退出

}

return 0;

}

int myDB::exeSQL()

{

// mysql_query()执行成功返回0，失败返回非0值。与PHP中不一样

std::string sql;

sql = "select * from user;";

if(mysql_query(connection, sql.c_str()))

{

std::cout <

exit(1);

}

else

{

result = mysql_use_result(connection); // 获取结果集

// mysql_field_count()返回connection查询的列数

for(int i=0; i 

{

// 获取下一行

row = mysql_fetch_row(result);

if(row <= 0)

{

break;

}

// mysql_num_fields()返回结果集中的字段数

for(int j=0; j 

{

std::cout <

}

std::cout <

}

std::cout <

// 释放结果集的内存

mysql_free_result(result);

}

return 0;

}

int myDB::insertSQL()

{

std::string username;

std::string psd;

std::string level;

std::cout <

std::cin >> username >> psd >> level;

string sql = "insert into user(id,username,password,level) values ( NULL,'" + username +"','"+ psd  +"', "+ level +");";

if(mysql_query(connection, sql.c_str()))

{

std::cout <

exit(1);

}

std::cout <

exeSQL();

return 0;

}

int myDB::deleteSQL()

{

std::string id;

std::cout <

std::cin >> id;

std::string sql;

sql = "delete from user where id = " + id + ";";

if(mysql_query(connection, sql.c_str()))

{

std::cout <

exit(1);

}

std::cout <

exeSQL();

return 0;

}

int myDB::updateSQL()

{

std::string id;

std::string thing;

std::cout <

std::cin >> id >> thing;

std::string sql;

sql = "update user set " + thing + "where id =" + id + ";";

if(mysql_query(connection, sql.c_str()))

{

std::cout <

exit(1);

}

std::cout <

exeSQL();

return 0;

}

void myDB::Run()

{

int i = 1;

while(i != 0)

{

int sel;

std::cout <

std::cin >> sel;

switch(sel)

{

case 1:exeSQL();

break;

case 2:insertSQL();

break;

case 3:deleteSQL();

break;

case 4:updateSQL();

break;

case 5:i = 0;

break;

default:std::cout <

break;

}

}

}

main.cpp

#include 

#include "myDB.h"

int main()

{

myDB db;

db.initDB("localhost", "root", "wei123", "wei");

db.Run();

return 0;

}

所有代码就都在这里了，下面是makefile文件

Makefile

mydb:main.cpp myDB.cpp

g++ -o mydb main.cpp myDB.cpp -lmysqlclient

clean:

rm -f *.o mydb

(g++ 和rm之前用Tab隔开)