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

同样，在 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)){

最后

自我介绍一下，小编13年上海交大毕业，曾经在小公司待过，也去过华为、OPPO等大厂，18年进入阿里一直到现在。

深知大多数Java工程师，想要提升技能，往往是自己摸索成长，自己不成体系的自学效果低效漫长且无助。

因此收集整理了一份《2024年Java开发全套学习资料》，初衷也很简单，就是希望能够帮助到想自学提升又不知道该从何学起的朋友，同时减轻大家的负担。

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，基本涵盖了95%以上Java开发知识点，不论你是刚入门Java开发的新手，还是希望在技术上不断提升的资深开发者，这些资料都将为你打开新的学习之门！

如果你觉得这些内容对你有帮助，需要这份全套学习资料的朋友可以戳我获取！！

由于文件比较大，这里只是将部分目录截图出来，每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频，并且会持续更新！
衷也很简单，就是希望能够帮助到想自学提升又不知道该从何学起的朋友，同时减轻大家的负担。**

[外链图片转存中…(img-e7fI3Uzg-1715619495593)]

[外链图片转存中…(img-aa4wEBXJ-1715619495594)]

[外链图片转存中…(img-cz0uZfiw-1715619495594)]

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，基本涵盖了95%以上Java开发知识点，不论你是刚入门Java开发的新手，还是希望在技术上不断提升的资深开发者，这些资料都将为你打开新的学习之门！

如果你觉得这些内容对你有帮助，需要这份全套学习资料的朋友可以戳我获取！！

由于文件比较大，这里只是将部分目录截图出来，每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频，并且会持续更新！