2021-06-16 12:49:58 +02:00

249 lines
7.6 KiB
C++

#include "ModelBase.h"
#include "sodium.h"
#include "../../ServerConfig.h"
#include "../../SingletonManager/ConnectionManager.h"
#include "Poco/URI.h"
namespace model {
namespace table {
ModelInsertTask::ModelInsertTask(Poco::AutoPtr<ModelBase> model, bool loadId, bool emailErrors /* = false */)
: UniLib::controller::CPUTask(ServerConfig::g_CPUScheduler), mModel(model), mEmailErrors(emailErrors), mLoadId(loadId)
{
#ifdef _UNI_LIB_DEBUG
setName(model->getTableName());
#endif
}
int ModelInsertTask::run()
{
auto result = mModel->insertIntoDB(mLoadId);
if (mModel->errorCount() > 0 && mEmailErrors) {
mModel->sendErrorsAsEmail();
}
return !result;
//return 0;
}
// ---------------------------------------------------------------------------------------------------
ModelBase::~ModelBase()
{
lock("~ModelBase");
assert(0 == mReferenceCount);
unlock();
}
bool ModelBase::insertIntoDB(bool loadId)
{
//printf("ModelBase::insertIntoDB with table: %s\n", getTableName());
auto cm = ConnectionManager::getInstance();
Poco::ScopedLock<Poco::Mutex> _poco_lock(mWorkMutex);
UNIQUE_LOCK;
auto session = cm->getConnection(CONNECTION_MYSQL_LOGIN_SERVER);
Poco::Data::Statement insert = _insertIntoDB(session);
size_t resultCount = 0;
try {
if (insert.execute(true) == 1) {
// load id from db
if (loadId) {
Poco::Data::Statement select = _loadIdFromDB(session);
try {
auto res = select.execute();
if (1 == res) { return true; }
}
catch (Poco::Exception& ex) {
addError(new ParamError(getTableName(), "mysql error by select id", ex.displayText().data()));
addError(new ParamError(getTableName(), "data set: ", toString().data()));
}
session = cm->getConnection(CONNECTION_MYSQL_LOGIN_SERVER);
select = _loadIdFromDB(session);
//Poco::Data::Statement select = _loadIdFromDB(session);
return 1 == select.execute();
}
else {
return true;
}
}
}
catch (Poco::Exception& ex) {
addError(new ParamError(getTableName(), "mysql error by insert", ex.displayText().data()));
addError(new ParamError(getTableName(), "data set: \n", toString().data()));
}
//printf("data valid: %s\n", toString().data());
return false;
}
bool ModelBase::isExistInDB()
{
auto cm = ConnectionManager::getInstance();
Poco::ScopedLock<Poco::Mutex> _poco_lock(mWorkMutex);
UNIQUE_LOCK;
auto session = cm->getConnection(CONNECTION_MYSQL_LOGIN_SERVER);
Poco::Data::Statement select = _loadIdFromDB(session);
try {
select.executeAsync();
return select.wait() == 1;
}
catch (Poco::Exception& ex) {
addError(new ParamError(getTableName(), "mysql error by select id, check if exist in db", ex.displayText().data()));
addError(new ParamError(getTableName(), "data set: ", toString().data()));
}
return false;
}
bool ModelBase::deleteFromDB()
{
Poco::ScopedLock<Poco::Mutex> _poco_lock(mWorkMutex);
UNIQUE_LOCK;
if (mID == 0) {
addError(new Error(getTableName(), "id is zero, couldn't delete from db"));
return false;
}
auto cm = ConnectionManager::getInstance();
auto session = cm->getConnection(CONNECTION_MYSQL_LOGIN_SERVER);
Poco::Data::Statement deleteStmt(session);
deleteStmt << "delete from " << getTableName() << " where id = ?", Poco::Data::Keywords::use(mID);
try {
return deleteStmt.execute() == 1;
}
catch (Poco::Exception& ex) {
addError(new ParamError(getTableName(), "mysql error by delete", ex.displayText().data()));
addError(new ParamError(getTableName(), "id: ", mID));
}
return false;
}
void ModelBase::duplicate()
{
//Poco::ScopedLock<Poco::Mutex> _lock(mWorkMutex);
std::string stack_details = "[ModelBase::duplicate] table: ";
stack_details += getTableName();
lock(stack_details.data());
mReferenceCount++;
printf("[ModelBase::duplicate] new value: %d, table name: %s\n", mReferenceCount, getTableName());
unlock();
//printf("[ModelBase::duplicate] new value: %d\n", mReferenceCount);
}
void ModelBase::release()
{
if(mReferenceCount <= 0) {
throw Poco::Exception("ModelBase already released", getTableName());
}
std::string stack_details = "[ModelBase::release] table: ";
stack_details += getTableName();
stack_details += ", reference count: ";
stack_details += std::to_string(mReferenceCount);
lock(stack_details.data());
mReferenceCount--;
printf("[ModelBase::release] new value: %d, table name: %s\n", mReferenceCount, getTableName());
if (0 == mReferenceCount) {
unlock();
delete this;
return;
}
unlock();
}
Poco::Data::Statement ModelBase::_loadFromDB(Poco::Data::Session session, const std::vector<std::string>& fieldNames, MysqlConditionType conditionType/* = MYSQL_CONDITION_AND*/)
{
std::string message = getTableName();
message += "::_loadFromDB with multiple fields not implemented";
throw Poco::Exception(message);
}
Poco::Data::Statement ModelBase::_loadMultipleFromDB(Poco::Data::Session session, const std::string& fieldName)
{
std::string message = getTableName();
message += "::_loadMultipleFromDB not implemented";
throw Poco::Exception(message);
}
Poco::Data::Statement ModelBase::_loadMultipleFromDB(Poco::Data::Session session, const std::vector<std::string> fieldNames, MysqlConditionType conditionType/* = MYSQL_CONDITION_AND*/)
{
std::string message = getTableName();
message += "::_loadMultipleFromDB multi not implemented";
throw Poco::Exception(message);
}
Poco::Data::Statement ModelBase::_loadAllFromDB(Poco::Data::Session session)
{
std::string message = getTableName();
message += "::_loadAllFromDB not implemented";
throw Poco::Exception(message);
}
Poco::DateTime ModelBase::parseElopageDate(std::string dateString)
{
std::string decodedDateString = "";
Poco::URI::decode(dateString, decodedDateString);
struct tm * parsedTime;
// used because localtime return an internal pointer, not thread safe
Poco::Mutex& timeMutex = ServerConfig::g_TimeMutex;
int year, month, day, hour, minute, second;
// ex: 2009-10-29
if (sscanf(decodedDateString.data(), "%d-%d-%dT%d:%dZ", &year, &month, &day, &hour, &minute) != EOF) {
time_t rawTime;
time(&rawTime);
// static, used for every thread
timeMutex.lock();
parsedTime = localtime(&rawTime);
// tm_year is years since 1900
parsedTime->tm_year = year - 1900;
// tm_months is months since january
parsedTime->tm_mon = month - 1;
parsedTime->tm_mday = day;
parsedTime->tm_hour = hour;
parsedTime->tm_min = minute;
parsedTime->tm_sec = 0;
rawTime = mktime(parsedTime);
timeMutex.unlock();
// rawTime is in seconds, poco timestamp in microseconds
return Poco::DateTime(Poco::Timestamp(rawTime * 1000000));
}
return Poco::DateTime(Poco::Timestamp());
}
std::string ModelBase::secondsToReadableDuration(Poco::UInt64 seconds)
{
Poco::UInt64 value = 0;
std::string result;
std::string unit_name;
if (seconds <= 60) {
return std::to_string(seconds) + " seconds";
}
else if (seconds <= 60 * 60) {
Poco::UInt64 minutes = round(seconds / 60);
return "~" + std::to_string(minutes) + " minutes";
}
else if (seconds <= 60 * 60 * 24) {
Poco::UInt64 hours = round(seconds / 60 / 60);
return "~" + std::to_string(hours) + " hours";
}
else {
Poco::UInt64 days = round(seconds / 60 / 60 / 24);
return "~" + std::to_string(days) + " days";
}
return "<unknown error>";
}
}
}