Add Environment::copy

src/environment.rs

@@ -13,7 +13,7 @@ use ffi;
 use error::{Result, lmdb_result};
 use database::Database;
 use transaction::{RoTransaction, RwTransaction, Transaction};
-use flags::{DatabaseFlags, EnvironmentFlags};
+use flags::{DatabaseFlags, EnvironmentCopyFlags, EnvironmentFlags};
 
 #[cfg(windows)]
 /// Adding a 'missing' trait from windows OsStrExt
@@ -138,6 +138,18 @@ impl Environment {
         }
     }
 
+    ///
+    /// Make a consistent copy of the environment in the given destination directory.
+    ///
+    /// Destination must exist, and be an empty directory.
+    ///
+    pub fn copy(&self, destination: &Path, flags: EnvironmentCopyFlags) -> Result<()> {
+        let c_destination = CString::new(destination.as_os_str().as_bytes()).unwrap();
+        unsafe {
+            lmdb_result(ffi::mdb_env_copy2(self.env(), c_destination.as_ptr(), flags.bits()))
+        }
+    }
+
     /// Closes the database handle. Normally unnecessary.
     ///
     /// Closing a database handle is not necessary, but lets `Transaction::open_database` reuse the
@@ -462,4 +474,24 @@ mod test {
         assert_eq!(stat.overflow_pages(), 0);
         assert_eq!(stat.entries(), 64);
     }
+
+    #[test]
+    fn test_copy() {
+        let dir = TempDir::new("test").unwrap();
+        let dir2 = TempDir::new("test-copy").unwrap();
+        {
+            let env = Environment::new().open(dir.path()).unwrap();
+            {
+                let db = env.open_db(None).unwrap();
+                let mut txn = env.begin_rw_txn().unwrap();
+                txn.put(db, b"key1", b"val1", WriteFlags::empty()).unwrap();
+                txn.commit().unwrap();
+            }
+            env.copy(dir2.path(), EnvironmentCopyFlags::empty()).unwrap()
+        }
+        let env = Environment::new().open(dir2.path()).unwrap();
+        let db = env.open_db(None).unwrap();
+        let txn = env.begin_ro_txn().unwrap();
+        assert_eq!(txn.get(db, b"key1").unwrap(), b"val1");
+    }
 }

src/flags.rs

@@ -171,3 +171,14 @@ bitflags! {
         const APPEND_DUP = MDB_APPENDDUP;
     }
 }
+
+bitflags! {
+    #[doc="Environment Copy Options"]
+    pub struct EnvironmentCopyFlags: c_uint {
+
+        #[doc="Perform compaction while copying: omit free pages and sequentially renumber all "]
+        #[doc="pages in output. This option consumes more CPU and runs more slowly than the "]
+        #[doc="default, but may produce a smaller output database."]
+        const COMPACT = MDB_CP_COMPACT;
+    }
+}