X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/21baffc9c880ba71baaddad5c4cf2b32352823ff..1cb912e4f90473f78d2f0f10ab5c9b2c5bd36440:/projects/stargazer/plugins/store/firebird/firebird_store_users.cpp diff --git a/projects/stargazer/plugins/store/firebird/firebird_store_users.cpp b/projects/stargazer/plugins/store/firebird/firebird_store_users.cpp index fbeacae5..ba21aa61 100644 --- a/projects/stargazer/plugins/store/firebird/firebird_store_users.cpp +++ b/projects/stargazer/plugins/store/firebird/firebird_store_users.cpp @@ -26,9 +26,15 @@ * */ -#include "stg/const.h" #include "firebird_store.h" + #include "stg/ibpp.h" +#include "stg/user_conf.h" +#include "stg/user_stat.h" +#include "stg/user_traff.h" +#include "stg/user_ips.h" +#include "stg/const.h" +#include "stg/common.h" //----------------------------------------------------------------------------- int FIREBIRD_STORE::GetUsersList(std::vector * usersList) const @@ -38,14 +44,13 @@ STG_LOCKER lock(&mutex); IBPP::Transaction tr = IBPP::TransactionFactory(db, IBPP::amRead, til, tlr); IBPP::Statement st = IBPP::StatementFactory(db, tr); -std::string name; - try { tr->Start(); st->Execute("select name from tb_users"); while (st->Fetch()) { + std::string name; st->Get(1, name); usersList->push_back(name); } @@ -118,7 +123,7 @@ catch (IBPP::Exception & ex) return 0; } //----------------------------------------------------------------------------- -int FIREBIRD_STORE::SaveUserStat(const USER_STAT & stat, +int FIREBIRD_STORE::SaveUserStat(const STG::UserStat & stat, const std::string & login) const { STG_LOCKER lock(&mutex); @@ -126,7 +131,7 @@ STG_LOCKER lock(&mutex); return SaveStat(stat, login); } //----------------------------------------------------------------------------- -int FIREBIRD_STORE::SaveStat(const USER_STAT & stat, +int FIREBIRD_STORE::SaveStat(const STG::UserStat & stat, const std::string & login, int year, int month) const @@ -223,7 +228,7 @@ catch (IBPP::Exception & ex) return 0; } //----------------------------------------------------------------------------- -int FIREBIRD_STORE::SaveUserConf(const USER_CONF & conf, +int FIREBIRD_STORE::SaveUserConf(const STG::UserConf & conf, const std::string & login) const { STG_LOCKER lock(&mutex); @@ -331,11 +336,11 @@ try st->Execute(); st->Prepare("insert into tb_allowed_ip (fk_user, ip, mask) values (?, ?, ?)"); - for(size_t i = 0; i < conf.ips.Count(); i++) + for(size_t j = 0; j < conf.ips.count(); j++) { st->Set(1, uid); - st->Set(2, (int32_t)conf.ips[i].ip); - st->Set(3, (int32_t)conf.ips[i].mask); + st->Set(2, (int32_t)conf.ips[j].ip); + st->Set(3, (int32_t)conf.ips[j].mask); st->Execute(); } tr->Commit(); @@ -351,7 +356,7 @@ catch (IBPP::Exception & ex) return 0; } //----------------------------------------------------------------------------- -int FIREBIRD_STORE::RestoreUserStat(USER_STAT * stat, +int FIREBIRD_STORE::RestoreUserStat(STG::UserStat * stat, const std::string & login) const { STG_LOCKER lock(&mutex); @@ -438,7 +443,7 @@ catch (IBPP::Exception & ex) return 0; } //----------------------------------------------------------------------------- -int FIREBIRD_STORE::RestoreUserConf(USER_CONF * conf, +int FIREBIRD_STORE::RestoreUserConf(STG::UserConf * conf, const std::string & login) const { STG_LOCKER lock(&mutex); @@ -537,14 +542,15 @@ try where fk_user = ?"); st->Set(1, uid); st->Execute(); - conf->ips.Erase(); + STG::UserIPs ips; while (st->Fetch()) { - IP_MASK im; + STG::IPMask im; st->Get(1, (int32_t &)im.ip); st->Get(2, (int32_t &)im.mask); - conf->ips.Add(im); + ips.add(im); } + conf->ips = ips; tr->Commit(); } @@ -656,10 +662,10 @@ return 0; } //----------------------------------------------------------------------------- int FIREBIRD_STORE::WriteUserDisconnect(const std::string & login, - const DIR_TRAFF & up, - const DIR_TRAFF & down, - const DIR_TRAFF & sessionUp, - const DIR_TRAFF & sessionDown, + const STG::DirTraff & up, + const STG::DirTraff & down, + const STG::DirTraff & sessionUp, + const STG::DirTraff & sessionDown, double /*cash*/, double /*freeMb*/, const std::string & /*reason*/) const @@ -708,7 +714,7 @@ catch (IBPP::Exception & ex) return 0; } //----------------------------------------------------------------------------- -int FIREBIRD_STORE::WriteDetailedStat(const std::map & statTree, +int FIREBIRD_STORE::WriteDetailedStat(const STG::TraffStat & statTree, time_t lastStat, const std::string & login) const { @@ -725,7 +731,7 @@ time_t2ts(lastStat, &statTime); try { tr->Start(); - std::map::const_iterator it; + STG::TraffStat::const_iterator it; it = statTree.begin(); st->Prepare("insert into tb_detail_stats \ (till_time, from_time, fk_user, dir_num, \ @@ -760,7 +766,7 @@ catch (IBPP::Exception & ex) return 0; } //----------------------------------------------------------------------------- -int FIREBIRD_STORE::SaveMonthStat(const USER_STAT & stat, int month, int year, const std::string & login) const +int FIREBIRD_STORE::SaveMonthStat(const STG::UserStat & stat, int month, int year, const std::string & login) const { STG_LOCKER lock(&mutex);