add rpc_accounting bin

This commit is contained in:
Bryan Stitt 2023-01-11 17:36:23 -08:00
parent 364097dcec
commit 80b84f3a60
3 changed files with 156 additions and 248 deletions

@ -1,241 +0,0 @@
use std::str::FromStr;
// select all requests for a timeline. sum bandwidth and request count. give `cost / byte` and `cost / request`.
use anyhow::Context;
use argh::FromArgs;
use entities::{rpc_accounting, rpc_key, user};
use ethers::types::Address;
use log::{debug, info};
use migration::{
sea_orm::{
self,
prelude::{DateTimeUtc, Decimal},
ColumnTrait, DatabaseConnection, EntityTrait, FromQueryResult, QueryFilter, QuerySelect,
},
Condition,
};
#[derive(Debug, PartialEq, Eq)]
pub enum TimeFrame {
Day,
Month,
}
impl TimeFrame {
pub fn as_seconds(&self) -> u64 {
match self {
Self::Day => 86_400,
Self::Month => 2_628_000,
}
}
}
impl FromStr for TimeFrame {
type Err = anyhow::Error;
fn from_str(s: &str) -> anyhow::Result<Self> {
match s.to_lowercase().as_ref() {
"day" => Ok(Self::Day),
"month" => Ok(Self::Month),
_ => Err(anyhow::anyhow!("Invalid string. Should be day or month.")),
}
}
}
/// calculate costs
#[derive(FromArgs, PartialEq, Debug, Eq)]
#[argh(subcommand, name = "cost_calculator")]
pub struct CostCalculatorSubCommand {
/// dollar cost of running web3-proxy
#[argh(positional)]
cost: Decimal,
#[argh(positional)]
cost_timeframe: TimeFrame,
/// the address of the user to check. If none, check all.
/// TODO: allow checking a single key
#[argh(option)]
address: Option<String>,
/// the chain id to check. If none, check all.
#[argh(option)]
chain_id: Option<u64>,
// TODO: query start and end dates
// TODO: goal price
}
impl CostCalculatorSubCommand {
pub async fn main(self, db_conn: &DatabaseConnection) -> anyhow::Result<()> {
#[derive(Debug, FromQueryResult)]
struct SelectResult {
pub total_frontend_requests: Decimal,
// pub total_backend_retries: Decimal,
// pub total_cache_misses: Decimal,
pub total_cache_hits: Decimal,
pub total_response_bytes: Decimal,
pub total_error_responses: Decimal,
// pub total_response_millis: Decimal,
pub first_period_datetime: DateTimeUtc,
pub last_period_datetime: DateTimeUtc,
}
let q = rpc_accounting::Entity::find()
.select_only()
.column_as(
rpc_accounting::Column::FrontendRequests.sum(),
"total_frontend_requests",
)
// .column_as(
// rpc_accounting::Column::BackendRequests.sum(),
// "total_backend_retries",
// )
// .column_as(
// rpc_accounting::Column::CacheMisses.sum(),
// "total_cache_misses",
// )
.column_as(rpc_accounting::Column::CacheHits.sum(), "total_cache_hits")
.column_as(
rpc_accounting::Column::SumResponseBytes.sum(),
"total_response_bytes",
)
.column_as(
// TODO: can we sum bools like this?
rpc_accounting::Column::ErrorResponse.sum(),
"total_error_responses",
)
// .column_as(
// rpc_accounting::Column::SumResponseMillis.sum(),
// "total_response_millis",
// )
.column_as(
rpc_accounting::Column::PeriodDatetime.min(),
"first_period_datetime",
)
.column_as(
rpc_accounting::Column::PeriodDatetime.max(),
"last_period_datetime",
);
let mut condition = Condition::all();
// TODO: do this with add_option? try operator is harder to use then
if let Some(address) = self.address {
let address: Vec<u8> = address.parse::<Address>()?.to_fixed_bytes().into();
// TODO: find_with_related
let u = user::Entity::find()
.filter(user::Column::Address.eq(address))
.one(db_conn)
.await?
.context("no user found")?;
// TODO: select_only
let u_keys = rpc_key::Entity::find()
.filter(rpc_key::Column::UserId.eq(u.id))
.all(db_conn)
.await?;
if u_keys.is_empty() {
return Err(anyhow::anyhow!("no user keys"));
}
let u_key_ids: Vec<_> = u_keys.iter().map(|x| x.id).collect();
condition = condition.add(rpc_accounting::Column::RpcKeyId.is_in(u_key_ids));
}
let q = q.filter(condition);
// TODO: make this work without into_json. i think we need to make a struct
let query_response = q
.into_model::<SelectResult>()
.one(db_conn)
.await?
.context("no query result")?;
info!("query_response: {:#?}", query_response);
// todo!("calculate cost/frontend request, cost/response_byte, calculate with a discount for cache hits");
let cost_seconds: Decimal = self.cost_timeframe.as_seconds().into();
debug!("cost_seconds: {}", cost_seconds);
let query_seconds: Decimal = query_response
.last_period_datetime
.signed_duration_since(query_response.first_period_datetime)
.num_seconds()
.into();
info!("query seconds: {}", query_seconds);
info!("$0.000005 = goal");
let x = costs(
query_response.total_frontend_requests,
query_seconds,
cost_seconds,
self.cost,
);
info!("${} = frontend request cost", x);
let x = costs(
query_response.total_frontend_requests - query_response.total_error_responses,
query_seconds,
cost_seconds,
self.cost,
);
info!("${} = frontend request cost excluding errors", x);
let x = costs(
query_response.total_frontend_requests
- query_response.total_error_responses
- query_response.total_cache_hits,
query_seconds,
cost_seconds,
self.cost,
);
info!(
"${} = frontend request cost excluding errors and cache hits",
x
);
let x = costs(
query_response.total_frontend_requests
- query_response.total_error_responses
- (query_response.total_cache_hits / Decimal::from(2)),
query_seconds,
cost_seconds,
self.cost,
);
info!(
"${} = frontend request cost excluding errors and half cache hits",
x
);
let x = costs(
query_response.total_response_bytes,
query_seconds,
cost_seconds,
self.cost,
);
info!("${} = response byte cost", x);
// TODO: another script that takes these numbers and applies to a single user?
Ok(())
}
}
fn costs(
query_total: Decimal,
query_seconds: Decimal,
cost_seconds: Decimal,
cost: Decimal,
) -> Decimal {
let requests_per_second = query_total / query_seconds;
let requests_per_cost_timeframe = requests_per_second * cost_seconds;
let request_cost_per_query = cost / requests_per_cost_timeframe;
request_cost_per_query.round_dp(9)
}

