PyObject *
pycbc_Bucket__fts_query(pycbc_Bucket *self, PyObject *args, PyObject *kwargs)
{
int rv;
PyObject *ret = NULL;
pycbc_MultiResult *mres;
pycbc_ViewResult *vres;
lcb_error_t rc;
lcb_CMDFTS cmd = { 0 };
const char *params;
pycbc_strlen_t nparams;
static char *kwlist[] = { "params", NULL };
rv = PyArg_ParseTupleAndKeywords(args, kwargs,
"s#", kwlist, ¶ms, &nparams);
if (!rv) {
PYCBC_EXCTHROW_ARGS();
return NULL;
}
if (-1 == pycbc_oputil_conn_lock(self)) {
return NULL;
}
if (self->pipeline_queue) {
PYCBC_EXC_WRAP(PYCBC_EXC_PIPELINE, 0,
"FTS queries cannot be executed in pipeline context");
}
mres = (pycbc_MultiResult *)pycbc_multiresult_new(self);
vres = (pycbc_ViewResult *)PYCBC_TYPE_CTOR(&pycbc_ViewResultType);
pycbc_httpresult_init(&vres->base, mres);
vres->rows = PyList_New(0);
vres->base.format = PYCBC_FMT_JSON;
vres->base.htype = PYCBC_HTTP_HN1QL;
cmd.callback = fts_row_callback;
cmd.query = params;
cmd.nquery = nparams;
cmd.handle = &vres->base.u.fts;
rc = lcb_fts_query(self->instance, mres, &cmd);
if (rc != LCB_SUCCESS) {
PYCBC_EXC_WRAP(PYCBC_EXC_LCBERR, rc, "Couldn't schedule fts query");
goto GT_DONE;
}
ret = (PyObject *)mres;
mres = NULL;
GT_DONE:
Py_XDECREF(mres);
pycbc_oputil_conn_unlock(self);
return ret;
}
PyObject *
pycbc_Bucket__view_request(pycbc_Bucket *self, PyObject *args, PyObject *kwargs)
{
int rv;
PyObject *ret = NULL;
pycbc_MultiResult *mres = NULL;
pycbc_ViewResult *vres = NULL;
lcb_CMDVIEWQUERY vcmd = { 0 };
viewpath_st vp = { NULL };
lcb_error_t rc;
const char *view = NULL, *design = NULL;
PyObject *options = NULL;
int flags;
static char *kwlist[] = { "design", "view", "options", "_flags", NULL };
rv = PyArg_ParseTupleAndKeywords(args, kwargs, "ss|Oi", kwlist,
&design, &view, &options, &flags);
if (!rv) {
PYCBC_EXCTHROW_ARGS();
return NULL;
}
if (-1 == pycbc_oputil_conn_lock(self)) {
return NULL;
}
if (self->pipeline_queue) {
PYCBC_EXC_WRAP(PYCBC_EXC_PIPELINE, 0,
"HTTP/View Requests cannot be executed in "
"pipeline context");
goto GT_DONE;
}
mres = (pycbc_MultiResult *)pycbc_multiresult_new(self);
vres = (pycbc_ViewResult *)PYCBC_TYPE_CTOR(&pycbc_ViewResultType);
vres->base.htype = PYCBC_HTTP_HVIEW;
pycbc_httpresult_init(&vres->base, mres);
rv = get_viewpath_str(self, &vp, options);
if (rv != 0) {
goto GT_DONE;
}
vcmd.ddoc = design;
vcmd.nddoc = strlen(design);
vcmd.view = view;
vcmd.nview = strlen(view);
vcmd.optstr = vp.optstr;
vcmd.noptstr = vp.noptstr;
vcmd.postdata = vp.body;
vcmd.npostdata = vp.nbody;
vcmd.handle = &vres->base.u.vh;
vcmd.callback = row_callback;
vcmd.cmdflags = flags;
vres->rows = PyList_New(0);
vres->base.format = PYCBC_FMT_JSON;
rc = lcb_view_query(self->instance, mres, &vcmd);
if (rc != LCB_SUCCESS) {
PYCBC_EXC_WRAP(PYCBC_EXC_LCBERR, rc, "Couldn't schedule view");
goto GT_DONE;
}
ret = (PyObject*)mres;
mres = NULL; /* Avoid GT_DONE decref */
GT_DONE:
Py_XDECREF(mres);
Py_XDECREF(vp.bk);
pycbc_oputil_conn_unlock(self);
return ret;
}
PyObject *
pycbc_Connection__stats(pycbc_Connection *self,
PyObject *args,
PyObject *kwargs)
{
int rv;
int ii;
Py_ssize_t ncmds;
lcb_error_t err;
PyObject *keys = NULL;
PyObject *ret = NULL;
pycbc_MultiResult *mres = NULL;
struct pycbc_common_vars cv = PYCBC_COMMON_VARS_STATIC_INIT;
static char *kwlist[] = { "keys", NULL };
rv = PyArg_ParseTupleAndKeywords(args, kwargs, "|O", kwlist, &keys);
if (!rv) {
PYCBC_EXCTHROW_ARGS();
return NULL;
}
if (keys == NULL || PyObject_IsTrue(keys) == 0) {
keys = NULL;
ncmds = 1;
} else {
if (!PySequence_Check(keys)) {
PYCBC_EXC_WRAP(PYCBC_EXC_ARGUMENTS, 0, "keys argument must be sequence");
return NULL;
}
ncmds = PySequence_Size(keys);
}
rv = pycbc_common_vars_init(&cv, ncmds, sizeof(lcb_server_stats_cmd_t), 0);
if (rv < 0) {
return NULL;
}
if (keys) {
for (ii =0; ii < ncmds; ii++) {
char *key;
Py_ssize_t nkey;
PyObject *newkey = NULL;
PyObject *curkey = PySequence_GetItem(keys, ii);
lcb_server_stats_cmd_t *cmd = cv.cmds.stats + ii;
rv = pycbc_BufFromString(curkey, &key, &nkey, &newkey);
if (rv < 0) {
PYCBC_EXC_WRAP_KEY(PYCBC_EXC_ARGUMENTS,
0,
"bad key type in stats",
curkey);
goto GT_DONE;
}
cmd->v.v0.name = key;
cmd->v.v0.nname = nkey;
cv.cmdlist.stats[ii] = cmd;
cv.enckeys[ii] = newkey;
}
} else {
cv.cmdlist.stats[0] = cv.cmds.stats;
}
mres = (pycbc_MultiResult*)pycbc_multiresult_new(self);
err = lcb_server_stats(self->instance, mres, ncmds, cv.cmdlist.stats);
if (err != LCB_SUCCESS) {
PYCBC_EXCTHROW_SCHED(err);
goto GT_DONE;
}
err = pycbc_oputil_wait_common(self);
if (err != LCB_SUCCESS) {
PYCBC_EXCTHROW_WAIT(err);
goto GT_DONE;
}
ret = (PyObject*)mres;
GT_DONE:
pycbc_common_vars_free(&cv);
/* Force multi, it's always a MultiResult */
pycbc_make_retval(PYCBC_ARGOPT_MULTI, &ret, &mres);
Py_XDECREF(mres);
return ret;
}
PyObject *
pycbc_Bucket__n1ql_query(pycbc_Bucket *self, PyObject *args, PyObject *kwargs)
{
int rv;
PyObject *ret = NULL;
pycbc_MultiResult *mres;
pycbc_ViewResult *vres;
lcb_error_t rc;
lcb_CMDN1QL cmd = { 0 };
const char *params;
pycbc_strlen_t nparams;
int prepared = 0, cross_bucket = 0;
static char *kwlist[] = { "params", "prepare", "cross_bucket", NULL };
rv = PyArg_ParseTupleAndKeywords(
args, kwargs, "s#|ii", kwlist, ¶ms, &nparams, &prepared, &cross_bucket);
if (!rv) {
PYCBC_EXCTHROW_ARGS();
return NULL;
}
if (-1 == pycbc_oputil_conn_lock(self)) {
return NULL;
}
if (self->pipeline_queue) {
PYCBC_EXC_WRAP(PYCBC_EXC_PIPELINE, 0,
"N1QL queries cannot be executed in "
"pipeline context");
}
mres = (pycbc_MultiResult *)pycbc_multiresult_new(self);
vres = (pycbc_ViewResult *)PYCBC_TYPE_CTOR(&pycbc_ViewResultType);
pycbc_httpresult_init(&vres->base, mres);
vres->rows = PyList_New(0);
vres->base.format = PYCBC_FMT_JSON;
vres->base.htype = PYCBC_HTTP_HN1QL;
cmd.content_type = "application/json";
cmd.callback = n1ql_row_callback;
cmd.query = params;
cmd.nquery = nparams;
cmd.handle = &vres->base.u.nq;
if (prepared) {
cmd.cmdflags |= LCB_CMDN1QL_F_PREPCACHE;
}
if (cross_bucket) {
cmd.cmdflags |= LCB_CMD_F_MULTIAUTH;
}
rc = lcb_n1ql_query(self->instance, mres, &cmd);
if (rc != LCB_SUCCESS) {
PYCBC_EXC_WRAP(PYCBC_EXC_LCBERR, rc, "Couldn't schedule n1ql query");
goto GT_DONE;
}
ret = (PyObject *)mres;
mres = NULL;
GT_DONE:
Py_XDECREF(mres);
pycbc_oputil_conn_unlock(self);
return ret;
}
static PyObject *
keyop_common(pycbc_Connection *self,
PyObject *args,
PyObject *kwargs,
int optype,
int argopts)
{
int rv;
int ii;
Py_ssize_t ncmds = 0;
pycbc_seqtype_t seqtype;
PyObject *casobj = NULL;
PyObject *ret = NULL;
PyObject *is_quiet = NULL;
PyObject *kobj = NULL;
pycbc_MultiResult *mres = NULL;
lcb_error_t err;
struct pycbc_common_vars cv = PYCBC_COMMON_VARS_STATIC_INIT;
static char *kwlist[] = { "keys", "cas", "quiet", NULL };
rv = PyArg_ParseTupleAndKeywords(args,
kwargs,
"O|OO",
kwlist,
&kobj,
&casobj,
&is_quiet);
if (!rv) {
PYCBC_EXCTHROW_ARGS();
return NULL;
}
if (argopts & PYCBC_ARGOPT_MULTI) {
rv = pycbc_oputil_check_sequence(kobj, 1, &ncmds, &seqtype);
if (rv < 0) {
return NULL;
}
if (casobj && PyObject_IsTrue(casobj)) {
PYCBC_EXC_WRAP(PYCBC_EXC_ARGUMENTS, 0, "Can't pass CAS for multiple keys");
}
} else {
ncmds = 1;
}
rv = pycbc_common_vars_init(&cv, ncmds, sizeof(lcb_remove_cmd_t), 0);
if (rv < 0) {
return NULL;
}
if (argopts & PYCBC_ARGOPT_MULTI) {
Py_ssize_t dictpos = 0;
PyObject *curseq, *iter = NULL;
curseq = pycbc_oputil_iter_prepare(seqtype, kobj, &iter, &dictpos);
if (!curseq) {
goto GT_DONE;
}
for (ii = 0; ii < ncmds; ii++) {
PyObject *curkey = NULL, *curvalue = NULL;
rv = pycbc_oputil_sequence_next(seqtype,
curseq,
&dictpos,
ii,
&curkey,
&curvalue);
if (rv < 0) {
goto GT_ITER_DONE;
}
rv = handle_single_keyop(self, curkey, curvalue, ii, optype, &cv);
Py_XDECREF(curkey);
Py_XDECREF(curvalue);
if (rv < 0) {
goto GT_ITER_DONE;
}
}
GT_ITER_DONE:
Py_XDECREF(iter);
if (rv < 0) {
goto GT_DONE;
}
} else {
rv = handle_single_keyop(self, kobj, casobj, 0, optype, &cv);
if (rv < 0) {
goto GT_DONE;
}
}
mres = (pycbc_MultiResult*)pycbc_multiresult_new(self);
if (optype == PYCBC_CMD_DELETE) {
if (pycbc_maybe_set_quiet(mres, is_quiet) == -1) {
goto GT_DONE;
}
err = lcb_remove(self->instance, mres, ncmds, cv.cmdlist.remove);
} else {
err = lcb_unlock(self->instance, mres, ncmds, cv.cmdlist.unlock);
}
if (err != LCB_SUCCESS) {
PYCBC_EXCTHROW_SCHED(err);
goto GT_DONE;
}
err = pycbc_oputil_wait_common(self);
if (err != LCB_SUCCESS) {
PYCBC_EXCTHROW_WAIT(err);
goto GT_DONE;
}
if (!pycbc_multiresult_maybe_raise(mres)) {
ret = (PyObject*)mres;
}
GT_DONE:
pycbc_common_vars_free(&cv);
ret = pycbc_make_retval(argopts, &ret, &mres);
Py_XDECREF(mres);
return ret;
}
