Add DbClosedError (#88)

* Add DbClosedError * Fix test * Clean up clippy lint for iter()
Congyuwang · Nov 3, 2023 · 02fe89b · 02fe89b
1 parent 1f73049
commit 02fe89b
Show file tree

Hide file tree

Showing 4 changed files with 328 additions and 388 deletions.
diff --git a/src/exceptions.rs b/src/exceptions.rs
@@ -0,0 +1,9 @@
+use pyo3::create_exception;
+use pyo3::exceptions::PyException;
+
+create_exception!(
+ rocksdict,
+ DbClosedError,
+ PyException,
+ "Raised when accessing a closed database instance."
+);
diff --git a/src/lib.rs b/src/lib.rs
@@ -1,5 +1,6 @@
 // #![feature(core_intrinsics)]
 mod encoder;
+mod exceptions;
 mod iter;
 mod options;
 mod rdict;
@@ -8,6 +9,7 @@ mod sst_file_writer;
 mod util;
 mod write_batch;
 
+use crate::exceptions::*;
 use crate::iter::*;
 use crate::options::*;
 use crate::rdict::*;
@@ -104,7 +106,7 @@ use pyo3::prelude::*;
 /// supports `pickle`.
 ///
 #[pymodule]
-fn rocksdict(_py: Python, m: &PyModule) -> PyResult<()> {
+fn rocksdict(py: Python, m: &PyModule) -> PyResult<()> {
  m.add_class::<Rdict>()?;
  m.add_class::<OptionsPy>()?;
  m.add_class::<MemtableFactoryPy>()?;
@@ -140,6 +142,9 @@ fn rocksdict(_py: Python, m: &PyModule) -> PyResult<()> {
  m.add_class::<BottommostLevelCompactionPy>()?;
  m.add_class::<ChecksumTypePy>()?;
  m.add_class::<KeyEncodingTypePy>()?;
+
+ m.add("DbClosedError", py.get_type::<DbClosedError>())?;
+
  pyo3_log::init();
  Ok(())
 }