expand/CCash
1
0
Fork 0
mirror of https://github.com/Expand-sys/CCash synced 2025-12-18 17:12:14 +11:00
Code Issues Projects Releases Packages Wiki Activity Actions

🎨 formatting

Browse source
This commit is contained in:
EntireTwix 2021-07-08 14:58:58 -07:00
parent 175976e796
commit 16494f971e
1 changed files with 64 additions and 50 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

42
src/bank.cpp
View file

@ -2,8 +2,6 @@
using namespace drogon; using namespace drogon;
#define INLINE __attribute__((always_inline)) inline
INLINE bool ValidUsrname(const std::string &name) noexcept INLINE bool ValidUsrname(const std::string &name) noexcept
{ {
if (name.size() < min_name_size || name.size() > max_name_size) if (name.size() < min_name_size || name.size() > max_name_size)
@ -68,14 +66,16 @@ bool Bank::GetChangeState() const noexcept
BankResponse Bank::GetBal(const std::string &name) const noexcept BankResponse Bank::GetBal(const std::string &name) const noexcept
{ {
static thread_local uint32_t res = 0; static thread_local uint32_t res = 0;
users.if_contains(name, [](const User &u) { res = u.balance + 1; }); users.if_contains(name, [](const User &u)
{ res = u.balance + 1; });
return res ? BankResponse(k200OK, std::to_string(res - 1)) : BankResponse(k404NotFound, "\"User not found\""); return res ? BankResponse(k200OK, std::to_string(res - 1)) : BankResponse(k404NotFound, "\"User not found\"");
} }
BankResponse Bank::GetLogs(const std::string &name) noexcept BankResponse Bank::GetLogs(const std::string &name) noexcept
{ {
#if MAX_LOG_SIZE > 0 #if MAX_LOG_SIZE > 0
BankResponse res; BankResponse res;
if (!users.modify_if(name, [&res](User &u) { res = BankResponse(k200OK, u.log.GetLog()); })) if (!users.modify_if(name, [&res](User &u)
{ res = BankResponse(k200OK, u.log.GetLog()); }))
{ {
return BankResponse(k404NotFound, "\"User not found\""); return BankResponse(k404NotFound, "\"User not found\"");
} }
@ -106,9 +106,11 @@ BankResponse Bank::SendFunds(const std::string &a_name, const std::string &b_nam
std::shared_lock<std::shared_mutex> lock{save_lock}; //about 10% of this function's cost std::shared_lock<std::shared_mutex> lock{save_lock}; //about 10% of this function's cost
#if MAX_LOG_SIZE > 0 #if MAX_LOG_SIZE > 0
static thread_local Transaction temp(a_name, b_name, amount); static thread_local Transaction temp(a_name, b_name, amount);
if (!users.modify_if(a_name, [amount](User &a) { if (!users.modify_if(a_name, [amount](User &a)
{
#else #else
if (!users.modify_if(a_name, [&state, amount](User &a) { if (!users.modify_if(a_name, [&state, amount](User &a)
{
#endif #endif
//if A can afford it //if A can afford it
if (a.balance < amount) if (a.balance < amount)
@ -130,12 +132,14 @@ BankResponse Bank::SendFunds(const std::string &a_name, const std::string &b_nam
if (state.first == k200OK) if (state.first == k200OK)
{ {
#if MAX_LOG_SIZE > 0 #if MAX_LOG_SIZE > 0
users.modify_if(b_name, [amount](User &b) { users.modify_if(b_name, [amount](User &b)
{
b.balance += amount; b.balance += amount;
b.log.AddTrans(std::move(temp)); b.log.AddTrans(std::move(temp));
}); //about 40% of this function's cost }); //about 40% of this function's cost
#else #else
users.modify_if(b_name, [amount](User &b) { b.balance += amount; }); users.modify_if(b_name, [amount](User &b)
{ b.balance += amount; });
#endif #endif
#if CONSERVATIVE_DISK_SAVE #if CONSERVATIVE_DISK_SAVE
#if MULTI_THREADED #if MULTI_THREADED
@ -150,13 +154,15 @@ BankResponse Bank::SendFunds(const std::string &a_name, const std::string &b_nam
bool Bank::VerifyPassword(const std::string &name, std::string_view &&attempt) const noexcept bool Bank::VerifyPassword(const std::string &name, std::string_view &&attempt) const noexcept
{ {
static thread_local bool res = false; static thread_local bool res = false;
users.if_contains(name, [&attempt](const User &u) { res = (u.password == xxHashStringGen{}(std::move(attempt))); }); users.if_contains(name, [&attempt](const User &u)
{ res = (u.password == xxHashStringGen{}(std::move(attempt))); });
return res; return res;
} }
void Bank::ChangePassword(const std::string &name, std::string &&new_pass) noexcept void Bank::ChangePassword(const std::string &name, std::string &&new_pass) noexcept
{ {
users.modify_if(name, [&new_pass](User &u) { u.password = xxHashStringGen{}(new_pass); }); users.modify_if(name, [&new_pass](User &u)
{ u.password = xxHashStringGen{}(new_pass); });
#if CONSERVATIVE_DISK_SAVE #if CONSERVATIVE_DISK_SAVE
#if MULTI_THREADED #if MULTI_THREADED
save_flag.SetChangesOn(); save_flag.SetChangesOn();
@ -167,7 +173,8 @@ void Bank::ChangePassword(const std::string &name, std::string &&new_pass) noexc
} }
BankResponse Bank::SetBal(const std::string &name, uint32_t amount) noexcept BankResponse Bank::SetBal(const std::string &name, uint32_t amount) noexcept
{ {
if (users.modify_if(name, [amount](User &u) { u.balance = amount; })) if (users.modify_if(name, [amount](User &u)
{ u.balance = amount; }))
{ {
#if CONSERVATIVE_DISK_SAVE #if CONSERVATIVE_DISK_SAVE
#if MULTI_THREADED #if MULTI_THREADED
@ -189,7 +196,8 @@ BankResponse Bank::ImpactBal(const std::string &name, int64_t amount) noexcept
{ {
return {k400BadRequest, "\"Amount cannot be 0\""}; return {k400BadRequest, "\"Amount cannot be 0\""};
} }
if (users.modify_if(name, [amount](User &u) { u.balance < (amount * -1) ? u.balance = 0 : u.balance += amount; })) if (users.modify_if(name, [amount](User &u)
{ u.balance < (amount * -1) ? u.balance = 0 : u.balance += amount; }))
{ {
#if CONSERVATIVE_DISK_SAVE #if CONSERVATIVE_DISK_SAVE
#if MULTI_THREADED #if MULTI_THREADED
@ -243,7 +251,9 @@ BankResponse Bank::DelUser(const std::string &name) noexcept
std::shared_lock<std::shared_mutex> lock{save_lock}; std::shared_lock<std::shared_mutex> lock{save_lock};
#if RETURN_ON_DEL #if RETURN_ON_DEL
uint32_t bal; uint32_t bal;
if (users.if_contains(name, [this, &bal](const User &u) { bal = u.balance; }) && bal) if (users.if_contains(name, [this, &bal](const User &u)
{ bal = u.balance; }) &&
bal)
{ {
users.modify_if(return_account, [ this, bal ](User & u)) users.modify_if(return_account, [ this, bal ](User & u))
{ {
@ -268,6 +278,8 @@ BankResponse Bank::DelUser(const std::string &name) noexcept
} }
} }
void Bank::Save() void Bank::Save()
{
if (GetChangeState())
{ {
static thread_local Json::Value temp; static thread_local Json::Value temp;
@ -277,7 +289,8 @@ void Bank::Save()
for (const auto &u : users) for (const auto &u : users)
{ {
//we know it contains this key but we call this func to grab mutex //we know it contains this key but we call this func to grab mutex
users.if_contains(u.first, [&u](const User &u_val) { temp[u.first] = u_val.Serialize(); }); users.if_contains(u.first, [&u](const User &u_val)
{ temp[u.first] = u_val.Serialize(); });
} }
} }
if (temp.isNull()) if (temp.isNull())
@ -300,6 +313,7 @@ void Bank::Save()
#endif #endif
#endif #endif
} }
}
//NOT THREAD SAFE, BY NO MEANS SHOULD THIS BE CALLED WHILE RECEIEVING REQUESTS //NOT THREAD SAFE, BY NO MEANS SHOULD THIS BE CALLED WHILE RECEIEVING REQUESTS
void Bank::Load() void Bank::Load()