MySQL数据库模块代码DB

最新推荐文章于 2025-09-17 08:33:25 发布

原创最新推荐文章于 2025-09-17 08:33:25 发布 · 522 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#数据库 #mysql #database

基于muduo的聊天服务器专栏收录该内容

15 篇文章

订阅专栏

本文介绍了一个用于简化MySQL数据库操作的C++类封装方案。通过该方案，可以实现数据库连接的初始化与释放、执行更新操作及查询操作等功能。

MySQL数据库代码封装

db.hpp
db.cpp

为了减少业务模块ChatService的复杂度，我们将数据库的增删改查操作封装成一个单独的类。

db.hpp

我们在include下的server下的db下，创建头文件db.hpp:

#ifndef DB_H
#define DB_H

#include <mysql/mysql.h>
#include <string>

using namespace std;

//数据库操作类
class MySQL
{
public:
    //初始化数据库连接
    MySQL();
    //释放数据库连接
    ~MySQL();
    //连接数据库
    bool connect();
    //更新操作
    bool update(string sql);
    //查询操作
    MYSQL_RES *query(string sql);
    //获取连接
    MYSQL *getConnection();

private:
    //数据库连接
    MYSQL *_conn;
};

#endif

db.cpp

我们在src下的server下的db下创建文件db.cpp：

#include "db.hpp"
#include <muduo/base/Logging.h>

//数据库配置信息
static string server = "127.0.0.1";
static string user = "root";
static string password = "123456";
static string dbname = "chat";

//初始化数据库连接
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)
    {
        //C和C++代码默认的编码字符是ASCII，如果不设置为gbk，从MySQL上拉下来的中文显示
        mysql_query(_conn, "set names gbk");
        LOG_INFO << "connect mysql success";
    }
    else
    {
        LOG_INFO << "connect mysql fail!";
    }

    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);
}

//获取连接
MYSQL *MySQL::getConnection()
{
    return _conn;
}