mirror/mCaptcha
1
0
Fork 1
mirror of https://github.com/mCaptcha/mCaptcha.git synced 2026-02-12 10:35:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

stats endpoint

Browse Source
This commit is contained in:
realaravinth
2021-07-25 21:15:59 +05:30
parent 0a8d36dc9f
commit 8830961e04
9 changed files with 239 additions and 92 deletions
Download Patch File Download Diff File Expand all files Collapse all files

143
src/stats/fetch.rs
View File

@@ -1,37 +1,60 @@
/*
* Copyright (C) 2021 Aravinth Manivannan <realaravinth@batsense.net>
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as
* published by the Free Software Foundation, either version 3 of the
* License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <https://www.gnu.org/licenses/>.
*/
* Copyright (C) 2021 Aravinth Manivannan <realaravinth@batsense.net>
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as
* published by the Free Software Foundation, either version 3 of the
* License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <https://www.gnu.org/licenses/>.
*/
use actix_web::{web, HttpResponse, Responder};
use serde::{Deserialize, Serialize};
use sqlx::PgPool;
use crate::date::Date;
use crate::errors::*;
use crate::AppData;
#[derive(Debug, Clone, Deserialize, Serialize)]
pub struct Stats {
pub struct StatsUnixTimestamp {
pub config_fetches: Vec<i64>,
pub solves: Vec<i64>,
pub confirms: Vec<i64>,
}
pub struct Stats {
pub config_fetches: Vec<Date>,
pub solves: Vec<Date>,
pub confirms: Vec<Date>,
}
#[derive(Debug, Clone, Deserialize, Serialize)]
pub struct StatsPayload {
pub key: String,
}
#[my_codegen::post(path = "crate::V1_API_ROUTES.auth.login", wrap = "crate::CheckLogin")]
async fn get_stats(
payload: web::Json<StatsPayload>,
data: AppData,
) -> ServiceResult<impl Responder> {
let stats = Stats::new(&payload.key, &data.db).await?;
let stats = StatsUnixTimestamp::from_stats(&stats);
Ok(HttpResponse::Ok().json(&stats))
}
impl Stats {
pub async fn new(key: &str, db: &PgPool) -> ServiceResult<Self> {
let config_fetches_fut = Self::fetch_config_fetched(key, db);
let solves_fut = Self::fetch_solve(key, db);
let confirms_fut = Self::fetch_confirm(key, db);
let config_fetches_fut = runners::fetch_config_fetched(key, db);
let solves_fut = runners::fetch_solve(key, db);
let confirms_fut = runners::fetch_confirm(key, db);
let (config_fetches, solves, confirms) =
futures::try_join!(config_fetches_fut, solves_fut, confirms_fut)?;
@@ -44,70 +67,81 @@ impl Stats {
Ok(res)
}
}
impl StatsUnixTimestamp {
pub fn from_stats(stats: &Stats) -> Self {
let config_fetches = Self::unix_timestamp(&stats.config_fetches);
let solves = Self::unix_timestamp(&stats.solves);
let confirms = Self::unix_timestamp(&stats.confirms);
Self {
config_fetches,
solves,
confirms,
}
}
/// featch PoWConfig confirms
#[inline]
fn unix_timestamp(dates: &Vec<Date>) -> Vec<i64> {
let mut res: Vec<i64> = Vec::with_capacity(dates.len());
dates
.iter()
.for_each(|record| res.push(record.time.unix_timestamp()));
res
}
}
pub mod runners {
use super::*;
/// featch PoWConfig fetches
#[inline]
pub async fn fetch_config_fetched(
key: &str,
db: &PgPool,
) -> ServiceResult<Vec<i64>> {
let records = sqlx::query!(
"SELECT fetched_at FROM mcaptcha_pow_fetched_stats WHERE config_id =
) -> ServiceResult<Vec<Date>> {
let records = sqlx::query_as!(
Date,
"SELECT time FROM mcaptcha_pow_fetched_stats WHERE config_id =
(SELECT config_id FROM mcaptcha_config where key = $1)",
&key,
)
.fetch_all(db)
.await?;
let mut res: Vec<i64> = Vec::with_capacity(records.len());
records
.iter()
.for_each(|record| res.push(record.fetched_at.unix_timestamp()));
Ok(res)
Ok(records)
}
/// featch PoWConfig solves
#[inline]
pub async fn fetch_solve(key: &str, db: &PgPool) -> ServiceResult<Vec<i64>> {
// "SELECT solved_at FROM mcaptcha_pow_solved_stats WHERE config_id =
// (SELECT config_id FROM mcaptcha_config where key = $1)"
let records = sqlx::query!(
"SELECT solved_at FROM mcaptcha_pow_solved_stats WHERE config_id =
pub async fn fetch_solve(key: &str, db: &PgPool) -> ServiceResult<Vec<Date>> {
let records = sqlx::query_as!(
Date,
"SELECT time FROM mcaptcha_pow_solved_stats WHERE config_id =
(SELECT config_id FROM mcaptcha_config where key = $1)",
&key,
)
.fetch_all(db)
.await?;
let mut res: Vec<i64> = Vec::with_capacity(records.len());
records
.iter()
.for_each(|record| res.push(record.solved_at.unix_timestamp()));
Ok(res)
Ok(records)
}
/// featch PoWConfig confirms
#[inline]
pub async fn fetch_confirm(key: &str, db: &PgPool) -> ServiceResult<Vec<i64>> {
let records = sqlx::query!(
"SELECT confirmed_at FROM mcaptcha_pow_confirmed_stats WHERE config_id = (
pub async fn fetch_confirm(key: &str, db: &PgPool) -> ServiceResult<Vec<Date>> {
let records = sqlx::query_as!(
Date,
"SELECT time FROM mcaptcha_pow_confirmed_stats WHERE config_id = (
SELECT config_id FROM mcaptcha_config where key = $1)",
&key
)
.fetch_all(db)
.await?;
let mut res: Vec<i64> = Vec::with_capacity(records.len());
records
.iter()
.for_each(|record| res.push(record.confirmed_at.unix_timestamp()));
Ok(res)
Ok(records)
}
}
@@ -148,5 +182,10 @@ mod tests {
assert_eq!(stats.config_fetches.len(), 1);
assert_eq!(stats.solves.len(), 1);
assert_eq!(stats.confirms.len(), 1);
let ustats = StatsUnixTimestamp::from_stats(&stats);
assert_eq!(ustats.config_fetches.len(), 1);
assert_eq!(ustats.solves.len(), 1);
assert_eq!(ustats.confirms.len(), 1);
}
}