heimoshuiyu/rust-synapse-compress-state
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Die if fail to write to file

Browse Source
This commit is contained in:
Erik Johnston
2019-03-19 13:34:06 +00:00
parent 2126625ef2
commit b9be6c5ed1
1 changed files with 14 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
src/main.rs
View File

@@ -236,33 +236,35 @@ fn main() {
if old_entry != new_entry { if old_entry != new_entry {
if transactions { if transactions {
writeln!(output, "BEGIN;"); writeln!(output, "BEGIN;").unwrap();
} }
writeln!( writeln!(
output, output,
"DELETE FROM state_group_edges WHERE state_group = {};", "DELETE FROM state_group_edges WHERE state_group = {};",
sg sg
); )
.unwrap();
if let Some(prev_sg) = new_entry.prev_state_group { if let Some(prev_sg) = new_entry.prev_state_group {
writeln!(output, "INSERT INTO state_group_edges (state_group, prev_state_group) VALUES ({}, {});", sg, prev_sg); writeln!(output, "INSERT INTO state_group_edges (state_group, prev_state_group) VALUES ({}, {});", sg, prev_sg).unwrap();
} }
writeln!( writeln!(
output, output,
"DELETE FROM state_groups_state WHERE state_group = {};", "DELETE FROM state_groups_state WHERE state_group = {};",
sg sg
); )
.unwrap();
if new_entry.state_map.len() > 0 { if new_entry.state_map.len() > 0 {
writeln!(output, "INSERT INTO state_groups_state (state_group, room_id, type, state_key, event_id) VALUES"); writeln!(output, "INSERT INTO state_groups_state (state_group, room_id, type, state_key, event_id) VALUES").unwrap();
let mut first = true; let mut first = true;
for ((t, s), e) in new_entry.state_map.iter() { for ((t, s), e) in new_entry.state_map.iter() {
if first { if first {
write!(output, " "); write!(output, " ").unwrap();
first = false; first = false;
} else { } else {
write!(output, " ,"); write!(output, " ,").unwrap();
} }
writeln!( writeln!(
output, output,
@@ -272,15 +274,16 @@ fn main() {
PGEscapse(t), PGEscapse(t),
PGEscapse(s), PGEscapse(s),
PGEscapse(e) PGEscapse(e)
); )
.unwrap();
} }
writeln!(output, ";"); writeln!(output, ";").unwrap();
} }
if transactions { if transactions {
writeln!(output, "COMMIT;"); writeln!(output, "COMMIT;").unwrap();
} }
writeln!(output); writeln!(output).unwrap();
} }
pb.inc(1); pb.inc(1);