This commit is contained in:
Philipp Moritz committed 2017-07-19 10:06:32 -07:00
1 parent 9044a01627
commit 5178ee7513
2 files changed
+12

No files matched your search

+11
View File
@@ -7,6 +7,9 @@ from libcpp.memory cimport shared_ptr, unique_ptr, make_shared
from libcpp.string cimport string as c_string
from libc.stdint cimport int64_t, uint8_t
from pyarrow.lib cimport Buffer
from pyarrow.includes.libarrow cimport CBuffer
cdef extern from "arrow/api.h" namespace "arrow" nogil:
# We can later add more of the common status factory methods as needed
cdef CStatus CStatus_OK "Status::OK"()
@@ -70,7 +73,15 @@ cdef class PlasmaClient:
def create(self, ObjectID object_id, data_size):
cdef uint8_t* data
cdef shared_ptr[CBuffer] buffer
self.client.get().Create(object_id.data, data_size, NULL, 0, &data)
buffer.reset(new CBuffer(data, data_size))
result = Buffer()
result.init(buffer)
return result
def seal(self, ObjectID object_id):
self.client.get().Seal(object_id)
def disconnect(self):
self.client.get().Disconnect()
+1
View File
@@ -148,6 +148,7 @@ cdef extern from "arrow/api.h" namespace "arrow" nogil:
CLoggingMemoryPool(CMemoryPool*)
cdef cppclass CBuffer" arrow::Buffer":
CBuffer(const uint8_t* data, int64_t size)
uint8_t* data()
int64_t size()
shared_ptr[CBuffer] parent()