C++搭建集群聊天室（九）：数据库代码及用户模型代码封装

---

接下来，实现这些函数

db.cpp

同样，在 src/server 目录下新建 db 文件夹，用于存放数据库文件。

新建 db.cpp

#include “db.hpp”

MySQL::MySQL() { _conn = mysql_init(nullptr); }

// 释放数据库连接资源

MySQL::~MySQL()

{

if (_conn != nullptr)

mysql_close(_conn);

}

// 连接数据库

bool MySQL::connect()

{

MYSQL *p = mysql_real_connect(_conn, server.c_str(), user.c_str(), password.c_str(), dbname.c_str(), 3306, nullptr, 0);

if (p != nullptr)

{

mysql_query(_conn, “set names gbk”);

}

return p;

}

// 更新操作

bool MySQL::update(string sql)

{

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

{

LOG_INFO << FILE << “:” << LINE << “:” << sql << “更新失败!”;

return false;

}

return true;

}

// 查询操作

MYSQL_RES * MySQL::query(string sql)

{

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

{

LOG_INFO << FILE << “:” << LINE << “:” << sql << “查询失败!”;

return nullptr;

}

return mysql_use_result(_conn);

}