集群聊天服务器项目【C++】（六）MySql数据库

前面已经介绍了网络模块和业务模块，本章介绍数据模块，同样保持模块解耦的特性，即业务模块不能出现数据模块内容，如出现SQL语句，接下来看看怎么实现的。

1.环境安装

第一章已经介绍了MySql安装，但注意需要安装MySql开发包，如果没安装则不能调用MySqlAPI，可以使用以下命令检查：

sudo find /usr -name libmysqlclient*

查询结果如下：
在这里插入图片描述

2.数据模块代码

在include/server创建db文件夹，再在改文件下创建db.h头文件，用来链接、操作（增删改查）数据库。

#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;//和mysql的一条连接 
};#endif

在src/server/创建db文件夹，在改文件夹下创建db.cpp，代码作用卸载注释了。

#include "db.h"
#include <muduo/base/Logging.h>  //使用了muduo的日志打印代码// 数据库配置信息
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!";  //muduo日志打印}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;
}

其中更新和查询都使用mysql_query()函数，区别在于返回值。

3.更新各目录下CMakeLists.txt内容

项目目录下的CMakeLists.txt更新为：

cmake_minimum_required(VERSION 3.0)
project(chat)# 配置编译选项
set(CMAKE_CXX_FLAGS ${CMAKE_CXX_FLAGS} -g)# 配置最终的可执行文件输出的路径
set(EXECUTABLE_OUTPUT_PATH ${PROJECT_SOURCE_DIR}/bin)# 配置头文件的搜索路径
include_directories(${PROJECT_SOURCE_DIR}/include)
include_directories(${PROJECT_SOURCE_DIR}/include/server)
include_directories(${PROJECT_SOURCE_DIR}/include/server/db) //新添db加头文件路径
include_directories(${PROJECT_SOURCE_DIR}/thirdparty)# 加载子目录
add_subdirectory(src)

src/server/CMakeLists.txt更新为：

# 定义了一个SRC_LIST变量，包含了该目录下所有的源文件
aux_source_directory(. SRC_LIST)
aux_source_directory(./db DB_LIST)  //添加db的源文件# 指定生成可执行文件
add_executable(ChatServer ${SRC_LIST} ${DB_LIST}) //添加db的源文件
# 指定可执行文件链接时需要依赖的库文件
target_link_libraries(ChatServer muduo_net muduo_base mysqlclient hiredis pthread) //添加mysqlclient库文件