astu/
db.rs

1mod sqlite;
2
3use anyhow::bail;
4use anyhow::Result;
5use async_trait::async_trait;
6use enum_dispatch::enum_dispatch;
7
8pub use self::sqlite::SqliteDb;
9
10#[async_trait]
11#[enum_dispatch]
12pub trait Db {
13    // Required
14
15    /// Saves a [`PingEntry`] to the database.
16    async fn save_ping(&self, entry: &PingEntry) -> Result<()>;
17
18    /// Loads a [`PingEntry`] from the database by `job_id`.
19    async fn load_ping(&self, job_id: &str) -> Result<Vec<PingEntry>>;
20
21    /// Saves an [`ExecEntry`] to the database.
22    async fn save_exec(&self, entry: &ExecEntry) -> Result<()>;
23
24    /// Loads an [`ExecEntry`] from the database by `job_id`.
25    async fn load_exec(&self, job_id: &str) -> Result<Vec<ExecEntry>>;
26
27    // Defaults
28
29    /// Migrates the database to the newest schema.
30    ///
31    /// By default this does nothing. Override this if needed.
32    async fn migrate(&self) -> Result<()> {
33        Ok(())
34    }
35}
36
37#[enum_dispatch(Db)]
38#[derive(Clone)]
39pub enum DbImpl {
40    Sqlite(SqliteDb),
41}
42
43impl DbImpl {
44    /// # Errors
45    ///
46    /// If any db connection fails.
47    pub async fn try_new(connection_string: &str) -> Result<Self> {
48        if connection_string.contains("sqlite") || connection_string.contains(".db") {
49            let db = SqliteDb::try_new(connection_string).await?;
50            return Ok(db.into());
51        }
52
53        bail!("unable to build a db impl");
54    }
55}
56
57/// Outcome of a `ping` run.
58#[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Ord, sqlx::FromRow)]
59pub struct PingEntry {
60    pub job_id: String,
61    pub target: String,
62    pub error: Option<String>,
63    pub message: Option<Vec<u8>>,
64}
65
66/// Outcome of an `exec` run.
67#[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Ord, sqlx::FromRow)]
68pub struct ExecEntry {
69    pub job_id: String,
70    pub target: String,
71    pub error: Option<String>,
72    pub exit_status: Option<u32>,
73    pub stdout: Option<Vec<u8>>,
74    pub stderr: Option<Vec<u8>>,
75}
astu/db.rs

astu/
db.rs
