def search(self, index, query, **params):
if not self.pb_search():
return self._search_mapred_emu(index, query)
req = riak_pb.RpbSearchQueryReq(index=index, q=query)
if 'rows' in params:
req.rows = params['rows']
if 'start' in params:
req.start = params['start']
if 'sort' in params:
req.sort = params['sort']
if 'filter' in params:
req.filter = params['filter']
if 'df' in params:
req.df = params['df']
if 'op' in params:
req.op = params['op']
if 'q.op' in params:
req.op = params['q.op']
if 'fl' in params:
if isinstance(params['fl'], list):
req.fl.extend(params['fl'])
else:
req.fl.append(params['fl'])
if 'presort' in params:
req.presort = params['presort']
msg_code, resp = self._request(MSG_CODE_SEARCH_QUERY_REQ, req,
MSG_CODE_SEARCH_QUERY_RESP)
result = {}
if resp.HasField('max_score'):
result['max_score'] = resp.max_score
if resp.HasField('num_found'):
result['num_found'] = resp.num_found
docs = []
for doc in resp.docs:
resultdoc = {}
for pair in doc.fields:
ukey = unicode(pair.key, 'utf-8')
uval = unicode(pair.value, 'utf-8')
resultdoc[ukey] = uval
docs.append(resultdoc)
result['docs'] = docs
return result
def search(self, index, query, **params):
if not self.pb_search():
return self._search_mapred_emu(index, query)
if isinstance(query, unicode):
query = query.encode('utf8')
req = riak_pb.RpbSearchQueryReq(index=index, q=query)
self._encode_search_query(req, params)
msg_code, resp = self._request(MSG_CODE_SEARCH_QUERY_REQ, req,
MSG_CODE_SEARCH_QUERY_RESP)
result = {}
if resp.HasField('max_score'):
result['max_score'] = resp.max_score
if resp.HasField('num_found'):
result['num_found'] = resp.num_found
result['docs'] = [self._decode_search_doc(doc) for doc in resp.docs]
return result