164 lines
7.5 KiB
Rust
164 lines
7.5 KiB
Rust
|
use egui::{Context, Ui};
|
||
|
|
use sqlx::mysql::MySqlQueryResult;
|
||
|
|
use std::error::Error;
|
||
|
|
use tokio::sync::mpsc::{channel, Receiver, Sender};
|
||
|
|
use tokio::sync::oneshot;
|
||
|
|
|
||
|
|
use crate::backend::backend_manager::Communication;
|
||
|
|
use crate::backend::database_handler::QueryResult;
|
||
|
|
|
||
|
|
pub struct DBTransactionWindow {
|
||
|
|
sender: Sender<Communication>,
|
||
|
|
/* (Query, Result) */
|
||
|
|
log_history: Vec<QueryResult>,
|
||
|
|
logs_receiver: Option<Receiver<QueryResult>>,
|
||
|
|
is_log_finished_receiver: Option<oneshot::Receiver<bool>>,
|
||
|
|
result: Option<Vec<Result<MySqlQueryResult, Box<dyn Error>>>>,
|
||
|
|
final_result_receiver: Option<oneshot::Receiver<QueryResult>>,
|
||
|
|
working_table_index: usize,
|
||
|
|
}
|
||
|
|
impl DBTransactionWindow {
|
||
|
|
pub fn default(
|
||
|
|
sender: Sender<Communication>,
|
||
|
|
working_table_index: usize,
|
||
|
|
) -> DBTransactionWindow {
|
||
|
|
DBTransactionWindow {
|
||
|
|
sender,
|
||
|
|
log_history: vec![],
|
||
|
|
result: None,
|
||
|
|
logs_receiver: None,
|
||
|
|
final_result_receiver: None,
|
||
|
|
is_log_finished_receiver: None,
|
||
|
|
working_table_index,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
impl DBTransactionWindow {
|
||
|
|
pub fn show(&mut self, ctx: &Context, ui: &mut Ui) {
|
||
|
|
egui::Window::new("Database Transactions")
|
||
|
|
.id(egui::Id::new("Database Transactions"))
|
||
|
|
.resizable(false)
|
||
|
|
.collapsible(true)
|
||
|
|
.title_bar(true)
|
||
|
|
.movable(false)
|
||
|
|
.scroll2([false, true])
|
||
|
|
.enabled(true)
|
||
|
|
.anchor(egui::Align2::CENTER_CENTER, egui::Vec2::ZERO)
|
||
|
|
.show(ctx, |ui| {
|
||
|
|
self.log();
|
||
|
|
self.ui(ctx, ui);
|
||
|
|
});
|
||
|
|
}
|
||
|
|
pub fn log(&mut self) {
|
||
|
|
if self.logs_receiver.is_some() {
|
||
|
|
if let Ok(log) = self.logs_receiver.as_mut().unwrap().try_recv() {
|
||
|
|
self.log_history.push(log);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if self.is_log_finished_receiver.is_some() {
|
||
|
|
if let Ok(finished) = self.is_log_finished_receiver.as_mut().unwrap().try_recv() {
|
||
|
|
//self.result = Some(finished);
|
||
|
|
println!("FINISHED THE QUERY!!!");
|
||
|
|
}
|
||
|
|
}
|
||
|
|
let (log_sender, log_receiver) = channel(2);
|
||
|
|
let (finished_sender, finished_receiver) = oneshot::channel();
|
||
|
|
self.is_log_finished_receiver = Some(finished_receiver);
|
||
|
|
self.logs_receiver = Some(log_receiver);
|
||
|
|
self.sender
|
||
|
|
.try_send(Communication::StartInserting(
|
||
|
|
self.working_table_index,
|
||
|
|
log_sender,
|
||
|
|
finished_sender,
|
||
|
|
))
|
||
|
|
.unwrap_or_else(|_| println!("Failed to send startInserting"));
|
||
|
|
}
|
||
|
|
pub fn ui(&mut self, ctx: &Context, ui: &mut Ui) {
|
||
|
|
ui.add_enabled_ui(self.result.is_some(), |ui| {
|
||
|
|
if ui.button("Roll back").clicked() {
|
||
|
|
if self.final_result_receiver.is_none() {
|
||
|
|
let (sender, receiver) = oneshot::channel();
|
||
|
|
self.final_result_receiver = Some(receiver);
|
||
|
|
self.sender
|
||
|
|
.try_send(Communication::TryRollBack(sender))
|
||
|
|
.unwrap_or_else(|_| println!("failed sending TryCommit receiver"));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if ui.button("Commit").clicked() {
|
||
|
|
if self.final_result_receiver.is_none() {
|
||
|
|
let (sender, receiver) = oneshot::channel();
|
||
|
|
self.final_result_receiver = Some(receiver);
|
||
|
|
self.sender
|
||
|
|
.try_send(Communication::TryCommit(sender))
|
||
|
|
.unwrap_or_else(|_| println!("failed sending TryCommit receiver"));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
});
|
||
|
|
/* DB Output Stuff */
|
||
|
|
egui::ScrollArea::vertical()
|
||
|
|
.auto_shrink([false; 2])
|
||
|
|
.show(ui, |ui| {
|
||
|
|
egui_extras::StripBuilder::new(ui)
|
||
|
|
.size(egui_extras::Size::remainder().at_least(100.0))
|
||
|
|
.size(egui_extras::Size::remainder())
|
||
|
|
.vertical(|mut strip| {
|
||
|
|
strip.cell(|ui| {
|
||
|
|
egui::ScrollArea::horizontal().show(ui, |ui| {
|
||
|
|
let mut table = egui_extras::TableBuilder::new(ui)
|
||
|
|
.striped(true)
|
||
|
|
.resizable(true)
|
||
|
|
.cell_layout(egui::Layout::left_to_right(egui::Align::LEFT))
|
||
|
|
.column(egui_extras::Column::auto().clip(false))
|
||
|
|
.column(egui_extras::Column::auto().clip(false))
|
||
|
|
.min_scrolled_height(0.0);
|
||
|
|
table
|
||
|
|
.header(20.0, |mut header| {
|
||
|
|
header.col(|ui| {
|
||
|
|
ui.strong("Query");
|
||
|
|
});
|
||
|
|
header.col(|ui| {
|
||
|
|
ui.strong("Result");
|
||
|
|
});
|
||
|
|
})
|
||
|
|
.body(|mut body| {
|
||
|
|
body.rows(
|
||
|
|
15.0,
|
||
|
|
self.log_history.len(),
|
||
|
|
|row_index, mut row| {
|
||
|
|
let log = self.log_history.get(row_index).unwrap();
|
||
|
|
row.col(|ui| {
|
||
|
|
ui.label(&log.query);
|
||
|
|
});
|
||
|
|
row.col(|ui| {
|
||
|
|
match &log.result {
|
||
|
|
Ok(rows) => {
|
||
|
|
ui.style_mut()
|
||
|
|
.visuals
|
||
|
|
.override_text_color =
|
||
|
|
Some(egui::Color32::RED);
|
||
|
|
ui.label(format!(
|
||
|
|
"Success! Rows affected: {}",
|
||
|
|
rows.rows_affected()
|
||
|
|
));
|
||
|
|
}
|
||
|
|
Err(e) => {
|
||
|
|
ui.style_mut()
|
||
|
|
.visuals
|
||
|
|
.override_text_color =
|
||
|
|
Some(egui::Color32::RED);
|
||
|
|
ui.label(format!("Error! {}", e));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
ui.reset_style();
|
||
|
|
});
|
||
|
|
},
|
||
|
|
);
|
||
|
|
});
|
||
|
|
ui.scroll_to_cursor(Some(egui::Align::BOTTOM));
|
||
|
|
});
|
||
|
|
});
|
||
|
|
})
|
||
|
|
});
|
||
|
|
}
|
||
|
|
}
|