Function run_db_migrations

chirpstack/src/storage/mod.rs:177–225

()

}

pub async fn run_db_migrations() -> Result<()> {
    info!("Applying schema migrations");

    let c = get_async_db_conn().await?;
    let mut c_wrapped: AsyncConnectionWrapper<AsyncDbPoolConnection> =
        AsyncConnectionWrapper::from(c);

    #[cfg(feature = "postgres")]
    let _ = task::spawn_blocking(move || -> Result<()> {
        c_wrapped
            .run_pending_migrations(MIGRATIONS)
            .map_err(|e| anyhow!("{}", e))?;

        Ok(())
    })
    .await?;

    #[cfg(feature = "sqlite")]
    let _ = task::spawn_blocking(move || -> Result<()> {
        use diesel::sql_types::Integer;
        use diesel::{QueryableByName, RunQueryDsl};

        #[derive(QueryableByName)]
        struct FkPragma {
            #[diesel(sql_type = Integer)]
            foreign_keys: i32,
        }

        let fk_pragma: i32 = diesel::sql_query("pragma foreign_keys")
            .get_result::<FkPragma>(&mut c_wrapped)?
            .foreign_keys;
        if fk_pragma != 0 {
            let _ = diesel::sql_query("pragma foreign_keys = 0").execute(&mut c_wrapped)?;
        }

        c_wrapped
            .run_pending_migrations(MIGRATIONS)
            .map_err(|e| anyhow!("{}", e))?;

        if fk_pragma != 0 {
            let _ = diesel::sql_query("pragma foreign_keys = 1").execute(&mut c_wrapped)?;
        }

        Ok(())
    })
    .await?;

    Ok(())
}

async fn set_async_redis_pool(p: AsyncRedisPool) {
    let mut pool_w = ASYNC_REDIS_POOL.write().await;