spin_trigger/cli/
summary.rs

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
use spin_core::async_trait;
use spin_factor_key_value::KeyValueFactor;
use spin_factor_sqlite::SqliteFactor;
use spin_factors::RuntimeFactors;
use spin_factors_executor::ExecutorHooks;

/// An [`ExecutorHooks`] that prints information about the default KV store.
pub struct KeyValueDefaultStoreSummaryHook;

#[async_trait]
impl<F: RuntimeFactors, U> ExecutorHooks<F, U> for KeyValueDefaultStoreSummaryHook {
    async fn configure_app(
        &self,
        configured_app: &spin_factors::ConfiguredApp<F>,
    ) -> anyhow::Result<()> {
        let Ok(kv_app_state) = configured_app.app_state::<KeyValueFactor>() else {
            return Ok(());
        };
        if !kv_app_state.store_is_used("default") {
            // We don't talk about unused default stores
            return Ok(());
        }
        if let Some(default_store_summary) = kv_app_state.store_summary("default") {
            println!("Storing default key-value data to {default_store_summary}.");
        }
        Ok(())
    }
}

/// An [`ExecutorHooks`] that prints information about the default KV store.
pub struct SqliteDefaultStoreSummaryHook;

#[async_trait]
impl<F: RuntimeFactors, U> ExecutorHooks<F, U> for SqliteDefaultStoreSummaryHook {
    async fn configure_app(
        &self,
        configured_app: &spin_factors::ConfiguredApp<F>,
    ) -> anyhow::Result<()> {
        let Ok(sqlite_app_state) = configured_app.app_state::<SqliteFactor>() else {
            return Ok(());
        };
        if !sqlite_app_state.database_is_used("default") {
            // We don't talk about unused default databases
            return Ok(());
        }
        if let Some(default_database_summary) = sqlite_app_state
            .get_connection("default")
            .await
            .and_then(Result::ok)
            .and_then(|conn| conn.summary())
        {
            println!("Storing default SQLite data to {default_database_summary}.");
        }
        Ok(())
    }
}