C++项目——集群聊天服务器项目(六)MySQL模块

news2025/3/4 12:03:41

Hello，大家好啊，最近比较忙，没来得及更新项目，实在抱歉~今天就恢复更新拉~

在验证完网络模块与业务模块代码可以正常使用后，需完成的操作是与底层数据库进行交互，为实现各类用户查询、增删业务奠定良好的基础。本节内容主要完成与底层数据库的交互操作，一起来看看把~

项目流程如下：

1、项目环境搭建 C++项目——集群聊天服务器项目(一)项目介绍、环境搭建、Boost库安装、Muduo库安装、Linux与vscode配置_c++集群聊天服务器-CSDN博客

2、Json第三方库介绍

C++项目——集群聊天服务器项目(二)Json第三方库-CSDN博客

3、muduo网络库介绍

C++项目——集群聊天服务器项目(三)muduo网络库-CSDN博客

4、MySQL数据库创建

C++项目——集群聊天服务器项目(四)MySQL数据库-CSDN博客

5、网络模块与业务模块代码编写

C++项目——集群聊天服务器项目(五)网络模块与业务模块-CSDN博客

一、Mysql环境

在第一节中环境搭建中，有提示大家安装MySQL开发包，可以使用命令

sudo find /usr -name libmysqlclient*

去查看MySQL开发包是否安装好，查询结果如下：

各位这里十分重要哈，这牵扯到后续项目是否能正常登录查询操作底层数据库中存储的数据。

二、MySQL模块编写

2.1 编写MySQL类

在include/server目录下创建文件夹db，表示database数据库，db文件夹下创建头文件db.hpp，负责连接、操作数据库的基本操作

dp.hpp头文件如下：

#ifndef DB_H
#define DB_H

#include <mysql/mysql.h>//mysql的头文件 
#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

相应的，在src/server/中创建文件夹db，db文件夹下创建库文件db.cpp,如图

对创建的MySQL类进行函数实现，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，如果不设置，从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;
}