jsyhsec/energy_storage
7
0
Fork 0
mirror of https://gitee.com/js-yhsec/energy_storage.git synced 2026-05-27 18:59:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

实现HTTP服务架构

Browse Source
This commit is contained in:
lixiaoyuan
2025-08-31 14:38:53 +08:00
parent 4af4e670d2
commit e0b64a20c4
46 changed files with 1436 additions and 545 deletions
Download Patch File Download Diff File Expand all files Collapse all files

116
src/database/MysqlClient.cpp
View File

@@ -2,6 +2,13 @@
#include "common/Utils.h"
//#include "Spdlogger.h"
#include "Logger.h"
#include <chrono>
using namespace std;
static int64_t GetTimestamp()
{
return chrono::duration_cast<chrono::seconds>(chrono::system_clock::now().time_since_epoch()).count();
}
MysqlClient::MysqlClient(MysqlOption option) : option(option)
{
@@ -13,22 +20,35 @@ MysqlClient::~MysqlClient()
this->close();
}
static int64_t g_tickErr {0};
int MysqlClient::conn()
{
if (mysql_)
{
return 0;
}
if (GetTimestamp() - g_tickErr <= 5)
{
return 1;
}
mysql_ = mysql_init(nullptr);
MYSQL* ret = mysql_real_connect(mysql_, option.host.c_str(), option.user.c_str(), option.password.c_str(), option.dbname.c_str(), option.port, NULL, 0);
MYSQL* ret = mysql_real_connect(mysql_,
option.host.c_str(),
option.user.c_str(),
option.password.c_str(),
option.dbname.c_str(),
option.port, NULL, 0);
if (ret == NULL)
{
std::string err = mysql_error(mysql_);
//Spdlogger::info("[mysql] connect failed: {}", mysql_error(mysql_));
mysql_ = nullptr;
g_tickErr = GetTimestamp();
}
else
{
g_tickErr = 0;
mysql_query(mysql_, "set names 'utf8';");
}
return 0;
@@ -48,70 +68,80 @@ void MysqlClient::close()
}
}
bool MysqlClient::exec(std::string sql)
static int MysqlQuery(MYSQL* mysql, const std::string& sql)
{
if (!mysql_)
int err = 0;
if (!mysql)
{
XLOGE() << "Mysql exec error, database is not connected.";
return false;
return err;
}
if (sql.empty())
{
XLOGE() << "Mysql exec error, sql is empty.";
return false;
return err;
}
int ret = mysql_query(mysql_, sql.c_str());
if (0 != ret)
err = mysql_query(mysql, sql.c_str());
if (0 != err)
{
XLOGE() << "Mysql exec error: " << mysql_error(mysql_) << ", sql=" << sql;
return false;
err = mysql_errno(mysql);
XLOGE() << "Mysql exec error: " << err << "," << mysql_error(mysql) << ", sql=" << sql;
return err;
}
return true;
return err;
}
bool MysqlClient::exec(std::string sql, vector<Fields>& result)
int MysqlClient::exec(std::string sql)
{
int err = MysqlQuery(mysql_, sql);
// 确保读取并释放结果集，否则会产生 [2014,Commands out of sync;] 错误
MYSQL_RES* res = mysql_store_result(mysql_);
if (res) { mysql_free_result(res); }
return err;
}
int MysqlClient::exec(std::string sql, vector<Fields>& result)
{
result.clear();
bool ret = MysqlClient::exec(sql);
if (!ret)
int err = MysqlQuery(mysql_, sql);
if (err != 0)
{
return false;
return err;
}
MYSQL_RES* res = mysql_store_result(mysql_);
if (!res)
if (res)
{
return false;
}
vector<string> fieldNames;
while (true)
{
MYSQL_FIELD* field = mysql_fetch_field(res);
if (!field)
vector<string> fieldNames;
while (true)
{
break;
}
fieldNames.push_back(field->name);
}
while (true)
{
MYSQL_ROW row = mysql_fetch_row(res);
if (!row)
{
break;
MYSQL_FIELD* field = mysql_fetch_field(res);
if (!field)
{
break;
}
fieldNames.push_back(field->name);
}
Fields rowData;
for (size_t i = 0; i < fieldNames.size(); ++i)
while (true)
{
string field_text = (row[i] == NULL) ? "" : row[i];
rowData.set(fieldNames[i], field_text);
}
result.push_back(rowData);
}
MYSQL_ROW row = mysql_fetch_row(res);
if (!row)
{
break;
}
// 释放结果集
mysql_free_result(res);
return true;
Fields rowData;
for (size_t i = 0; i < fieldNames.size(); ++i)
{
string field_text = (row[i] == NULL) ? "" : row[i];
rowData.set(fieldNames[i], field_text);
}
result.push_back(rowData);
}
// 释放结果集
mysql_free_result(res);
}
return 0;
}