Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Allow history command to be executed while asb is running by opening database in read-only mode #1722

Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions swap/src/api.rs
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
pub mod request;
use crate::cli::command::{Bitcoin, Monero, Tor};
use crate::database::open_db;
use crate::database::{open_db, AccessMode};
use crate::env::{Config as EnvConfig, GetConfig, Mainnet, Testnet};
use crate::fs::system_data_dir;
use crate::network::rendezvous::XmrBtcNamespace;
Expand Down Expand Up @@ -224,7 +224,7 @@ impl Context {
let tor_socks5_port = tor.map_or(9050, |tor| tor.tor_socks5_port);

let context = Context {
db: open_db(data_dir.join("sqlite"), false).await?,
db: open_db(data_dir.join("sqlite"), AccessMode::ReadWrite).await?,
bitcoin_wallet,
monero_wallet,
monero_rpc_process,
Expand Down Expand Up @@ -259,7 +259,7 @@ impl Context {
bitcoin_wallet: Some(bob_bitcoin_wallet),
monero_wallet: Some(bob_monero_wallet),
config,
db: open_db(db_path, false)
db: open_db(db_path, AccessMode::ReadWrite)
.await
.expect("Could not open sqlite database"),
monero_rpc_process: None,
Expand Down
16 changes: 8 additions & 8 deletions swap/src/bin/asb.rs
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,7 @@ use swap::asb::config::{
};
use swap::asb::{cancel, punish, redeem, refund, safely_abort, EventLoop, Finality, KrakenRate};
use swap::common::check_latest_version;
use swap::database::open_db;
use swap::database::{open_db, AccessMode};
use swap::network::rendezvous::XmrBtcNamespace;
use swap::network::swarm;
use swap::protocol::alice::{run, AliceState};
Expand Down Expand Up @@ -100,7 +100,7 @@ async fn main() -> Result<()> {

match cmd {
Command::Start { resume_only } => {
let db = open_db(config.data.dir.join("sqlite"), false).await?;
let db = open_db(config.data.dir.join("sqlite"), AccessMode::ReadWrite).await?;

// check and warn for duplicate rendezvous points
let mut rendezvous_addrs = config.network.rendezvous_point.clone();
Expand Down Expand Up @@ -225,7 +225,7 @@ async fn main() -> Result<()> {
event_loop.run().await;
}
Command::History => {
let db = open_db(config.data.dir.join("sqlite"), true).await?;
let db = open_db(config.data.dir.join("sqlite"), AccessMode::ReadOnly).await?;

let mut table = Table::new();

Expand Down Expand Up @@ -272,7 +272,7 @@ async fn main() -> Result<()> {
tracing::info!(%bitcoin_balance, %monero_balance, "Current balance");
}
Command::Cancel { swap_id } => {
let db = open_db(config.data.dir.join("sqlite"), false).await?;
let db = open_db(config.data.dir.join("sqlite"), AccessMode::ReadWrite).await?;

let bitcoin_wallet = init_bitcoin_wallet(&config, &seed, env_config).await?;

Expand All @@ -281,7 +281,7 @@ async fn main() -> Result<()> {
tracing::info!("Cancel transaction successfully published with id {}", txid);
}
Command::Refund { swap_id } => {
let db = open_db(config.data.dir.join("sqlite"), false).await?;
let db = open_db(config.data.dir.join("sqlite"), AccessMode::ReadWrite).await?;

let bitcoin_wallet = init_bitcoin_wallet(&config, &seed, env_config).await?;
let monero_wallet = init_monero_wallet(&config, env_config).await?;
Expand All @@ -297,7 +297,7 @@ async fn main() -> Result<()> {
tracing::info!("Monero successfully refunded");
}
Command::Punish { swap_id } => {
let db = open_db(config.data.dir.join("sqlite"), false).await?;
let db = open_db(config.data.dir.join("sqlite"), AccessMode::ReadWrite).await?;

let bitcoin_wallet = init_bitcoin_wallet(&config, &seed, env_config).await?;

Expand All @@ -306,7 +306,7 @@ async fn main() -> Result<()> {
tracing::info!("Punish transaction successfully published with id {}", txid);
}
Command::SafelyAbort { swap_id } => {
let db = open_db(config.data.dir.join("sqlite"), false).await?;
let db = open_db(config.data.dir.join("sqlite"), AccessMode::ReadWrite).await?;

safely_abort(swap_id, db).await?;

Expand All @@ -316,7 +316,7 @@ async fn main() -> Result<()> {
swap_id,
do_not_await_finality,
} => {
let db = open_db(config.data.dir.join("sqlite"), false).await?;
let db = open_db(config.data.dir.join("sqlite"), AccessMode::ReadWrite).await?;

let bitcoin_wallet = init_bitcoin_wallet(&config, &seed, env_config).await?;

Expand Down
12 changes: 9 additions & 3 deletions swap/src/database.rs
Original file line number Diff line number Diff line change
Expand Up @@ -83,19 +83,25 @@ impl Swap {
}
}

#[derive(Debug, Clone, Copy, Serialize, Deserialize, Hash, PartialEq)]
pub enum AccessMode {
ReadWrite,
ReadOnly,
}

pub async fn open_db(
sqlite_path: impl AsRef<Path>,
read_only: bool,
access_mode: AccessMode,
) -> Result<Arc<dyn Database + Send + Sync>> {
if sqlite_path.as_ref().exists() {
tracing::debug!("Using existing sqlite database.");
let sqlite = SqliteDatabase::open(sqlite_path, read_only).await?;
let sqlite = SqliteDatabase::open(sqlite_path, access_mode).await?;
Ok(Arc::new(sqlite))
} else {
tracing::debug!("Creating and using new sqlite database.");
ensure_directory_exists(sqlite_path.as_ref())?;
tokio::fs::File::create(&sqlite_path).await?;
let sqlite = SqliteDatabase::open(sqlite_path, read_only).await?;
let sqlite = SqliteDatabase::open(sqlite_path, access_mode).await?;
Ok(Arc::new(sqlite))
}
}
8 changes: 6 additions & 2 deletions swap/src/database/sqlite.rs
Original file line number Diff line number Diff line change
Expand Up @@ -12,15 +12,19 @@ use std::str::FromStr;
use time::OffsetDateTime;
use uuid::Uuid;

use super::AccessMode;

pub struct SqliteDatabase {
pool: Pool<Sqlite>,
}

impl SqliteDatabase {
pub async fn open(path: impl AsRef<Path>, read_only: bool) -> Result<Self>
pub async fn open(path: impl AsRef<Path>, access_mode: AccessMode) -> Result<Self>
where
Self: std::marker::Sized,
{
let read_only = matches!(access_mode, AccessMode::ReadOnly);

let path_str = format!("sqlite:{}", path.as_ref().display());
let options = SqliteConnectOptions::from_str(&path_str)?.read_only(read_only);

Expand Down Expand Up @@ -515,7 +519,7 @@ mod tests {
// file has to exist in order to connect with sqlite
File::create(temp_db.clone()).unwrap();

let db = SqliteDatabase::open(temp_db, false).await?;
let db = SqliteDatabase::open(temp_db, AccessMode::ReadWrite).await?;

Ok(db)
}
Expand Down
8 changes: 4 additions & 4 deletions swap/tests/harness/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ use std::sync::Arc;
use std::time::Duration;
use swap::asb::FixedRate;
use swap::bitcoin::{CancelTimelock, PunishTimelock, TxCancel, TxPunish, TxRedeem, TxRefund};
use swap::database::SqliteDatabase;
use swap::database::{AccessMode, SqliteDatabase};
use swap::env::{Config, GetConfig};
use swap::fs::ensure_directory_exists;
use swap::network::rendezvous::XmrBtcNamespace;
Expand Down Expand Up @@ -232,7 +232,7 @@ async fn start_alice(
tokio::fs::File::create(&db_path).await.unwrap();
}
let db = Arc::new(
SqliteDatabase::open(db_path.as_path(), false)
SqliteDatabase::open(db_path.as_path(), AccessMode::ReadWrite)
.await
.unwrap(),
);
Expand Down Expand Up @@ -437,7 +437,7 @@ impl BobParams {
if !self.db_path.exists() {
tokio::fs::File::create(&self.db_path).await?;
}
let db = Arc::new(SqliteDatabase::open(&self.db_path, false).await?);
let db = Arc::new(SqliteDatabase::open(&self.db_path, AccessMode::ReadWrite).await?);

let (event_loop, handle) = self.new_eventloop(swap_id, db.clone()).await?;

Expand Down Expand Up @@ -467,7 +467,7 @@ impl BobParams {
if !self.db_path.exists() {
tokio::fs::File::create(&self.db_path).await?;
}
let db = Arc::new(SqliteDatabase::open(&self.db_path, false).await?);
let db = Arc::new(SqliteDatabase::open(&self.db_path, AccessMode::ReadWrite).await?);

let (event_loop, handle) = self.new_eventloop(swap_id, db.clone()).await?;

Expand Down
Loading