Add checkpoint functionality (#4)

* add checkpoint support

* remove .gitignore entries

* pass wrapped_db to checkpoint create

fix import syntax

Co-authored-by: Zhou Xiaoqiang <[email protected]>

Author: wbarnha

rocksdb/_rocksdb.pyx

@@ -25,6 +25,7 @@ from . cimport snapshot
 from . cimport db
 from . cimport iterator
 from . cimport backup
+from . cimport checkpoint
 from . cimport env
 from . cimport table_factory
 from . cimport memtablerep
@@ -2861,6 +2862,34 @@ cdef class ReversedIterator(object):
         check_status(self.it.ptr.status())
         return ret
 
+
+cdef class Checkpoint(object):
+    cdef checkpoint.Checkpoint* checkpoint
+
+    def  __cinit__(self, DB db):
+        cdef Status st
+        self.checkpoint = NULL
+        st = checkpoint.Checkpoint_Create(
+            db.wrapped_db,
+            cython.address(self.checkpoint))
+
+        check_status(st)
+
+    def __dealloc__(self):
+        if not self.checkpoint == NULL:
+            with nogil:
+                del self.checkpoint
+
+    def create_checkpoint(self, checkpoint_dir):
+        cdef Status st
+        cdef string c_checkpoint_dir
+        c_checkpoint_dir = path_to_string(checkpoint_dir)
+
+        with nogil:
+            st = self.checkpoint.CreateCheckpoint(c_checkpoint_dir)
+        check_status(st)
+
+
 cdef class BackupEngine(object):
     cdef backup.BackupEngine* engine
 

rocksdb/checkpoint.pxd

@@ -0,0 +1,11 @@
+from libcpp.string cimport string
+from .db cimport DB
+from .status cimport Status
+
+cdef extern from "rocksdb/utilities/checkpoint.h" namespace "rocksdb":
+    cdef cppclass Checkpoint:
+        Status CreateCheckpoint(const string&) nogil except+
+
+    cdef Status Checkpoint_Create "rocksdb::Checkpoint::Create"(
+        DB*,
+        Checkpoint**) nogil except+