1

wangky

2024-11-04 4b032971d376b063480e53842045561f7c9b399b

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

@@ -1,174 +1,115 @@
#include "MysqlConn.h"
#include <regex>
// åˆå§‹åŒ–连接
MysqlConn::MysqlConn() {
   mysql_ = mysql_init( mysql_ );
   // è®¾ç½®å­—符集
   if( mysql_ )   mysql_set_character_set(mysql_ , "gbk");
}

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; // æç¤º
// è¿žæŽ¥æ•°æ®åº“
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;
}
bool MysqlConn::query(const string sql)
{

// é‡Šæ”¾èµ„源
MysqlConn::~MysqlConn() {
   if (mysql_) {
      mysql_close(mysql_);
      mysql_ = nullptr;
   }
   freeRes();
   int res = mysql_real_query(m_conn, sql.c_str(), static_cast<unsigned int>(sql.size()));
   if (res != 0) {
      return false; // æç¤º
}
// æ›´æ–°æ•°æ®
bool MysqlConn::update( std::string sql ) {
   if (!isSqlSafe(sql))
   {
      return false;
   }
   m_result = mysql_store_result(m_conn);
   int ret = mysql_query( mysql_ , sql.c_str());
   if( ret != 0 ){
      return false;
   }
   return true;
}
bool MysqlConn::getRes()
{
   if (m_result != nullptr) {
      // char** èŽ·å–单行记录
      m_mysqlRow = mysql_fetch_row(m_result);
      if (m_mysqlRow != nullptr) {
         return true;
      }
      freeRes();

// æŸ¥è¯¢æ•°æ®åº“
bool MysqlConn::query(std::string sql) {
   freeRes( );
   if (!isSqlSafe(sql))
   {
      return false;
   }
   int ret = mysql_query(mysql_, sql.c_str());
   if (ret != 0)   return false;
	
   // èŽ·å–查询结果
   res_ = mysql_store_result(mysql_);
   if (!res_)   return false;
   return true;
}

// å¾—到结果集
bool MysqlConn::getResult() {
   if (res_) {
      row_ = mysql_fetch_row(res_);
      if(row_)  return true;
   }
   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();

// èŽ·å–结果集的字段
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;
   }
}