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

数据库模块

db.hpp

首先，在 include/server 目录下新增文件夹 db，用于存放数据库相关文件。

在 db 文件夹下新增文件：db.hpp

#ifndef DB_H_ #define DB_H_ #include #include #include using namespace std; // 数据库配置信息 static string server = "127.0.0.1"; static string user = "root"; static string password = "123456"; static string dbname = "chat"; // 数据库操作类 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" 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); } MYSQL* MySQL::getconnection(){ return _conn; }

用户通用模块

user.hpp

封装数据。并提供数据的访问。

#ifndef USER_H_ #define USER_H_ #include using namespace std; //User表的映射类 class User{ private: int id; string name; string password; string state; public: User(int id = -1,string name = "",string pwd = "",string state = "offline"){ this->id = id; this->name = name; this->password = pwd; this->state = state; } void setID(int id){ this->id = id; } void setname(string name){ this->name = name; } void setpassword(string password){ this->password = password; } void setstate(string state){ this->state = state; } int getID(){ return this->id; } string getname(){ return this->name; } string getpassword(){ return this->password; } string getstate(){ return this->state; } }; #endif

东西不多，直接写这里。

usermodel.hpp

如果说，上面那个文件是 ==从数据库到用户层的映射==，那么这组文件就是 ==业务层对用户业务的操作==

#ifndef USERMODEL_H_ #define USERMODEL_H_ #include "user.hpp" //User表的操作类 class UserModel{ public: bool insert(User &user); User query(int id); bool updateState(User user); private: }; #endif

usermodel.cpp

#include "usermodel.hpp" #include "db.hpp" #include using namespace std; bool UserModel::insert(User &user){ char sql[128] = {0}; sprintf("insert into User(name,password,state) values('%s','%s','%s')", user.getname().c_str(),user.getpassword().c_str(),user.getstate().c_str()); MySQL mysql; if(mysql.connect()){ if(mysql.update(sql)){ user.setID(mysql_insert_id(mysql.getconnection())); return true; } } return false; } // 根据用户号码查询用户信息 User UserModel::query(int id) { // 1.组装sql语句 char sql[1024] = {0}; sprintf(sql, "select * from User where id = %d", id); MySQL mysql; if (mysql.connect()) { MYSQL_RES *res = mysql.query(sql); if (res != nullptr) { MYSQL_ROW row = mysql_fetch_row(res); if (row != nullptr) { User user; user.setID(atoi(row[0])); user.setname(row[1]); user.setpassword(row[2]); user.setstate(row[3]); mysql_free_result(res); return user; } } } return User(); } bool UserModel::updateState(User user){ char sql[128] = {0}; sprintf("updata User set state = '%s' where id = '%d'",user.getstate().c_str(),user.getID()); MySQL mysql; if(mysql.connect()){ if(mysql.update(sql)){ return true; } } return false; }

cmake 文件修改

最外层的 cmake 文件需要适当的做出调整。

加上下面这条

include_directories(${PROJECT_SOURCE_DIR}/include/server/db)

server 里面的 cmake 也要适当的做调整：

# 定义一个SRC_LIST变量，存放该目录下所有的源文件 aux_source_directory(. SRC_LIST) aux_source_directory(./db SRC_LIST) # 指定生成可执行文件 add_executable(ChatServer ${SRC_LIST}) # 指定可执行文件生成时需要链接的外部库 target_link_libraries(ChatServer muduo_net muduo_base pthread mysqlclient)

加一个源文件的路径，加一个外库。

编译遇到问题请直接转文档 问题解决 专栏，这次的问题很难找到适配的资料，需要靠经验嘿嘿。

C++ 数据库

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。