bestool-psql 1.7.2

psql-inspired client for PostgreSQL
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

use std::ops::ControlFlow;

use super::{state::ReplContext, transaction::TransactionState};

pub async fn handle_exit(ctx: &mut ReplContext<'_>) -> ControlFlow<()> {
	let write_mode = {
		let state = ctx.repl_state.lock().unwrap();
		state.write_mode
	};

	if write_mode {
		let tx_state = TransactionState::check(ctx.monitor_client, ctx.backend_pid).await;
		if matches!(tx_state, TransactionState::Active) {
			eprintln!("Cannot exit while in an active transaction. COMMIT or ROLLBACK first.");
			return ControlFlow::Continue(());
		}
	}

	ControlFlow::Break(())
}