tornadocontrib/web3-proxy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

only create owned key if active premium

Browse Source
This commit is contained in:
Bryan Stitt 2023-07-20 21:13:55 -07:00
parent e222572d09
commit c674871c98
1 changed files with 17 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
web3_proxy/src/stats/stat_buffer.rs
View File

@ -219,9 +219,7 @@ impl StatBuffer {
// update the latest balance // update the latest balance
// do this BEFORE emitting any stats // do this BEFORE emitting any stats
let mut approximate_balance_remaining = 0.into(); let mut approximate_balance_remaining = 0.into();
let mut active_premium = false;
// TODO: re-enable this once I know its not the cause of Polygon W3P crashing all the time
// TODO: we want to do this even if the db is down. we need to buffer if there is an outage!
if let Ok(db_conn) = global_db_conn().await { if let Ok(db_conn) = global_db_conn().await {
let user_id = stat.authorization.checks.user_id; let user_id = stat.authorization.checks.user_id;
@ -246,6 +244,7 @@ impl StatBuffer {
// check if they still have premium // check if they still have premium
if user_balance.active_premium() { if user_balance.active_premium() {
// TODO: referall credits here? i think in the save_db section still makes sense for those // TODO: referall credits here? i think in the save_db section still makes sense for those
active_premium = true;
} else if let Err(err) = self } else if let Err(err) = self
.user_balance_cache .user_balance_cache
.invalidate(&user_balance.user_id, &db_conn, &self.rpc_secret_key_cache) .invalidate(&user_balance.user_id, &db_conn, &self.rpc_secret_key_cache)
@ -256,6 +255,8 @@ impl StatBuffer {
warn!(?err, "unable to clear caches"); warn!(?err, "unable to clear caches");
} }
} else if user_balance.active_premium() { } else if user_balance.active_premium() {
active_premium = true;
// paid credits were not used, but now we have active premium. invalidate the caches // paid credits were not used, but now we have active premium. invalidate the caches
// TODO: this seems unliekly. should we warn if this happens so we can investigate? // TODO: this seems unliekly. should we warn if this happens so we can investigate?
if let Err(err) = self if let Err(err) = self
@ -285,12 +286,14 @@ impl StatBuffer {
if self.influxdb_client.is_some() { if self.influxdb_client.is_some() {
// TODO: round the timestamp at all? // TODO: round the timestamp at all?
if let Some(opt_in_timeseries_key) = stat.owned_timeseries_key() { if active_premium {
self.opt_in_timeseries_buffer if let Some(opt_in_timeseries_key) = stat.owned_timeseries_key() {
.entry(opt_in_timeseries_key) self.opt_in_timeseries_buffer
.or_default() .entry(opt_in_timeseries_key)
.add(stat.clone(), approximate_balance_remaining) .or_default()
.await; .add(stat.clone(), approximate_balance_remaining)
.await;
}
} }
let global_timeseries_key = stat.global_timeseries_key(); let global_timeseries_key = stat.global_timeseries_key();
@ -349,6 +352,7 @@ impl StatBuffer {
) )
.await .await
{ {
// TODO: save the stat and retry later!
error!(?err, "unable to save accounting entry!"); error!(?err, "unable to save accounting entry!");
}; };
} }
@ -380,6 +384,7 @@ impl StatBuffer {
points.push(point); points.push(point);
} }
Err(err) => { Err(err) => {
// TODO: what can cause this?
error!(?err, "unable to build global stat!"); error!(?err, "unable to build global stat!");
} }
}; };
@ -395,8 +400,8 @@ impl StatBuffer {
points.push(point); points.push(point);
} }
Err(err) => { Err(err) => {
// TODO: if this errors, we throw away some of the pending stats! we should probably buffer them somewhere to be tried again // TODO: what can cause this?
error!("unable to build opt-in stat! err={:?}", err); error!(?err, "unable to build opt-in stat!");
} }
}; };
} }
@ -426,7 +431,7 @@ impl StatBuffer {
) )
.await .await
{ {
// TODO: if this errors, we throw away some of the pending stats! we should probably buffer them somewhere to be tried again // TODO: if this errors, we throw away some of the pending stats! retry any failures! (but not successes. it can have partial successes!)
error!(?err, batch_size, "unable to save tsdb stats!"); error!(?err, batch_size, "unable to save tsdb stats!");
// TODO: we should probably wait a second to give errors a chance to settle // TODO: we should probably wait a second to give errors a chance to settle
} }