Add command to get metrics

Signed-off-by: Matthias Beyer <mail@beyermatthias.de>

1 files changed, 110 insertions, 0 deletions

diff --git a/src/commands/metrics.rs b/src/commands/metrics.rs
new file mode 100644
index 0000000..d7f68fb
--- /dev/null
+++ b/src/commands/metrics.rs
@@ -0,0 +1,110 @@
+//
+// Copyright (c) 2020-2021 science+computing ag and other contributors
+//
+// This program and the accompanying materials are made
+// available under the terms of the Eclipse Public License 2.0
+// which is available at https://www.eclipse.org/legal/epl-2.0/
+//
+// SPDX-License-Identifier: EPL-2.0
+//
+
+use std::path::Path;
+use std::io::Write;
+
+use anyhow::Error;
+use anyhow::Result;
+use diesel::PgConnection;
+use diesel::QueryDsl;
+use diesel::RunQueryDsl;
+use walkdir::WalkDir;
+
+use crate::config::Configuration;
+use crate::repository::Repository;
+
+pub async fn metrics(
+    repo_path: &Path,
+    config: &Configuration,
+    repo: Repository,
+    conn: PgConnection,
+) -> Result<()> {
+    let mut out = std::io::stdout();
+
+    let nfiles = WalkDir::new(repo_path)
+        .follow_links(true)
+        .into_iter()
+        .filter_map(Result::ok)
+        .filter(|d| d.file_type().is_file())
+        .filter(|f| {
+            f.path()
+                .file_name()
+                .map(|name| name == "pkg.toml")
+                .unwrap_or(false)
+        })
+        .count();
+
+    let n_artifacts     = async { crate::schema::artifacts::table.count().get_result::<i64>(&conn) };
+    let n_endpoints     = async { crate::schema::endpoints::table.count().get_result::<i64>(&conn) };
+    let n_envvars       = async { crate::schema::envvars::table.count().get_result::<i64>(&conn) };
+    let n_githashes     = async { crate::schema::githashes::table.count().get_result::<i64>(&conn) };
+    let n_images        = async { crate::schema::images::table.count().get_result::<i64>(&conn) };
+    let n_jobs          = async { crate::schema::jobs::table.count().get_result::<i64>(&conn) };
+    let n_packages      = async { crate::schema::packages::table.count().get_result::<i64>(&conn) };
+    let n_releasestores = async { crate::schema::release_stores::table.count().get_result::<i64>(&conn) };
+    let n_releases      = async { crate::schema::releases::table.count().get_result::<i64>(&conn) };
+    let n_submits       = async { crate::schema::submits::table.count().get_result::<i64>(&conn) };
+
+    let (
+        n_artifacts,
+        n_endpoints,
+        n_envvars,
+        n_githashes,
+        n_images,
+        n_jobs,
+        n_packages,
+        n_releasestores,
+        n_releases,
+        n_submits,
+    ) = tokio::try_join!(n_artifacts, n_endpoints, n_envvars, n_githashes, n_images, n_jobs, n_packages, n_releasestores, n_releases, n_submits)?;
+
+    write!(out, "{}", indoc::formatdoc!(r#"
+        Butido release {release}
+
+        {configured_endpoints} Configured endpoints
+        {configured_images} Configured images
+        {configured_release_stores} Configured release stores
+        {configured_phases} Configures phases
+
+        {nfiles} packages in repository
+        {repo_packages} packages in repository
+
+        {n_artifacts} artifacts in database
+        {n_endpoints} endpoints in database
+        {n_envvars} envvars in database
+        {n_githashes} githashes in database
+        {n_images} images in database
+        {n_jobs} jobs in database
+        {n_packages} packages in database
+        {n_releasestores} releasestores in database
+        {n_releases} releases in database
+        {n_submits} submits in database
+    "#,
+        release = clap::crate_version!(),
+        configured_endpoints = config.docker().endpoints().len(),
+        configured_images = config.docker().images().len(),
+        configured_release_stores = config.release_stores().len(),
+        configured_phases = config.available_phases().len(),
+        nfiles = nfiles,
+        repo_packages = repo.packages().count(),
+        n_artifacts = n_artifacts,
+        n_endpoints = n_endpoints,
+        n_envvars = n_envvars,
+        n_githashes = n_githashes,
+        n_images = n_images,
+        n_jobs = n_jobs,
+        n_packages = n_packages,
+        n_releasestores = n_releasestores,
+        n_releases = n_releases,
+        n_submits = n_submits,
+    )).map_err(Error::from)
+}
+