LIBCOUCHBASE_API
libcouchbase_error_t libcouchbase_mget_by_key(libcouchbase_t instance,
const void *command_cookie,
const void *hashkey,
libcouchbase_size_t nhashkey,
libcouchbase_size_t num_keys,
const void * const *keys,
const libcouchbase_size_t *nkey,
const libcouchbase_time_t *exp)
{
libcouchbase_server_t *server = NULL;
protocol_binary_request_noop noop;
libcouchbase_size_t ii, *affected_servers = NULL;
int vb, idx;
struct server_info_st *servers = NULL;
/* we need a vbucket config before we can start getting data.. */
if (instance->vbucket_config == NULL) {
return libcouchbase_synchandler_return(instance, LIBCOUCHBASE_ETMPFAIL);
}
if (num_keys == 1) {
return libcouchbase_single_get(instance, command_cookie, hashkey,
nhashkey, keys[0], nkey[0], exp);
}
affected_servers = calloc(instance->nservers, sizeof(libcouchbase_size_t));
if (affected_servers == NULL) {
return libcouchbase_synchandler_return(instance, LIBCOUCHBASE_ENOMEM);
}
if (nhashkey != 0) {
(void)vbucket_map(instance->vbucket_config, hashkey, nhashkey, &vb, &idx);
if (idx < 0 || (libcouchbase_size_t)idx > instance->nservers) {
/* the config says that there is no server yet at that position (-1) */
return libcouchbase_synchandler_return(instance, LIBCOUCHBASE_NETWORK_ERROR);
}
server = instance->servers + (libcouchbase_size_t)idx;
affected_servers[idx]++;
} else {
servers = malloc(num_keys * sizeof(struct server_info_st));
if (servers == NULL) {
return libcouchbase_synchandler_return(instance, LIBCOUCHBASE_ENOMEM);
}
for (ii = 0; ii < num_keys; ++ii) {
(void)vbucket_map(instance->vbucket_config, keys[ii], nkey[ii], &servers[ii].vb, &servers[ii].idx);
if (servers[ii].idx < 0 || (libcouchbase_size_t)servers[ii].idx > instance->nservers) {
/* the config says that there is no server yet at that position (-1) */
free(servers);
return libcouchbase_synchandler_return(instance, LIBCOUCHBASE_NETWORK_ERROR);
}
affected_servers[servers[ii].idx]++;
}
}
for (ii = 0; ii < num_keys; ++ii) {
protocol_binary_request_gat req;
if (nhashkey == 0) {
server = instance->servers + (libcouchbase_size_t)servers[ii].idx;
vb = servers[ii].vb;
}
memset(&req, 0, sizeof(req));
req.message.header.request.magic = PROTOCOL_BINARY_REQ;
req.message.header.request.keylen = ntohs((libcouchbase_uint16_t)nkey[ii]);
req.message.header.request.datatype = PROTOCOL_BINARY_RAW_BYTES;
req.message.header.request.vbucket = ntohs((libcouchbase_uint16_t)vb);
req.message.header.request.bodylen = ntohl((libcouchbase_uint32_t)(nkey[ii]));
req.message.header.request.opaque = ++instance->seqno;
if (!exp) {
req.message.header.request.opcode = PROTOCOL_BINARY_CMD_GETQ;
libcouchbase_server_start_packet(server, command_cookie, req.bytes,
sizeof(req.bytes) - 4);
} else {
req.message.header.request.opcode = PROTOCOL_BINARY_CMD_GATQ;
req.message.header.request.extlen = 4;
req.message.body.expiration = ntohl((libcouchbase_uint32_t)exp[ii]);
req.message.header.request.bodylen = ntohl((libcouchbase_uint32_t)(nkey[ii]) + 4);
libcouchbase_server_start_packet(server, command_cookie, req.bytes,
sizeof(req.bytes));
}
libcouchbase_server_write_packet(server, keys[ii], nkey[ii]);
libcouchbase_server_end_packet(server);
}
free(servers);
memset(&noop, 0, sizeof(noop));
noop.message.header.request.magic = PROTOCOL_BINARY_REQ;
noop.message.header.request.opcode = PROTOCOL_BINARY_CMD_NOOP;
noop.message.header.request.datatype = PROTOCOL_BINARY_RAW_BYTES;
/*
** We don't know which server we sent the data to, so examine
** where to send the noop
*/
for (ii = 0; ii < instance->nservers; ++ii) {
if (affected_servers[ii]) {
server = instance->servers + ii;
noop.message.header.request.opaque = ++instance->seqno;
libcouchbase_server_complete_packet(server, command_cookie,
noop.bytes, sizeof(noop.bytes));
libcouchbase_server_send_packets(server);
}
}
free(affected_servers);
return libcouchbase_synchandler_return(instance, LIBCOUCHBASE_SUCCESS);
}
LIBCOUCHBASE_API
libcouchbase_error_t libcouchbase_mget_by_key(libcouchbase_t instance,
const void *hashkey,
size_t nhashkey,
size_t num_keys,
const void * const *keys,
const size_t *nkey,
const time_t *exp)
{
uint16_t vb;
libcouchbase_server_t *server;
protocol_binary_request_noop noop;
size_t ii;
// we need a vbucket config before we can start getting data..
libcouchbase_ensure_vbucket_config(instance);
assert(instance->vbucket_config);
if (nhashkey != 0) {
vb = (uint16_t)vbucket_get_vbucket_by_key(instance->vbucket_config,
hashkey, nhashkey);
server = instance->servers + instance->vb_server_map[vb];
}
for (ii = 0; ii < num_keys; ++ii) {
protocol_binary_request_gat req;
if (nhashkey == 0) {
vb = (uint16_t)vbucket_get_vbucket_by_key(instance->vbucket_config,
keys[ii], nkey[ii]);
server = instance->servers + instance->vb_server_map[vb];
}
memset(&req, 0, sizeof(req));
req.message.header.request.magic = PROTOCOL_BINARY_REQ;
req.message.header.request.keylen = ntohs((uint16_t)nkey[ii]);
req.message.header.request.datatype = PROTOCOL_BINARY_RAW_BYTES;
req.message.header.request.vbucket = ntohs(vb);
req.message.header.request.bodylen = ntohl((uint32_t)(nkey[ii]));
req.message.header.request.opaque = ++instance->seqno;
if (!exp) {
req.message.header.request.opcode = PROTOCOL_BINARY_CMD_GETQ;
libcouchbase_server_start_packet(server, req.bytes,
sizeof(req.bytes) - 4);
} else {
req.message.header.request.opcode = PROTOCOL_BINARY_CMD_GATQ;
req.message.header.request.extlen = 4;
req.message.body.expiration = ntohl((uint32_t)exp[ii]);
libcouchbase_server_start_packet(server, req.bytes,
sizeof(req.bytes));
}
libcouchbase_server_write_packet(server, keys[ii], nkey[ii]);
libcouchbase_server_end_packet(server);
}
memset(&noop, 0, sizeof(noop));
noop.message.header.request.magic = PROTOCOL_BINARY_REQ;
noop.message.header.request.opcode = PROTOCOL_BINARY_CMD_NOOP;
noop.message.header.request.datatype = PROTOCOL_BINARY_RAW_BYTES;
if (nhashkey == 0) {
// We don't know which server we sent the data to, so examine
// where to send the noop
for (ii = 0; ii < instance->nservers; ++ii) {
server = instance->servers + ii;
if (server->output.avail > 0 || server->pending.avail > 0) {
noop.message.header.request.opaque = ++instance->seqno;
libcouchbase_server_complete_packet(server, noop.bytes,
sizeof(noop.bytes));
libcouchbase_server_send_packets(server);
}
}
} else {
noop.message.header.request.opaque = ++instance->seqno;
libcouchbase_server_complete_packet(server, noop.bytes,
sizeof(noop.bytes));
libcouchbase_server_send_packets(server);
}
return LIBCOUCHBASE_SUCCESS;
}
