Release the GIL if calls into rocksdb are made
This commit is contained in:
parent
86e6aef6cb
commit
ecb6e26546
1 changed files with 130 additions and 72 deletions
|
@ -15,13 +15,13 @@ cimport options
|
|||
cimport merge_operator
|
||||
cimport filter_policy
|
||||
cimport comparator
|
||||
cimport slice_
|
||||
cimport cache
|
||||
cimport logger
|
||||
cimport snapshot
|
||||
cimport db
|
||||
cimport iterator
|
||||
|
||||
from slice_ cimport Slice
|
||||
from status cimport Status
|
||||
|
||||
import sys
|
||||
|
@ -33,7 +33,7 @@ import traceback
|
|||
import errors
|
||||
|
||||
cdef extern from "cpp/utils.hpp" namespace "py_rocks":
|
||||
cdef const slice_.Slice* vector_data(vector[slice_.Slice]&)
|
||||
cdef const Slice* vector_data(vector[Slice]&)
|
||||
|
||||
## Here comes the stuff to wrap the status to exception
|
||||
cdef check_status(const Status& st):
|
||||
|
@ -71,10 +71,10 @@ cdef string bytes_to_string(path) except *:
|
|||
cdef string_to_bytes(string ob):
|
||||
return PyBytes_FromStringAndSize(ob.c_str(), ob.size())
|
||||
|
||||
cdef slice_.Slice bytes_to_slice(ob) except *:
|
||||
return slice_.Slice(PyBytes_AsString(ob), PyBytes_Size(ob))
|
||||
cdef Slice bytes_to_slice(ob) except *:
|
||||
return Slice(PyBytes_AsString(ob), PyBytes_Size(ob))
|
||||
|
||||
cdef slice_to_bytes(slice_.Slice sl):
|
||||
cdef slice_to_bytes(Slice sl):
|
||||
return PyBytes_FromStringAndSize(sl.data(), sl.size())
|
||||
|
||||
## only for filsystem paths
|
||||
|
@ -149,8 +149,8 @@ cdef class PyBytewiseComparator(PyComparator):
|
|||
|
||||
cdef int compare_callback(
|
||||
void* ctx,
|
||||
const slice_.Slice& a,
|
||||
const slice_.Slice& b) with gil:
|
||||
const Slice& a,
|
||||
const Slice& b) with gil:
|
||||
|
||||
return (<object>ctx).compare(slice_to_bytes(a), slice_to_bytes(b))
|
||||
|
||||
|
@ -197,7 +197,7 @@ cdef class PyGenericFilterPolicy(PyFilterPolicy):
|
|||
|
||||
cdef void create_filter_callback(
|
||||
void* ctx,
|
||||
const slice_.Slice* keys,
|
||||
const Slice* keys,
|
||||
int n,
|
||||
string* dst) with gil:
|
||||
|
||||
|
@ -207,8 +207,8 @@ cdef void create_filter_callback(
|
|||
|
||||
cdef cpp_bool key_may_match_callback(
|
||||
void* ctx,
|
||||
const slice_.Slice& key,
|
||||
const slice_.Slice& filt) with gil:
|
||||
const Slice& key,
|
||||
const Slice& filt) with gil:
|
||||
|
||||
return (<object>ctx).key_may_match(
|
||||
slice_to_bytes(key),
|
||||
|
@ -229,7 +229,7 @@ cdef class PyBloomFilterPolicy(PyFilterPolicy):
|
|||
|
||||
def create_filter(self, keys):
|
||||
cdef string dst
|
||||
cdef vector[slice_.Slice] c_keys
|
||||
cdef vector[Slice] c_keys
|
||||
|
||||
for key in keys:
|
||||
c_keys.push_back(bytes_to_slice(key))
|
||||
|
@ -294,9 +294,9 @@ cdef class PyMergeOperator(object):
|
|||
|
||||
cdef cpp_bool merge_callback(
|
||||
void* ctx,
|
||||
const slice_.Slice& key,
|
||||
const slice_.Slice* existing_value,
|
||||
const slice_.Slice& value,
|
||||
const Slice& key,
|
||||
const Slice* existing_value,
|
||||
const Slice& value,
|
||||
string* new_value,
|
||||
logger.Logger* log) with gil:
|
||||
|
||||
|
@ -326,8 +326,8 @@ cdef cpp_bool merge_callback(
|
|||
|
||||
cdef cpp_bool full_merge_callback(
|
||||
void* ctx,
|
||||
const slice_.Slice& key,
|
||||
const slice_.Slice* existing_value,
|
||||
const Slice& key,
|
||||
const Slice* existing_value,
|
||||
const deque[string]& op_list,
|
||||
string* new_value,
|
||||
logger.Logger* log) with gil:
|
||||
|
@ -358,9 +358,9 @@ cdef cpp_bool full_merge_callback(
|
|||
|
||||
cdef cpp_bool partial_merge_callback(
|
||||
void* ctx,
|
||||
const slice_.Slice& key,
|
||||
const slice_.Slice& left_op,
|
||||
const slice_.Slice& right_op,
|
||||
const Slice& key,
|
||||
const Slice& left_op,
|
||||
const Slice& right_op,
|
||||
string* new_value,
|
||||
logger.Logger* log) with gil:
|
||||
|
||||
|
@ -966,65 +966,85 @@ cdef class DB(object):
|
|||
cdef db.DB* db
|
||||
|
||||
def __cinit__(self, db_name, Options opts, read_only=False):
|
||||
if read_only:
|
||||
check_status(
|
||||
db.DB_OpenForReadOnly(
|
||||
deref(opts.opts),
|
||||
path_to_string(db_name),
|
||||
cython.address(self.db),
|
||||
False))
|
||||
else:
|
||||
check_status(
|
||||
db.DB_Open(
|
||||
deref(opts.opts),
|
||||
path_to_string(db_name),
|
||||
cython.address(self.db)))
|
||||
cdef Status st
|
||||
cdef string db_path = path_to_string(db_name)
|
||||
|
||||
if read_only:
|
||||
with nogil:
|
||||
st = db.DB_OpenForReadOnly(
|
||||
deref(opts.opts),
|
||||
db_path,
|
||||
cython.address(self.db),
|
||||
False)
|
||||
else:
|
||||
with nogil:
|
||||
st = db.DB_Open(
|
||||
deref(opts.opts),
|
||||
db_path,
|
||||
cython.address(self.db))
|
||||
|
||||
check_status(st)
|
||||
self.opts = opts
|
||||
|
||||
def __dealloc__(self):
|
||||
del self.db
|
||||
|
||||
def put(self, key, value, sync=False, disable_wal=False):
|
||||
cdef Status st
|
||||
cdef options.WriteOptions opts
|
||||
opts.sync = sync
|
||||
opts.disableWAL = disable_wal
|
||||
|
||||
check_status(
|
||||
self.db.Put(opts, bytes_to_slice(key), bytes_to_slice(value)))
|
||||
cdef Slice c_key = bytes_to_slice(key)
|
||||
cdef Slice c_value = bytes_to_slice(value)
|
||||
|
||||
with nogil:
|
||||
st = self.db.Put(opts, c_key, c_value)
|
||||
check_status(st)
|
||||
|
||||
def delete(self, key, sync=False, disable_wal=False):
|
||||
cdef Status st
|
||||
cdef options.WriteOptions opts
|
||||
opts.sync = sync
|
||||
opts.disableWAL = disable_wal
|
||||
|
||||
check_status(
|
||||
self.db.Delete(opts, bytes_to_slice(key)))
|
||||
cdef Slice c_key = bytes_to_slice(key)
|
||||
with nogil:
|
||||
st = self.db.Delete(opts, c_key)
|
||||
check_status(st)
|
||||
|
||||
def merge(self, key, value, sync=False, disable_wal=False):
|
||||
cdef Status st
|
||||
cdef options.WriteOptions opts
|
||||
opts.sync = sync
|
||||
opts.disableWAL = disable_wal
|
||||
|
||||
check_status(
|
||||
self.db.Merge(opts, bytes_to_slice(key), bytes_to_slice(value)))
|
||||
cdef Slice c_key = bytes_to_slice(key)
|
||||
cdef Slice c_value = bytes_to_slice(value)
|
||||
with nogil:
|
||||
st = self.db.Merge(opts, c_key, c_value)
|
||||
check_status(st)
|
||||
|
||||
def write(self, WriteBatch batch, sync=False, disable_wal=False):
|
||||
cdef Status st
|
||||
cdef options.WriteOptions opts
|
||||
opts.sync = sync
|
||||
opts.disableWAL = disable_wal
|
||||
|
||||
check_status(
|
||||
self.db.Write(opts, batch.batch))
|
||||
with nogil:
|
||||
st = self.db.Write(opts, batch.batch)
|
||||
check_status(st)
|
||||
|
||||
def get(self, key, *args, **kwargs):
|
||||
cdef string res
|
||||
cdef Status st
|
||||
cdef options.ReadOptions opts
|
||||
|
||||
st = self.db.Get(
|
||||
self.build_read_opts(self.__parse_read_opts(*args, **kwargs)),
|
||||
bytes_to_slice(key),
|
||||
cython.address(res))
|
||||
opts = self.build_read_opts(self.__parse_read_opts(*args, **kwargs))
|
||||
cdef Slice c_key = bytes_to_slice(key)
|
||||
|
||||
with nogil:
|
||||
st = self.db.Get(opts, c_key, cython.address(res))
|
||||
|
||||
if st.ok():
|
||||
return string_to_bytes(res)
|
||||
|
@ -1037,14 +1057,19 @@ cdef class DB(object):
|
|||
cdef vector[string] values
|
||||
values.resize(len(keys))
|
||||
|
||||
cdef vector[slice_.Slice] c_keys
|
||||
cdef vector[Slice] c_keys
|
||||
for key in keys:
|
||||
c_keys.push_back(bytes_to_slice(key))
|
||||
|
||||
cdef vector[Status] res = self.db.MultiGet(
|
||||
self.build_read_opts(self.__parse_read_opts(*args, **kwargs)),
|
||||
c_keys,
|
||||
cython.address(values))
|
||||
cdef options.ReadOptions opts
|
||||
opts = self.build_read_opts(self.__parse_read_opts(*args, **kwargs))
|
||||
|
||||
cdef vector[Status] res
|
||||
with nogil:
|
||||
res = self.db.MultiGet(
|
||||
opts,
|
||||
c_keys,
|
||||
cython.address(values))
|
||||
|
||||
cdef dict ret_dict = {}
|
||||
for index in range(len(keys)):
|
||||
|
@ -1062,15 +1087,20 @@ cdef class DB(object):
|
|||
cdef cpp_bool value_found
|
||||
cdef cpp_bool exists
|
||||
cdef options.ReadOptions opts
|
||||
cdef Slice c_key
|
||||
opts = self.build_read_opts(self.__parse_read_opts(*args, **kwargs))
|
||||
|
||||
c_key = bytes_to_slice(key)
|
||||
exists = False
|
||||
|
||||
if fetch:
|
||||
value_found = False
|
||||
exists = self.db.KeyMayExist(
|
||||
opts,
|
||||
bytes_to_slice(key),
|
||||
cython.address(value),
|
||||
cython.address(value_found))
|
||||
with nogil:
|
||||
exists = self.db.KeyMayExist(
|
||||
opts,
|
||||
c_key,
|
||||
cython.address(value),
|
||||
cython.address(value_found))
|
||||
|
||||
if exists:
|
||||
if value_found:
|
||||
|
@ -1080,10 +1110,11 @@ cdef class DB(object):
|
|||
else:
|
||||
return (False, None)
|
||||
else:
|
||||
exists = self.db.KeyMayExist(
|
||||
opts,
|
||||
bytes_to_slice(key),
|
||||
cython.address(value))
|
||||
with nogil:
|
||||
exists = self.db.KeyMayExist(
|
||||
opts,
|
||||
c_key,
|
||||
cython.address(value))
|
||||
|
||||
return (exists, None)
|
||||
|
||||
|
@ -1092,7 +1123,8 @@ cdef class DB(object):
|
|||
cdef KeysIterator it
|
||||
opts = self.build_read_opts(self.__parse_read_opts(*args, **kwargs))
|
||||
it = KeysIterator(self)
|
||||
it.ptr = self.db.NewIterator(opts)
|
||||
with nogil:
|
||||
it.ptr = self.db.NewIterator(opts)
|
||||
return it
|
||||
|
||||
def itervalues(self, prefix=None, *args, **kwargs):
|
||||
|
@ -1100,7 +1132,8 @@ cdef class DB(object):
|
|||
cdef ValuesIterator it
|
||||
opts = self.build_read_opts(self.__parse_read_opts(*args, **kwargs))
|
||||
it = ValuesIterator(self)
|
||||
it.ptr = self.db.NewIterator(opts)
|
||||
with nogil:
|
||||
it.ptr = self.db.NewIterator(opts)
|
||||
return it
|
||||
|
||||
def iteritems(self, prefix=None, *args, **kwargs):
|
||||
|
@ -1108,7 +1141,8 @@ cdef class DB(object):
|
|||
cdef ItemsIterator it
|
||||
opts = self.build_read_opts(self.__parse_read_opts(*args, **kwargs))
|
||||
it = ItemsIterator(self)
|
||||
it.ptr = self.db.NewIterator(opts)
|
||||
with nogil:
|
||||
it.ptr = self.db.NewIterator(opts)
|
||||
return it
|
||||
|
||||
def snapshot(self):
|
||||
|
@ -1116,8 +1150,13 @@ cdef class DB(object):
|
|||
|
||||
def get_property(self, prop):
|
||||
cdef string value
|
||||
cdef Slice c_prop = bytes_to_slice(prop)
|
||||
cdef cpp_bool ret = False
|
||||
|
||||
if self.db.GetProperty(bytes_to_slice(prop), cython.address(value)):
|
||||
with nogil:
|
||||
ret = self.db.GetProperty(c_prop, cython.address(value))
|
||||
|
||||
if ret:
|
||||
return string_to_bytes(value)
|
||||
else:
|
||||
return None
|
||||
|
@ -1125,7 +1164,8 @@ cdef class DB(object):
|
|||
def get_live_files_metadata(self):
|
||||
cdef vector[db.LiveFileMetaData] metadata
|
||||
|
||||
self.db.GetLiveFilesMetaData(cython.address(metadata))
|
||||
with nogil:
|
||||
self.db.GetLiveFilesMetaData(cython.address(metadata))
|
||||
|
||||
ret = []
|
||||
for ob in metadata:
|
||||
|
@ -1181,10 +1221,12 @@ cdef class Snapshot(object):
|
|||
|
||||
def __cinit__(self, DB db):
|
||||
self.db = db
|
||||
self.ptr = db.db.GetSnapshot()
|
||||
with nogil:
|
||||
self.ptr = db.db.GetSnapshot()
|
||||
|
||||
def __dealloc__(self):
|
||||
self.db.db.ReleaseSnapshot(self.ptr)
|
||||
with nogil:
|
||||
self.db.db.ReleaseSnapshot(self.ptr)
|
||||
|
||||
|
||||
@cython.internal
|
||||
|
@ -1208,20 +1250,25 @@ cdef class BaseIterator(object):
|
|||
raise StopIteration()
|
||||
|
||||
cdef object ret = self.get_ob()
|
||||
self.ptr.Next()
|
||||
with nogil:
|
||||
self.ptr.Next()
|
||||
return ret
|
||||
|
||||
def __reversed__(self):
|
||||
return ReversedIterator(self)
|
||||
|
||||
cpdef seek_to_first(self):
|
||||
self.ptr.SeekToFirst()
|
||||
with nogil:
|
||||
self.ptr.SeekToFirst()
|
||||
|
||||
cpdef seek_to_last(self):
|
||||
self.ptr.SeekToLast()
|
||||
with nogil:
|
||||
self.ptr.SeekToLast()
|
||||
|
||||
cpdef seek(self, key):
|
||||
self.ptr.Seek(bytes_to_slice(key))
|
||||
cdef Slice c_key = bytes_to_slice(key)
|
||||
with nogil:
|
||||
self.ptr.Seek(c_key)
|
||||
|
||||
cdef object get_ob(self):
|
||||
return None
|
||||
|
@ -1229,17 +1276,28 @@ cdef class BaseIterator(object):
|
|||
@cython.internal
|
||||
cdef class KeysIterator(BaseIterator):
|
||||
cdef object get_ob(self):
|
||||
return slice_to_bytes(self.ptr.key())
|
||||
cdef Slice c_key
|
||||
with nogil:
|
||||
c_key = self.ptr.key()
|
||||
return slice_to_bytes(c_key)
|
||||
|
||||
@cython.internal
|
||||
cdef class ValuesIterator(BaseIterator):
|
||||
cdef object get_ob(self):
|
||||
return slice_to_bytes(self.ptr.value())
|
||||
cdef Slice c_value
|
||||
with nogil:
|
||||
c_value = self.ptr.value()
|
||||
return slice_to_bytes(c_value)
|
||||
|
||||
@cython.internal
|
||||
cdef class ItemsIterator(BaseIterator):
|
||||
cdef object get_ob(self):
|
||||
return (slice_to_bytes(self.ptr.key()), slice_to_bytes(self.ptr.value()))
|
||||
cdef Slice c_key
|
||||
cdef Slice c_value
|
||||
with nogil:
|
||||
c_key = self.ptr.key()
|
||||
c_value = self.ptr.value()
|
||||
return (slice_to_bytes(c_key), slice_to_bytes(c_value))
|
||||
|
||||
@cython.internal
|
||||
cdef class ReversedIterator(object):
|
||||
|
|
Loading…
Reference in a new issue