Rust Example

@quilling.dev · 1mo ago · rust · 64 loc · raw · 0 comments
1//! This module is responsible for handling the database operations.23use sqlx::{sqlite::{SqliteConnectOptions, SqliteJournalMode, SqlitePool}, Executor};4use std::str::FromStr;56use crate::{types, webrequest::Agent};78/// The main function for the database module.9#[tracing::instrument]10pub async fn main_database() -> Result<(), Box<dyn std::error::Error + Send + Sync>> {11    const STATE: u8 = 1;12    let pool_opts = SqliteConnectOptions::from_str("sqlite://prod.db").expect("Expected to be able to configure the database, but failed.")13        .journal_mode(SqliteJournalMode::Wal);14    let pool = SqlitePool::connect_with(pool_opts).await.expect("Expected to be able to connect to the database at sqlite://prod.db but failed.");15    match STATE {16        0 => {17            if initialize_database(&pool).await.is_err() {18                tracing::debug!("Database already initialized");19            }20        },21        1 => {22            validate_labels(&mut Agent::default(), &pool).await?23        },24        _ => (),25    }26    Ok(())27}28async fn initialize_database(pool: &SqlitePool) -> Result<(), sqlx::Error> {29    tracing::debug!("Initializing database");30    let mut connection = pool.acquire().await?;31    _ = connection.execute("PRAGMA foreign_keys=on").await?;32    _ = connection33        .execute("CREATE TABLE profile (did STRING PRIMARY KEY)")34        .await?;35    _ = connection36        .execute(37            "CREATE TABLE profile_stats (38			did	STRING PRIMARY KEY,39			created_at    DATETIME NOT NULL,40			follower_count    INTEGER NOT NULL,41			post_count    INTEGER NOT NULL,42			checked_at	DATETIME NOT NULL,43			FOREIGN KEY(did) REFERENCES profile (did)44		)",45        )46        .await?;47    _ = connection48        .execute(49            "CREATE TABLE profile_labels (50			seq INTEGER PRIMARY KEY AUTOINCREMENT,51			uri	STRING NOT NULL,52            cid STRING,53			val STRING NOT NULL,54            neg BOOLEAN,55			cts	DATETIME NOT NULL,56            exp DATETIME,57            sig BLOB NOT NULL,58			FOREIGN KEY(uri) REFERENCES profile (did)59		)",60        )61        .await?;62    tracing::info!("Database initialized");63    Ok(())64}