cratery 1.11.1

Cratery -- a private cargo registry
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84

/*******************************************************************************
 * Copyright (c) 2024 Cénotélie Opérations SAS (cenotelie.fr)
 ******************************************************************************/

//! Service for persisting information in the database
//! API related to statistics

use super::Database;
use crate::model::stats::GlobalStats;
use crate::model::CrateVersion;
use crate::utils::apierror::ApiError;

impl Database {
    /// Gets the global statistics for the registry
    pub async fn get_crates_stats(&self) -> Result<GlobalStats, ApiError> {
        let total_crates = sqlx::query!("SELECT COUNT(name) AS total_crates FROM Package")
            .fetch_one(&mut *self.transaction.borrow().await)
            .await?
            .total_crates;
        let total_downloads = sqlx::query!("SELECT SUM(downloadCount) AS total_downloads FROM PackageVersion")
            .fetch_one(&mut *self.transaction.borrow().await)
            .await?
            .total_downloads
            .unwrap_or_default();

        let rows = sqlx::query!(
            "SELECT name, version, upload
            FROM Package INNER JOIN PackageVersion ON package = name
            WHERE (SELECT COUNT(version) FROM PackageVersion WHERE package = name) = 1
            ORDER BY upload DESC
            LIMIT 10"
        )
        .fetch_all(&mut *self.transaction.borrow().await)
        .await?;
        let crates_newest = rows
            .into_iter()
            .map(|row| CrateVersion {
                package: row.name,
                version: row.version,
            })
            .collect::<Vec<_>>();

        let rows = sqlx::query!(
            "SELECT name, SUM(downloadCount) AS download_count
            FROM Package INNER JOIN PackageVersion ON package = name
            GROUP BY package
            ORDER BY download_count DESC
            LIMIT 10"
        )
        .fetch_all(&mut *self.transaction.borrow().await)
        .await?;
        let crates_most_downloaded = rows
            .into_iter()
            .map(|row| CrateVersion {
                package: row.name,
                version: String::new(),
            })
            .collect::<Vec<_>>();

        let rows = sqlx::query!(
            "SELECT package, version, upload
            FROM PackageVersion
            ORDER BY upload DESC
            LIMIT 10"
        )
        .fetch_all(&mut *self.transaction.borrow().await)
        .await?;
        let crates_last_updated = rows
            .into_iter()
            .map(|row| CrateVersion {
                package: row.package,
                version: row.version,
            })
            .collect::<Vec<_>>();

        Ok(GlobalStats {
            total_downloads,
            total_crates,
            crates_newest,
            crates_most_downloaded,
            crates_last_updated,
        })
    }
}