RepoMirrors/ceph
1
0
Fork 0
mirror of https://github.com/ceph/ceph synced 2025-02-22 10:37:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

pybind/rados: add WriteOp::writesame() and test WriteOp::writesame()

Browse Source 
Signed-off-by: Zhang Jiao <zhangjiao@cmss.chinamobile.com>
This commit is contained in:
zhangjiao 2019-12-03 12:21:57 +08:00
parent 8d2e311f47
commit 65cbfc8508
2 changed files with 26 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/pybind/rados/rados.pyx
View File

@ -306,6 +306,7 @@ cdef extern from "rados/librados.h" nogil:
void rados_write_op_truncate(rados_write_op_t write_op, uint64_t offset)
void rados_write_op_zero(rados_write_op_t write_op, uint64_t offset, uint64_t len)
void rados_write_op_exec(rados_write_op_t write_op, const char *cls, const char *method, const char *in_buf, size_t in_len, int *prval)
void rados_write_op_writesame(rados_write_op_t write_op, const char *buffer, size_t data_len, size_t write_len, uint64_t offset)
void rados_read_op_omap_get_vals2(rados_read_op_t read_op, const char * start_after, const char * filter_prefix, uint64_t max_return, rados_omap_iter_t * iter, unsigned char *pmore, int * prval)
void rados_read_op_omap_get_keys2(rados_read_op_t read_op, const char * start_after, uint64_t max_return, rados_omap_iter_t * iter, unsigned char *pmore, int * prval)
void rados_read_op_omap_get_vals_by_keys(rados_read_op_t read_op, const char * const* keys, size_t keys_len, rados_omap_iter_t * iter, int * prval)
@ -2164,6 +2165,25 @@ cdef class WriteOp(object):
with nogil:
rados_write_op_exec(self.write_op, _cls, _method, _data, _data_len, NULL)
@requires(('to_write', bytes), ('write_len', int), ('offset', int))
def writesame(self, to_write, write_len, offset=0):
"""
Write the same buffer multiple times
:param to_write: data to write
:type to_write: bytes
:param write_len: total number of bytes to write
:type len: int
:param offset: byte offset in the object to begin writing at
:type offset: int
"""
cdef:
char *_to_write = to_write
size_t _data_len = len(to_write)
size_t _write_len = write_len
uint64_t _offset = offset
with nogil:
rados_write_op_writesame(self.write_op, _to_write, _data_len, _write_len, _offset)
class WriteOpCtx(WriteOp, OpCtx):
"""write operation context manager"""

6
src/test/pybind/test_rados.py
View File

@ -538,6 +538,12 @@ class TestIoctx(object):
self.ioctx.operate_write_op(write_op, "object")
eq(self.ioctx.read('object'), b"Hello, ebs!")
def test_writesame_op(self):
with WriteOpCtx() as write_op:
write_op.writesame(b'rzx', 9)
self.ioctx.operate_write_op(write_op, 'abc')
eq(self.ioctx.read('abc'), b'rzxrzxrzx')
def test_get_omap_vals_by_keys(self):
keys = ("1", "2", "3", "4")
values = (b"aaa", b"bbb", b"ccc", b"\x04\x04\x04\x04")