mirror of
https://github.com/gitbutlerapp/gitbutler.git
synced 2024-12-19 15:41:31 +03:00
72291ce4cb
This is done one-time (for now) using a nightly feature of cargo-fmt
as defined in `rustfmt-nightly.toml.`
It's planned to make this the default so imports will always be sorted.
For now it can be run manually with:
cargo +nightly fmt -- --config-path rustfmt-nightly.toml
or
pnpm rustfmtTBD
Please enter the message for your patch. Lines starting with
48 lines
1.5 KiB
Rust
48 lines
1.5 KiB
Rust
use std::{path, sync::Arc};
|
|
|
|
use anyhow::{Context, Result};
|
|
use r2d2::Pool;
|
|
use r2d2_sqlite::SqliteConnectionManager;
|
|
use refinery::config::Config;
|
|
use rusqlite::Transaction;
|
|
|
|
mod embedded {
|
|
use refinery::embed_migrations;
|
|
embed_migrations!("src/database/migrations");
|
|
}
|
|
|
|
#[derive(Clone)]
|
|
pub struct Database {
|
|
pool: Arc<Pool<SqliteConnectionManager>>,
|
|
}
|
|
|
|
impl Database {
|
|
pub fn open_in_directory<P: AsRef<path::Path>>(path: P) -> Result<Self> {
|
|
let path = path.as_ref().to_path_buf().join("database.sqlite3");
|
|
let manager = SqliteConnectionManager::file(&path);
|
|
let pool = r2d2::Pool::new(manager)?;
|
|
let mut cfg = Config::new(refinery::config::ConfigDbType::Sqlite)
|
|
.set_db_path(path.as_path().to_str().unwrap());
|
|
embedded::migrations::runner()
|
|
.run(&mut cfg)
|
|
.map(|report| {
|
|
report
|
|
.applied_migrations()
|
|
.iter()
|
|
.for_each(|migration| tracing::info!(%migration, "migration applied"));
|
|
})
|
|
.context("Failed to run migrations")?;
|
|
Ok(Self {
|
|
pool: Arc::new(pool),
|
|
})
|
|
}
|
|
|
|
pub fn transaction<T>(&self, f: impl FnOnce(&Transaction) -> Result<T>) -> Result<T> {
|
|
let mut conn = self.pool.get()?;
|
|
let tx = conn.transaction().context("Failed to start transaction")?;
|
|
let result = f(&tx)?;
|
|
tx.commit().context("Failed to commit transaction")?;
|
|
Ok(result)
|
|
}
|
|
}
|