def _stats_callback(self, instance, cbtype, resp):
resp = ffi.cast('lcb_RESPSTATS*', resp)
mres = ffi.from_handle(resp.cookie)
if resp.rc:
r = ValueResult()
r.key = '__dummy__'
mres._add_bad_rc(resp.rc, r)
if resp.key == ffi.NULL and resp.server == ffi.NULL:
self._chk_op_done(mres)
return
if resp.rc:
return
kbuf = bytes(ffi.buffer(resp.key, resp.nkey))
key = self._tc.decode_key(kbuf)
if resp.nvalue:
value = from_cstring(resp.value, resp.nvalue)
try:
value = int(value)
except ValueError:
pass
else:
value = None
server = from_cstring(resp.server)
mres.setdefault(key, {})[server] = value
def _on_single_row(self, instance, cbtype, resp):
mres = ffi.from_handle(resp.cookie)
if resp.rflags & C.LCB_RESP_F_FINAL:
self._handle_done(resp, mres)
return
if resp.rc != C.LCB_SUCCESS:
mres._add_bad_rc(resp.rc, self)
return
row = {}
if resp.nkey:
row['key'] = PyCBC.json_decode(buf2str(resp.key, resp.nkey))
if resp.nvalue:
row['value'] = PyCBC.json_decode(buf2str(resp.value, resp.nvalue))
if resp.docid:
# Document ID is always a simple string, so no need to decode
row['id'] = buf2str(resp.docid, resp.ndocid)
if resp.docresp:
py_doc = ValueResult()
l_doc = resp.docresp
row['__DOCRESULT__'] = py_doc
py_doc.key = row['id']
py_doc.flags = l_doc.itmflags
py_doc.cas = l_doc.cas
py_doc.rc = l_doc.rc
if not resp.docresp.rc:
buf = bytes(ffi.buffer(l_doc.value, l_doc.nvalue))
try:
tc = self._parent._tc
py_doc.value = tc.decode_value(buf, py_doc.flags)
except:
py_doc.value = buf[::]
# So now that we have a row..
self.rows.append(row)
if self._parent._is_async:
self._invoke_async(mres)
def make_result(self, key, value):
vr = ValueResult()
vr.key = key
vr.value = []
return vr
def make_result(self, key, _):
vr = ValueResult()
vr.key = key
return vr
