Skip to content

Commit 01c0961

Browse files
committed
Refactor: Streamline the sync API
- Replace sync and sync_no_fail with sync_and_save - Remove some redundant save_to_disk - Update tests to use sync_and_save
1 parent 4d66f7a commit 01c0961

File tree

19 files changed

+54
-61
lines changed

19 files changed

+54
-61
lines changed

src/bin/taker.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -330,7 +330,7 @@ fn main() -> Result<(), TakerError> {
330330

331331
println!("{txid}");
332332

333-
taker.get_wallet_mut().sync_no_fail();
333+
taker.get_wallet_mut().sync_and_save()?;
334334
}
335335
Commands::FetchOffers => {
336336
let all_offers = {

src/maker/api2.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -329,7 +329,7 @@ impl Maker {
329329
config.write_to_file(&data_dir.join("config.toml"))?;
330330

331331
log::info!("Initializing wallet sync");
332-
wallet.sync()?;
332+
wallet.sync_and_save()?;
333333
log::info!("Completed wallet sync");
334334

335335
let network_port = config.network_port;
@@ -483,7 +483,7 @@ impl Maker {
483483
let mut wallet = self.wallet.write()?;
484484

485485
// Sync wallet to get latest UTXO state
486-
wallet.sync()?;
486+
wallet.sync_and_save()?;
487487

488488
let balance = wallet.get_balances()?;
489489
if balance.spendable < connection_state.swap_amount {

src/maker/handlers.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -303,7 +303,7 @@ impl Maker {
303303
.get(funding_output_index as usize)
304304
.expect("funding output expected at this index");
305305

306-
self.wallet.write()?.sync_no_fail();
306+
self.wallet.write()?.sync_and_save()?;
307307

308308
let receiver_contract_tx = create_receivers_contract_tx(
309309
OutPoint {

src/maker/rpc/server.rs

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -105,7 +105,7 @@ fn handle_request<M: MakerRpc>(maker: &Arc<M>, socket: &mut TcpStream) -> Result
105105

106106
let txid = maker.wallet().read()?.send_tx(&tx)?;
107107

108-
maker.wallet().write()?.sync_no_fail();
108+
maker.wallet().write()?.sync_and_save()?;
109109

110110
RpcMsgResp::SendToAddressResp(txid.to_string())
111111
}
@@ -136,11 +136,10 @@ fn handle_request<M: MakerRpc>(maker: &Arc<M>, socket: &mut TcpStream) -> Result
136136
RpcMsgReq::SyncWallet => {
137137
log::info!("Initializing wallet sync");
138138
let mut wallet = maker.wallet().write()?;
139-
if let Err(e) = wallet.sync() {
139+
if let Err(e) = wallet.sync_and_save() {
140140
RpcMsgResp::ServerError(format!("{e:?}"))
141141
} else {
142142
log::info!("Completed wallet sync");
143-
wallet.save_to_disk()?;
144143
RpcMsgResp::Pong
145144
}
146145
}

src/maker/server.rs

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -147,7 +147,7 @@ fn setup_fidelity_bond(maker: &Maker, maker_address: &str) -> Result<FidelityPro
147147
while !maker.shutdown.load(Relaxed) {
148148
sleep_multiplier += 1;
149149
// sync the wallet
150-
maker.get_wallet().write()?.sync_no_fail();
150+
maker.get_wallet().write()?.sync_and_save()?;
151151

152152
let fidelity_result = maker.get_wallet().write()?.create_fidelity(
153153
amount,
@@ -196,8 +196,7 @@ fn setup_fidelity_bond(maker: &Maker, maker_address: &str) -> Result<FidelityPro
196196
*proof = Some(highest_proof);
197197

198198
// sync and save the wallet data to disk
199-
maker.get_wallet().write()?.sync_no_fail();
200-
maker.get_wallet().read()?.save_to_disk()?;
199+
maker.get_wallet().write()?.sync_and_save()?;
201200
break;
202201
}
203202
}
@@ -217,7 +216,7 @@ fn check_swap_liquidity(maker: &Maker) -> Result<(), MakerError> {
217216
let mut sleep_duration = 0;
218217
let addr = maker.get_wallet().write()?.get_next_external_address()?;
219218
while !maker.shutdown.load(Relaxed) {
220-
maker.get_wallet().write()?.sync_no_fail();
219+
maker.get_wallet().write()?.sync_and_save()?;
221220
let offer_max_size = maker.get_wallet().read()?.store.offer_maxsize;
222221

223222
let min_required = maker.config.min_swap_amount;
@@ -501,9 +500,8 @@ pub fn start_maker_server(maker: Arc<Maker>) -> Result<(), MakerError> {
501500
maker.thread_pool.join_all_threads()?;
502501

503502
log::info!("Shutdown wallet sync initiated.");
504-
maker.get_wallet().write()?.sync_no_fail();
503+
maker.get_wallet().write()?.sync_and_save()?;
505504
log::info!("Shutdown wallet syncing completed.");
506-
maker.get_wallet().read()?.save_to_disk()?;
507505
log::info!("Wallet file saved to disk.");
508506
log::info!("Maker Server is shut down successfully.");
509507
Ok(())

src/maker/server2.rs

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -163,7 +163,7 @@ fn setup_fidelity_bond_taproot(
163163
while !maker.shutdown.load(Relaxed) {
164164
sleep_multiplier += 1;
165165
// sync the wallet
166-
maker.wallet().write()?.sync_no_fail();
166+
maker.wallet().write()?.sync_and_save()?;
167167

168168
let fidelity_result = maker.wallet().write()?.create_fidelity(
169169
amount,
@@ -225,8 +225,7 @@ fn setup_fidelity_bond_taproot(
225225
};
226226

227227
// sync and save the wallet data to disk
228-
maker.wallet().write()?.sync_no_fail();
229-
maker.wallet().read()?.save_to_disk()?;
228+
maker.wallet().write()?.sync_and_save()?;
230229

231230
// Store the fidelity proof in maker
232231
*maker.highest_fidelity_proof.write()? = Some(highest_proof.clone());

src/taker/api.rs

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2163,9 +2163,7 @@ impl Taker {
21632163
}
21642164
}
21652165

2166-
self.wallet.sync_no_fail();
2167-
2168-
self.wallet.save_to_disk()?;
2166+
self.wallet.sync_and_save()?;
21692167

21702168
self.clear_ongoing_swaps();
21712169

src/taker/api2.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -407,7 +407,7 @@ impl Taker {
407407
};
408408

409409
log::info!("Initializing wallet sync...");
410-
wallet.sync()?;
410+
wallet.sync_and_save()?;
411411
log::info!("Completed wallet sync");
412412

413413
Ok(Self {

src/wallet/api.rs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2035,7 +2035,7 @@ impl Wallet {
20352035
completed_swapcoins.len()
20362036
);
20372037

2038-
self.sync_no_fail();
2038+
self.sync_and_save()?;
20392039

20402040
for (multisig_redeemscript, _) in completed_swapcoins {
20412041
let utxo_info = self
@@ -2128,7 +2128,7 @@ impl Wallet {
21282128
}
21292129
let reedem_script = incoming.get_multisig_redeemscript();
21302130
let next_internal = &self.get_next_internal_addresses(1)?[0];
2131-
self.sync()?;
2131+
self.sync_and_save()?;
21322132

21332133
let hashlock_spend =
21342134
self.create_hashlock_spend(&incoming, next_internal, MIN_FEE_RATE)?;
@@ -2159,7 +2159,7 @@ impl Wallet {
21592159
let reedem_script = outgoing.get_multisig_redeemscript();
21602160
let timelock = outgoing.get_timelock()?;
21612161
let next_internal = &self.get_next_internal_addresses(1)?[0];
2162-
self.sync()?;
2162+
self.sync_and_save()?;
21632163

21642164
let timelock_spend =
21652165
self.create_timelock_spend(&outgoing, next_internal, MIN_FEE_RATE)?;

src/wallet/backup.rs

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -125,8 +125,7 @@ impl Wallet {
125125
store,
126126
store_enc_material: restored_enc_material,
127127
};
128-
tmp_wallet.sync()?;
129-
tmp_wallet.save_to_disk()?;
128+
tmp_wallet.sync_and_save()?;
130129

131130
Ok(tmp_wallet)
132131
}

0 commit comments

Comments
 (0)