moar cleanup

Author: faukah

src/diff.rs

@@ -259,6 +259,9 @@ pub fn write_package_diff(
   .map_err(Error::from);
 
   tracing::info!(diff_count = ?count.as_ref().ok(), "package diff complete");
+
+  connection.close()?;
+
   count
 }
 
@@ -861,10 +864,14 @@ pub fn spawn_size_diff(
     let mut connection = create_backend(force_correctness);
     connection.connect()?;
 
-    Ok::<_, Error>((
+    let result = (
       connection.query_closure_size(&path_old)?,
       connection.query_closure_size(&path_new)?,
-    ))
+    );
+
+    connection.close()?;
+
+    Ok::<_, Error>(result)
   })
 }
 

src/store.rs

@@ -1,4 +1,3 @@
-#![allow(clippy::mem_forget)]
 //! Provides an interface for querying data from the nix store.
 //!
 //! - [`LazyDBConnection`] is a lazy connection the underlying sqlite database.
@@ -10,10 +9,6 @@ mod queries;
 
 #[cfg(test)] mod test_utils;
 
-pub mod store {
-  pub use crate::store::db_lazy::LazyDBConnection;
-}
-
 use std::{
   fmt::Display,
   iter::Iterator,
@@ -31,8 +26,8 @@ use crate::{
   StorePath,
   store::{
     db_eager::EagerDBConnection,
+    db_lazy::LazyDBConnection,
     nix_command::CommandBackend,
-    store::LazyDBConnection,
   },
 };
 /// The normal database connection