Merge pull request #17 from ginkgobioworks/simple-transaction

Add in simple transactions
ginkgobioworks · Aug 27, 2024 · 1eac034 · 1eac034
2 parents e455722 + 77fc26c
commit 1eac034
Showing 1 changed file with 18 additions and 8 deletions.
diff --git a/src/main.rs b/src/main.rs
@@ -283,21 +283,31 @@ fn main() {
             name,
             db,
             shallow,
-        }) => import_fasta(fasta, name, *shallow, &mut get_connection(db)),
+        }) => {
+            let mut conn = get_connection(db);
+            conn.execute("BEGIN TRANSACTION", []).unwrap();
+            import_fasta(fasta, name, *shallow, &mut conn);
+            conn.execute("END TRANSACTION", []).unwrap();
+        }
         Some(Commands::Update {
             name,
             db,
             fasta,
             vcf,
             genotype,
             sample,
-        }) => update_with_vcf(
-            vcf,
-            name,
-            genotype.clone().unwrap_or("".to_string()),
-            sample.clone().unwrap_or("".to_string()),
-            &mut get_connection(db),
-        ),
+        }) => {
+            let mut conn = get_connection(db);
+            conn.execute("BEGIN TRANSACTION", []).unwrap();
+            update_with_vcf(
+                vcf,
+                name,
+                genotype.clone().unwrap_or("".to_string()),
+                sample.clone().unwrap_or("".to_string()),
+                &mut conn,
+            );
+            conn.execute("END TRANSACTION", []).unwrap();
+        }
         None => {}
     }
 }