Merge branch 'master' of ssh://115.28.86.8:29418/~admin/昆仑_1025

lzx

2024-11-05 6cc341355e736952329b328232c2560dc6f29060

 Server/ÍõçûÔª/code/MysqlConn.cpp

@@ -1,174 +1,144 @@
#include "stdafx.h"
#include "MysqlConn.h"
#include <regex>

MysqlConn::MysqlConn()//初始化数据库
{
   m_result = nullptr;
   m_mysqlRow = nullptr;
   // ä¼ å…¥nullptr空指针时，会自动分配一个MYSQL对象
   m_conn = mysql_init(nullptr);
}
MysqlConn::~MysqlConn()//释放数据库连接
{
   freeRes(); // é‡Šæ”¾ç»“果集
   if (m_conn != nullptr) {
      mysql_close(m_conn);
      m_conn = nullptr;
   }
}
void MysqlConn::freeRes()//结果集释放
{
   if (m_result) {
      mysql_free_result(m_result);
      m_result = nullptr;
   }
}
bool MysqlConn::connect(const string user, const string passwd, \
   const string dbName, string ip, \
   const unsigned short port)//连接数据库
{
   MYSQL* res = mysql_real_connect(m_conn, ip.c_str(), user.c_str(), \
      passwd.c_str(), dbName.c_str(), port, nullptr, 0);
   // ä¿®æ”¹ç¼–码
   mysql_set_character_set(m_conn, "gb2312");
   return res != nullptr;
}
bool MysqlConn::update(const string sql) const
{
   // æ‰§è¡ŒæˆåŠŸè¿”回0；
   int res = mysql_real_query(m_conn, sql.c_str(), static_cast<unsigned int>(sql.size()));
   if (res != 0) {
      return false; // æç¤º
   }
   return true;
}
bool MysqlConn::query(const string sql)
{
   freeRes();
   int res = mysql_real_query(m_conn, sql.c_str(), static_cast<unsigned int>(sql.size()));
   if (res != 0) {
      return false; // æç¤º
   }
   m_result = mysql_store_result(m_conn);
   return true;
}
bool MysqlConn::getRes()
{
   if (m_result != nullptr) {
      // char** èŽ·å–单行记录
      m_mysqlRow = mysql_fetch_row(m_result);
      if (m_mysqlRow != nullptr) {
         return true;
      }
      freeRes();
   }
   return false;
}
string MysqlConn::getValue(const int fieldIndex) const
{
   int fieldCount = mysql_num_fields(m_result);
   if (fieldIndex >= fieldCount || fieldIndex < 0) {
      return string(); // æç¤º
   }
   char* value = m_mysqlRow[fieldIndex];
   // å¾—到一个保存各字段值长度的数组
   unsigned long* len = mysql_fetch_lengths(m_result);
   unsigned long length = len[fieldIndex];
   // é˜²æ­¢ç»“果中存在\0导致数据丢失
   return string(value, length);
}
bool MysqlConn::selectDB(const string dbName) const
{
   int res = mysql_select_db(m_conn, dbName.c_str());
   if (res != 0) {
      return false;  // æç¤º
   }
   return true;
}
void MysqlConn::backupCurrentDB(const string path)
{
   string sql = "show tables";
   int r = mysql_real_query(m_conn, sql.c_str(), static_cast<unsigned long>(sql.size()));
   if (r != 0) {
      return; // æç¤º
   }
   MYSQL_RES* tableRes = mysql_store_result(m_conn);
   for (int i = 0; i < mysql_num_rows(tableRes); ++i) {
      MYSQL_ROW tableName = mysql_fetch_row(tableRes);
      backupCurrentTable(path, tableName[0]);
   }
}
void MysqlConn::backupCurrentTable(const string path, const string tableName)
{
   string file = path + tableName + ".sql";
   ofstream ofs(file);
   if (!ofs.is_open()) {
      return; // æç¤º
   }
   // è¡¨ç»“构写入
   string showCreate = "show create table " + tableName + ";";
   bool res = query(showCreate);
   if (!res) {
      return; // æç¤º
   }
   if (getRes()) {
      string writeSQL = getValue(1) + ";\n";
      ofs.write(writeSQL.c_str(), writeSQL.size());
      // cout << writeSQL << endl;
   }
   // è¡¨æ•°æ®å†™å…¥
   string sql = "select * from " + tableName + ";";
   res = query(sql);
   if (!res) {
      return; // æç¤º
   }
   while (getRes()) {
      string writeSQL = "insert into `" + tableName + "` values(";
      for (int i = 0; !getValue(i).empty(); ++i) {
         if (i != 0) {
            writeSQL += ",";
         }
         MYSQL_FIELD* valueType = mysql_fetch_field_direct(m_result, i);
         if (valueType->type == MYSQL_TYPE_DECIMAL
            || valueType->type == MYSQL_TYPE_TINY
            || valueType->type == MYSQL_TYPE_SHORT
            || valueType->type == MYSQL_TYPE_LONG
            || valueType->type == MYSQL_TYPE_FLOAT
            || valueType->type == MYSQL_TYPE_DOUBLE
            || valueType->type == MYSQL_TYPE_TIMESTAMP
            || valueType->type == MYSQL_TYPE_LONGLONG
            || valueType->type == MYSQL_TYPE_INT24) {
            writeSQL += getValue(i);
         }
         else {
            writeSQL += "'" + getValue(i) + "'";
         }
      }
      writeSQL += ");\n";
      ofs.write(writeSQL.c_str(), writeSQL.size());
   }
   ofs.close();
}
bool MysqlConn::transaction() const
{
   // å°†äº‹åŠ¡æäº¤è®¾ç½®ä¸ºæ‰‹åŠ¨æäº¤
   return mysql_autocommit(m_conn, false);
}
bool MysqlConn::commit() const
{
   return mysql_commit(m_conn);
}
bool MysqlConn::rollback() const
{
   return mysql_rollback(m_conn);
}
void MysqlConn::refreashAliveTime()
{
   m_alivetime = steady_clock::now();
}
ll MysqlConn::getAliveTime()
{
   // æ¯«ç§’ <<= çº³ç§’:精度降低
   milliseconds res = duration_cast<milliseconds>(steady_clock::now() - m_alivetime);
   return res.count();

// åˆå§‹åŒ–连接
MysqlConn::MysqlConn() {
    mysql_ = mysql_init(mysql_);
    // è®¾ç½®å­—符集
    if (mysql_) mysql_set_character_set(mysql_, "gbk");
}

// è¿žæŽ¥æ•°æ®åº“
bool MysqlConn::connect(std::string ip, std::string userName, std::string passwd, std::string db, int port) {
    mysql_ = mysql_real_connect(mysql_, ip.c_str(), userName.c_str(), passwd.c_str(), db.c_str(), port, nullptr, 0);
    if (!mysql_) {
        return false;
    }
    return true;
}

// é‡Šæ”¾èµ„源
MysqlConn::~MysqlConn() {
    if (mysql_) {
        mysql_close(mysql_);
        mysql_ = nullptr;
    }
    freeRes();
}

// æ›´æ–°æ•°æ®
bool MysqlConn::update(std::string sql) {
    // å‚数化查询优化后的安全检查
    MYSQL_STMT* stmt = mysql_stmt_init(mysql_);
    if (!stmt) {
        return false;
    }
    if (mysql_stmt_prepare(stmt, sql.c_str(), sql.length())) {
        mysql_stmt_close(stmt);
        return false;
    }
    int paramCount = mysql_stmt_param_count(stmt);
    if (paramCount > 0) {
        // å¦‚果有参数，需要进行参数绑定等操作，这里暂不实现
        mysql_stmt_close(stmt);
        return false;
    }
    if (mysql_stmt_execute(stmt)) {
        mysql_stmt_close(stmt);
        return false;
    }
    mysql_stmt_close(stmt);
    return true;
}

// æŸ¥è¯¢æ•°æ®åº“
bool MysqlConn::query(std::string sql) {
    freeRes();
    // å‚数化查询优化后的安全检查
    MYSQL_STMT* stmt = mysql_stmt_init(mysql_);
    if (!stmt) {
        return false;
    }
    if (mysql_stmt_prepare(stmt, sql.c_str(), sql.length())) {
        mysql_stmt_close(stmt);
        return false;
    }
    int paramCount = mysql_stmt_param_count(stmt);
    if (paramCount > 0) {
        // å¦‚果有参数，需要进行参数绑定等操作，这里暂不实现
        mysql_stmt_close(stmt);
        return false;
    }
    if (mysql_stmt_execute(stmt)) {
        mysql_stmt_close(stmt);
        return false;
    }
    res_ = mysql_stmt_result_metadata(stmt);
    if (!res_) {
        mysql_stmt_close(stmt);
        return false;
    }
    mysql_stmt_close(stmt);
    return true;
}

// å¾—到结果集
bool MysqlConn::getResult() {
    if (res_) {
        row_ = mysql_fetch_row(res_);
        if (row_) return true;
    }
    return false;
}

// èŽ·å–结果集的字段
std::string MysqlConn::getField(int index) {
    int cols = mysql_num_fields(res_);
    if (index >= cols || index < 0) return std::string("");

    char* value = row_[index];
    unsigned long len = mysql_fetch_lengths(res_)[index];
    return std::string(value, len);
}

// äº‹åŠ¡æ“ä½œ
bool MysqlConn::transaction() {
    return mysql_autocommit(mysql_, false);
}

// æäº¤äº‹åŠ¡
bool MysqlConn::commit() {
    return mysql_commit(mysql_);
}

// äº‹åŠ¡å›žæ»š
bool MysqlConn::rollback() {
    return mysql_rollback(mysql_);
}

void MysqlConn::refreshActiveTime() {
    activeTime_ = std::chrono::steady_clock::now();
}

long long MysqlConn::getActiveTime() {
    // çº³ç±³
    std::chrono::nanoseconds nased = std::chrono::steady_clock::now() - activeTime_;
    // è½¬æ¢æˆæ¯«ç±³
    std::chrono::microseconds millsed = std::chrono::duration_cast<std::chrono::microseconds>(nased);
    return millsed.count();   // å¤šå°‘毫秒
}

// å®‰å…¨æ ¡éªŒå®žçŽ°ï¼Œè¿™é‡Œç®€å•ä½¿ç”¨æ­£åˆ™è¡¨è¾¾å¼åˆ¤æ–­æ˜¯å¦åŒ…含危险字符
bool MysqlConn::isSqlSafe(const std::string& sql) {
    std::regex dangerousPattern(".*(['\";\\-+=]).*");
    return!std::regex_search(sql, dangerousPattern);
}

void MysqlConn::freeRes() {
    if (res_) {
        mysql_free_result(res_);
        res_ = nullptr;
    }
}