@ -3,12 +3,12 @@ mod change_user_tier;
mod change_user_tier_by_address;
mod change_user_tier_by_key;
mod check_config;
mod cost_calculator;
mod count_users;
mod create_user;
mod drop_migration_lock;
mod health_compass;
mod list_user_tier;
mod rpc_accounting;
mod transfer_key;
mod user_export;
mod user_import;
@ -47,11 +47,11 @@ enum SubCommand {
ChangeUserTierByAddress(change_user_tier_by_address::ChangeUserTierByAddressSubCommand),
ChangeUserTierByKey(change_user_tier_by_key::ChangeUserTierByKeySubCommand),
CheckConfig(check_config::CheckConfigSubCommand),
CostCalculator(cost_calculator::CostCalculatorSubCommand),
CountUsers(count_users::CountUsersSubCommand),
CreateUser(create_user::CreateUserSubCommand),
DropMigrationLock(drop_migration_lock::DropMigrationLockSubCommand),
HealthCompass(health_compass::HealthCompassSubCommand),
RpcAccounting(rpc_accounting::RpcAccountingSubCommand),
TransferKey(transfer_key::TransferKeySubCommand),
UserExport(user_export::UserExportSubCommand),
UserImport(user_import::UserImportSubCommand),
@ -116,11 +116,6 @@ async fn main() -> anyhow::Result<()> {
x.main(&db_conn).await
}
SubCommand::CostCalculator(x) => {
let db_conn = get_db(cli_config.db_url, 1, 1).await?;
x.main(&db_conn).await
}
SubCommand::CountUsers(x) => {
let db_conn = get_db(cli_config.db_url, 1, 1).await?;
@ -133,6 +128,11 @@ async fn main() -> anyhow::Result<()> {
x.main(&db_conn).await
}
SubCommand::HealthCompass(x) => x.main().await,
SubCommand::RpcAccounting(x) => {
let db_conn = get_migrated_db(cli_config.db_url, 1, 1).await?;
x.main(&db_conn).await
}
SubCommand::TransferKey(x) => {
let db_conn = get_db(cli_config.db_url, 1, 1).await?;

@ -0,0 +1,149 @@
// select all requests for a timeline. sum bandwidth and request count. give `cost / byte` and `cost / request`.
use anyhow::Context;
use argh::FromArgs;
use entities::{rpc_accounting, rpc_key, user};
use ethers::types::Address;
use log::info;
use migration::{
sea_orm::{
self,
prelude::{DateTimeUtc, Decimal},
ColumnTrait, DatabaseConnection, EntityTrait, FromQueryResult, QueryFilter, QuerySelect,
},
Condition,
};
/// count requests
#[derive(FromArgs, PartialEq, Debug, Eq)]
#[argh(subcommand, name = "rpc_accounting")]
pub struct RpcAccountingSubCommand {
/// the address of the user to check. If none, check all.
/// TODO: allow checking a single key
#[argh(option)]
address: Option<String>,
/// the chain id to check. If none, check all.
#[argh(option)]
chain_id: Option<u64>,
/// unix epoch timestamp of earliest entry
#[argh(option)]
start_timestamp: Option<u64>,
/// unix epoch timestamp of last entry
#[argh(option)]
end_timestamp: Option<u64>,
}
impl RpcAccountingSubCommand {
pub async fn main(self, db_conn: &DatabaseConnection) -> anyhow::Result<()> {
#[derive(Debug, FromQueryResult)]
struct SelectResult {
total_frontend_requests: Decimal,
// pub total_backend_retries: Decimal,
// pub total_cache_misses: Decimal,
total_cache_hits: Decimal,
total_response_bytes: Decimal,
total_error_responses: Decimal,
// pub total_response_millis: Decimal,
first_period_datetime: DateTimeUtc,
last_period_datetime: DateTimeUtc,
}
let mut q = rpc_accounting::Entity::find()
.select_only()
.column_as(
rpc_accounting::Column::FrontendRequests.sum(),
"total_frontend_requests",
)
// .column_as(
// rpc_accounting::Column::BackendRequests.sum(),
// "total_backend_retries",
// )
// .column_as(
// rpc_accounting::Column::CacheMisses.sum(),
// "total_cache_misses",
// )
.column_as(rpc_accounting::Column::CacheHits.sum(), "total_cache_hits")
.column_as(
rpc_accounting::Column::SumResponseBytes.sum(),
"total_response_bytes",
)
.column_as(
// TODO: can we sum bools like this?
rpc_accounting::Column::ErrorResponse.sum(),
"total_error_responses",
)
// .column_as(
// rpc_accounting::Column::SumResponseMillis.sum(),
// "total_response_millis",
// )
.column_as(
rpc_accounting::Column::PeriodDatetime.min(),
"first_period_datetime",
)
.column_as(
rpc_accounting::Column::PeriodDatetime.max(),
"last_period_datetime",
);
let mut condition = Condition::all();
// TODO: do this with add_option? try operator is harder to use then
if let Some(address) = self.address {
let address: Vec<u8> = address.parse::<Address>()?.to_fixed_bytes().into();
// TODO: find_with_related
let u = user::Entity::find()
.filter(user::Column::Address.eq(address))
.one(db_conn)
.await?
.context("no user found")?;
// TODO: select_only
let u_keys = rpc_key::Entity::find()
.filter(rpc_key::Column::UserId.eq(u.id))
.all(db_conn)
.await?;
if u_keys.is_empty() {
return Err(anyhow::anyhow!("no user keys"));
}
let u_key_ids: Vec<_> = u_keys.iter().map(|x| x.id).collect();
condition = condition.add(rpc_accounting::Column::RpcKeyId.is_in(u_key_ids));
}
if let Some(start_timestamp) = self.start_timestamp {
condition = condition.add(rpc_accounting::Column::PeriodDatetime.gte(start_timestamp))
}
if let Some(end_timestamp) = self.end_timestamp {
condition = condition.add(rpc_accounting::Column::PeriodDatetime.lte(end_timestamp))
}
q = q.filter(condition);
// TODO: make this work without into_json. i think we need to make a struct
let query_response = q
.into_model::<SelectResult>()
.one(db_conn)
.await?
.context("no query result")?;
info!(
"query_response for chain {:?}: {:#?}",
self.chain_id, query_response
);
// let query_seconds: Decimal = query_response
// .last_period_datetime
// .signed_duration_since(query_response.first_period_datetime)
// .num_seconds()
// .into();
// info!("query seconds: {}", query_seconds);
Ok(())
}
}