chat集群聊天室项目 代码+讲解(三):映射层 + 持久层

举报
看,未来 发表于 2021/09/14 21:14:37 2021/09/14
【摘要】 @[toc] 项目简单架构图 db代码#pragma#include<string>#include<mysql/mysql.h>#include<muduo/base/Logging.h>using namespace std;// 数据库配置信息 static string server = "127.0.0.1"; static string user = "root"; static...

在这里插入图片描述

@[toc]

项目简单架构图

在这里插入图片描述

db代码

#pragma

#include<string>
#include<mysql/mysql.h>
#include<muduo/base/Logging.h>

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

讲解

这里为什么不用单例?

第一次在项目中用MySQL做数据库,有点紧张,就没弄单例了,怕出意外。


代码

#ifndef FRIENDMODEL_H_
#define FRIENDMODEL_H_

#include<vector>
#include"user.hpp"

using namespace std;

//好友信息方法
class FriendModel{
public:
    void insert(int userid,int friendid);

    vector<User> query(int userid);

private:

};

#endif
#include"friendmodel.hpp"
#include"db.hpp"

void FriendModel::insert(int userid,int friendid){
    char sql[128] = {0};

    sprintf(sql,"insert into Friend(userid,friendid) values(%d,%d)",userid,friendid);

    MySQL mysql;
    if(mysql.connect()){
        mysql.update(sql);
    }
}

vector<User> FriendModel::query(int userid){
    // 1.组装sql语句
    char sql[256] = {0};
    sprintf(sql, "select a.id,a.name,a.state from User a inner join Friend b on b.friendid = a.id where b.userid = %d", userid);

    vector<User> vec;

    MySQL mysql;
    if (mysql.connect())
    {
        MYSQL_RES *res = mysql.query(sql);
        
        if (res != nullptr)
        {
            MYSQL_ROW row;
            while((row = mysql_fetch_row(res)) != nullptr){
                User user;
                user.setID(atoi(row[0]));
                user.setname(row[1]);
                user.setstate(row[2]);

                vec.push_back(user);
            }
            mysql_free_result(res);
        }
    }

    return vec;
}

代码讲解

数据库操作的时候为什么不上锁?

这些映射层的代码和业务层一样的千篇一律,我就挑个短的来就好了。

其实当时弄完就挺后悔的,整个单例多好。

这里面,为什么对数据库操作的时候不上锁啊?

以前上锁,是因为以前连接的数据库是 sqlite3,MySQL是没有自带锁吗?要我操心这个。

【版权声明】本文为华为云社区用户原创内容,转载时必须标注文章的来源(华为云社区)、文章链接、文章作者等基本信息, 否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。