static void stat_callback(lcb_t instance,
const void *cookie,
lcb_error_t err,
const lcb_server_stat_resp_t *resp)
{
pycbc_MultiResultObject *mres;
PyObject *value;
PyObject *skey, *knodes;
mres = (pycbc_MultiResultObject*)cookie;
CB_THR_END(mres->parent);
if (err != LCB_SUCCESS) {
if (mres->errop == NULL) {
pycbc_ResultBaseObject *res =
(pycbc_ResultBaseObject*)pycbc_result_new(mres->parent);
res->rc = err;
res->key = Py_None; Py_INCREF(res->key);
maybe_push_operr(mres, res, err, 0);
}
CB_THR_BEGIN(mres->parent);
return;
}
if (!resp->v.v0.server_endpoint) {
CB_THR_BEGIN(mres->parent);
return;
}
skey = pycbc_SimpleStringN(resp->v.v0.key, resp->v.v0.nkey);
value = pycbc_SimpleStringN(resp->v.v0.bytes, resp->v.v0.nbytes);
{
PyObject *intval = pycbc_maybe_convert_to_int(value);
if (intval) {
Py_DECREF(value);
value = intval;
} else {
PyErr_Clear();
}
}
knodes = PyDict_GetItem((PyObject*)mres, skey);
if (!knodes) {
knodes = PyDict_New();
PyDict_SetItem((PyObject*)mres, skey, knodes);
Py_DECREF(knodes);
}
PyDict_SetItemString(knodes, resp->v.v0.server_endpoint, value);
Py_DECREF(skey);
Py_DECREF(value);
CB_THR_BEGIN(mres->parent);
(void)instance;
}
static int get_common_objects(PyObject *cookie,
const void *key,
size_t nkey,
lcb_error_t err,
pycbc_ConnectionObject **conn,
pycbc_ResultBaseObject **res,
int restype,
pycbc_MultiResultObject **mres)
{
PyObject *hkey;
int rv;
assert(Py_TYPE(cookie) == &pycbc_MultiResultType);
*mres = (pycbc_MultiResultObject*)cookie;
*conn = (*mres)->parent;
CB_THR_END(*conn);
rv = pycbc_tc_decode_key(*conn, key, nkey, &hkey);
if (rv < 0) {
push_fatal_error(*mres);
return -1;
}
/**
* Now, get/set the result object
*/
if (restype == RESTYPE_BASE) {
*res = (pycbc_ResultBaseObject*)pycbc_result_new(*conn);
} else if (restype == RESTYPE_OPERATION) {
*res = (pycbc_ResultBaseObject*)pycbc_opresult_new(*conn);
} else if (restype == RESTYPE_VALUE) {
*res = (pycbc_ResultBaseObject*)pycbc_valresult_new(*conn);
} else {
abort();
}
assert(PyDict_Contains((PyObject*)*mres, hkey) == 0);
PyDict_SetItem((PyObject*)*mres, hkey, (PyObject*)*res);
Py_DECREF(*res);
(*res)->key = hkey;
(*res)->rc = err;
if (err != LCB_SUCCESS) {
(*mres)->all_ok = 0;
}
return 0;
}
static void
stats_callback(lcb_t instance, int cbtype, const lcb_RESPBASE *resp_base)
{
pycbc_MultiResult *mres;
PyObject *value;
PyObject *skey, *knodes;
PyObject *mrdict;
pycbc_Bucket *parent;
const lcb_RESPSTATS *resp = (const lcb_RESPSTATS *)resp_base;
int do_return = 0;
mres = (pycbc_MultiResult*)resp->cookie;
parent = mres->parent;
CB_THR_END(parent);
if (resp->rc != LCB_SUCCESS) {
do_return = 1;
if (mres->errop == NULL) {
pycbc_Result *res = (pycbc_Result*)pycbc_result_new(parent);
res->rc = resp->rc;
res->key = Py_None; Py_INCREF(res->key);
maybe_push_operr(mres, res, resp->rc, 0);
}
}
if (resp->rflags & LCB_RESP_F_FINAL) {
/* Note this can happen in both success and error cases! */
do_return = 1;
operation_completed(parent, mres);
}
if (do_return) {
CB_THR_BEGIN(parent);
return;
}
skey = pycbc_SimpleStringN(resp->key, resp->nkey);
value = pycbc_SimpleStringN(resp->value, resp->nvalue);
{
PyObject *intval = pycbc_maybe_convert_to_int(value);
if (intval) {
Py_DECREF(value);
value = intval;
} else {
PyErr_Clear();
}
}
mrdict = pycbc_multiresult_dict(mres);
knodes = PyDict_GetItem(mrdict, skey);
if (!knodes) {
knodes = PyDict_New();
PyDict_SetItem(mrdict, skey, knodes);
}
PyDict_SetItemString(knodes, resp->server, value);
Py_DECREF(skey);
Py_DECREF(value);
CB_THR_BEGIN(parent);
(void)instance;
}
/**
* Call this function for each callback. Note that even if this function
* returns nonzero, CB_THR_BEGIN() must still be called, and the `conn`
* and `mres` out parameters are considered valid
* @param resp base response object
* @param[out] conn the bucket object
* @param[out] res the result object for the individual operation
* @param restype What type should `res` be if it needs to be created
* @param[out] mres the context for the current operation
* @return 0 if operation processing may proceed, nonzero if operation
* processing has completed. In both cases the `conn` and `mres` paramters
* are valid, however.
*/
static int
get_common_objects(const lcb_RESPBASE *resp, pycbc_Bucket **conn,
pycbc_Result **res, int restype, pycbc_MultiResult **mres)
{
PyObject *hkey;
PyObject *mrdict;
int rv;
pycbc_assert(pycbc_multiresult_check(resp->cookie));
*mres = (pycbc_MultiResult*)resp->cookie;
*conn = (*mres)->parent;
CB_THR_END(*conn);
rv = pycbc_tc_decode_key(*conn, resp->key, resp->nkey, &hkey);
if (rv < 0) {
pycbc_multiresult_adderr(*mres);
return -1;
}
mrdict = pycbc_multiresult_dict(*mres);
*res = (pycbc_Result*)PyDict_GetItem(mrdict, hkey);
if (*res) {
int exists_ok = (restype & RESTYPE_EXISTS_OK) ||
( (*mres)->mropts & PYCBC_MRES_F_UALLOCED);
if (!exists_ok) {
if ((*conn)->flags & PYCBC_CONN_F_WARNEXPLICIT) {
PyErr_WarnExplicit(PyExc_RuntimeWarning,
"Found duplicate key",
__FILE__, __LINE__,
"couchbase._libcouchbase",
NULL);
} else {
PyErr_WarnEx(PyExc_RuntimeWarning,
"Found duplicate key",
1);
}
/**
* We need to destroy the existing object and re-create it.
*/
PyDict_DelItem(mrdict, hkey);
*res = NULL;
} else {
Py_XDECREF(hkey);
}
}
if (*res == NULL) {
/* Now, get/set the result object */
if ( (*mres)->mropts & PYCBC_MRES_F_ITEMS) {
*res = (pycbc_Result*)pycbc_item_new(*conn);
} else if (restype & RESTYPE_BASE) {
*res = (pycbc_Result*)pycbc_result_new(*conn);
} else if (restype & RESTYPE_OPERATION) {
*res = (pycbc_Result*)pycbc_opresult_new(*conn);
} else if (restype & RESTYPE_VALUE) {
*res = (pycbc_Result*)pycbc_valresult_new(*conn);
} else {
abort();
}
PyDict_SetItem(mrdict, hkey, (PyObject*)*res);
(*res)->key = hkey;
Py_DECREF(*res);
}
if (resp->rc) {
(*res)->rc = resp->rc;
}
if (resp->rc != LCB_SUCCESS) {
(*mres)->all_ok = 0;
}
return 0;
}
static int
get_common_objects(PyObject *cookie,
const void *key,
size_t nkey,
lcb_error_t err,
pycbc_Connection **conn,
pycbc_Result **res,
int restype,
pycbc_MultiResult **mres)
{
PyObject *hkey;
int rv;
pycbc_assert(Py_TYPE(cookie) == &pycbc_MultiResultType);
*mres = (pycbc_MultiResult*)cookie;
*conn = (*mres)->parent;
if (!(restype & RESTYPE_VARCOUNT)) {
maybe_breakout(*conn);
}
CB_THR_END(*conn);
rv = pycbc_tc_decode_key(*conn, key, nkey, &hkey);
if (rv < 0) {
push_fatal_error(*mres);
return -1;
}
*res = (pycbc_Result*)PyDict_GetItem((PyObject*)*mres, hkey);
if (*res) {
int exists_ok = (restype & RESTYPE_EXISTS_OK) ||
( (*mres)->mropts & PYCBC_MRES_F_UALLOCED);
if (!exists_ok) {
if ((*conn)->flags & PYCBC_CONN_F_WARNEXPLICIT) {
PyErr_WarnExplicit(PyExc_RuntimeWarning,
"Found duplicate key",
__FILE__, __LINE__,
"couchbase._libcouchbase",
NULL);
} else {
PyErr_WarnEx(PyExc_RuntimeWarning,
"Found duplicate key",
1);
}
/**
* We need to destroy the existing object and re-create it.
*/
PyDict_DelItem((PyObject*)*mres, hkey);
*res = NULL;
} else {
Py_XDECREF(hkey);
}
}
if (*res == NULL) {
/**
* Now, get/set the result object
*/
if ( (*mres)->mropts & PYCBC_MRES_F_ITEMS) {
*res = (pycbc_Result*)pycbc_item_new(*conn);
} else if (restype & RESTYPE_BASE) {
*res = (pycbc_Result*)pycbc_result_new(*conn);
} else if (restype & RESTYPE_OPERATION) {
*res = (pycbc_Result*)pycbc_opresult_new(*conn);
} else if (restype & RESTYPE_VALUE) {
*res = (pycbc_Result*)pycbc_valresult_new(*conn);
} else {
abort();
}
PyDict_SetItem((PyObject*)*mres, hkey, (PyObject*)*res);
(*res)->key = hkey;
Py_DECREF(*res);
}
if (err) {
(*res)->rc = err;
}
if (err != LCB_SUCCESS) {
(*mres)->all_ok = 0;
}
return 0;
}
