LemmyNet/lemmy
2
0
Fork 0
mirror of https://github.com/LemmyNet/lemmy.git synced 2024-11-27 14:51:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

attempt 2

Browse source
This commit is contained in:
Dull Bananas 2024-05-20 19:35:42 +00:00
parent 592a127954
commit a909d2d643
1 changed files with 14 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
crates/db_schema/src/schema_setup/diff_check.rs
View file

@ -16,7 +16,7 @@ enum DumpAction {
pub struct DiffChecker { pub struct DiffChecker {
snapshot_conn: PgConnection, snapshot_conn: PgConnection,
handles: Vec<thread::JoinHandle<()>>, handles: Vec<thread::JoinHandle<()>>,
snapshot_sender: Option<Sender<(String, DumpAction)>>, snapshot_sender: Option<Sender<((String,PgConnection), DumpAction)>>,
error: Receiver<Box<dyn Any + Send + 'static>>, error: Receiver<Box<dyn Any + Send + 'static>>,
// todo rename to channels // todo rename to channels
//dump_receivers: Arc<Mutex<HashMap<String, Receiver<String>>>>, //dump_receivers: Arc<Mutex<HashMap<String, Receiver<String>>>>,
@ -43,13 +43,14 @@ impl DiffChecker {
let rx2 = rx.clone(); let rx2 = rx.clone();
let error_t = error_t.clone(); let error_t = error_t.clone();
handles.push(thread::spawn(move || if let Err(e) = std::panic::catch_unwind(move || { handles.push(thread::spawn(move || if let Err(e) = std::panic::catch_unwind(move || {
while let Ok((snapshot, action)) = rx2.recv() { while let Ok(((snapshot,conn), action)) = rx2.recv() {
let snapshot_arg = format!("--snapshot={snapshot}"); let snapshot_arg = format!("--snapshot={snapshot}");
let output = Command::new("pg_dump") let output = Command::new("pg_dump")
.args(["--schema-only", &snapshot_arg]) .args(["--schema-only", &snapshot_arg])
.env("DATABASE_URL", SETTINGS.get_database_url()) .env("DATABASE_URL", SETTINGS.get_database_url())
.output() .output()
.expect("failed to start pg_dump process"); .expect("failed to start pg_dump process");
std::mem::drop(conn);
if !output.status.success() { if !output.status.success() {
panic!("{}", String::from_utf8(output.stderr).expect("")); panic!("{}", String::from_utf8(output.stderr).expect(""));
@ -77,7 +78,8 @@ impl DiffChecker {
fn check_err(&mut self) { fn check_err(&mut self) {
if let Ok(e) = self.error.try_recv() { if let Ok(e) = self.error.try_recv() {
std::panic::resume_unwind(e); //std::panic::resume_unwind(e);
panic!();
} }
} }
@ -89,10 +91,15 @@ impl DiffChecker {
self.check_err(); self.check_err();
} }
fn get_snapshot(&mut self) -> String { fn get_snapshot(&mut self) -> (String, PgConnection) {
diesel::select(pg_export_snapshot()) let db_url = SETTINGS.get_database_url();
.get_result::<String>(&mut self.snapshot_conn) let mut conn = PgConnection::establish(&db_url).expect("conn2");
.expect("pg_export_snapshot failed") conn.batch_execute("BEGIN;").expect("begin");
//conn.batch_execute("SELECT * FROM __diesel_schema_migrations;").expect("b");
let snapshot = diesel::select(pg_export_snapshot())
.get_result::<String>(&mut conn)
.expect("pg_export_snapshot failed");
(snapshot,conn)
} }
pub fn get_dump(&mut self) -> Receiver<String> { pub fn get_dump(&mut self) -> Receiver<String> {