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; }