84 lines
3.0 KiB
Rust
84 lines
3.0 KiB
Rust
|
use crate::errors::{Web3ProxyResponse, Web3ProxyResult};
|
||
|
|
use fstrings::{f, format_args_f};
|
||
|
|
use migration::sea_orm::{
|
||
|
|
DbBackend, DbConn, FromQueryResult, JsonValue, QueryResult, SqlxMySqlPoolConnection, Statement,
|
||
|
|
};
|
||
|
|
use migration::{sea_orm, ConnectionTrait};
|
||
|
|
use rust_decimal::Decimal;
|
||
|
|
use serde::{Deserialize, Serialize};
|
||
|
|
use serde_json::Value;
|
||
|
|
use std::num::NonZeroU64;
|
||
|
|
use tracing::info;
|
||
|
|
|
||
|
|
/// Implements the balance getter
|
||
|
|
#[derive(Clone, Debug, Default, Serialize, Deserialize, FromQueryResult)]
|
||
|
|
pub struct Balance {
|
||
|
|
pub user_id: u64,
|
||
|
|
pub total_spent_paid_credits: Decimal,
|
||
|
|
pub total_spent: Decimal,
|
||
|
|
pub total_deposits: Decimal,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Balance {
|
||
|
|
pub fn remaining(&self) -> Decimal {
|
||
|
|
self.total_deposits - self.total_spent_paid_credits
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub async fn try_get_balance_from_db(
|
||
|
|
db_conn: &DbConn,
|
||
|
|
user_id: u64,
|
||
|
|
) -> Web3ProxyResult<Option<Balance>> {
|
||
|
|
// Return early if user_id == 0
|
||
|
|
if user_id == 0 {
|
||
|
|
return Ok(None);
|
||
|
|
}
|
||
|
|
|
||
|
|
// Injecting the variable directly, should be fine because Rust is typesafe, especially with primitives
|
||
|
|
let raw_sql = f!(r#"
|
||
|
|
SELECT
|
||
|
|
user.id AS user_id,
|
||
|
|
COALESCE(SUM(admin_receipt.amount), 0) + COALESCE(SUM(chain_receipt.amount), 0) + COALESCE(SUM(stripe_receipt.amount), 0) + COALESCE(SUM(referee.one_time_bonus_applied_for_referee), 0) + COALESCE(referrer_bonus.total_bonus, 0) AS total_deposits,
|
||
|
|
COALESCE(SUM(accounting.sum_credits_used), 0) AS total_spent_paid_credits,
|
||
|
|
COALESCE(SUM(accounting.sum_incl_free_credits_used), 0) AS total_spent
|
||
|
|
FROM
|
||
|
|
user
|
||
|
|
LEFT JOIN
|
||
|
|
admin_increase_balance_receipt AS admin_receipt ON user.id = admin_receipt.deposit_to_user_id
|
||
|
|
LEFT JOIN
|
||
|
|
increase_on_chain_balance_receipt AS chain_receipt ON user.id = chain_receipt.deposit_to_user_id
|
||
|
|
LEFT JOIN
|
||
|
|
stripe_increase_balance_receipt AS stripe_receipt ON user.id = stripe_receipt.deposit_to_user_id
|
||
|
|
LEFT JOIN
|
||
|
|
referee ON user.id = referee.user_id
|
||
|
|
LEFT JOIN
|
||
|
|
(SELECT referrer.user_id, SUM(referee.credits_applied_for_referrer) AS total_bonus
|
||
|
|
FROM referrer
|
||
|
|
JOIN referee ON referrer.id = referee.used_referral_code
|
||
|
|
GROUP BY referrer.user_id) AS referrer_bonus ON user.id = referrer_bonus.user_id
|
||
|
|
LEFT JOIN
|
||
|
|
rpc_key ON user.id = rpc_key.user_id
|
||
|
|
LEFT JOIN
|
||
|
|
rpc_accounting_v2 AS accounting ON rpc_key.id = accounting.rpc_key_id
|
||
|
|
LEFT JOIN
|
||
|
|
user_tier ON user.user_tier_id = user_tier.id
|
||
|
|
WHERE
|
||
|
|
user.id = {};
|
||
|
|
"#, user_id).to_string();
|
||
|
|
|
||
|
|
let balance: Balance = match Balance::find_by_statement(Statement::from_string(
|
||
|
|
DbBackend::MySql,
|
||
|
|
raw_sql,
|
||
|
|
// [.into()],
|
||
|
|
))
|
||
|
|
.one(db_conn)
|
||
|
|
.await?
|
||
|
|
{
|
||
|
|
None => return Ok(None),
|
||
|
|
Some(x) => x,
|
||
|
|
};
|
||
|
|
|
||
|
|
// Return None if there is no entry
|
||
|
|
Ok(Some(balance))
|
||
|
|
}
|