module.exports = function(max_key, callback) {
var cluster = new Riak.Cluster.Builder()
.withNodeManager(single_node_manager)
.withExecutionAttmpts(1000)
.withRiakNodes([new Riak.Node({ remoteAddress: "127.0.0.1", remotePort: "8087"}) ])
.withQueueCommands(1e9)
.build();
var client = new Riak.Client(cluster);
var r = {};
r.read_one_random_record = function(next, old_key) {
var key = old_key;
if (typeof old_key == "undefined") {
key = "key" + Math.floor((Math.random() * max_key) + 1);
}
client.fetchValue({ bucket: "storage_test", key: key }, function(err) {
count++;
if (err) {
process.nextTick(function() { r.read_one_random_record(next, key); });
return;
}
next();
});
};
r.teardown = function() {
client.shutdown(function() {});
};
client.ping(function (err) {
callback(err, r);
})
};
function (async_cb) {
client.storeValue({
bucket: orderSummariesBucket,
key: orderSummary.customerId,
value: orderSummary
}, async_cb);
}
client.secondaryIndexQuery({ bucket: ordersBucket, indexName: 'SalespersonId_int', indexKey: 9000 }, function (err, rslt) {
checkErr(err);
if (rslt.values.length > 0) {
Array.prototype.push.apply(janesOrders,
rslt.values.map(function (value) {
return value.objectKey;
})
);
}
if (rslt.done) {
logger.info("Jane's Orders: %s", janesOrders.join(', '));
var octoberOrders = [];
client.secondaryIndexQuery({ bucket: ordersBucket, indexName: 'OrderDate_bin',
rangeStart: '2013-10-01', rangeEnd: '2013-10-31' }, function (err, rslt) {
checkErr(err);
if (rslt.values.length > 0) {
Array.prototype.push.apply(octoberOrders,
rslt.values.map(function (value) {
return value.objectKey;
})
);
}
if (rslt.done) {
logger.info("October's Orders: %s", octoberOrders.join(', '));
client_shutdown();
}
});
}
});
function client_shutdown() {
client.shutdown(function (state) {
if (state === Riak.Cluster.State.SHUTDOWN) {
logger.info("cluster stopped");
process.exit();
}
});
}
client.fetchValue({ bucket: ordersBucket, key : '3' }, function (err, rslt) {
checkErr(err);
var fetchedOrderObj = rslt.values.shift();
fetchedOrderObj.addToIndex('SalespersonId_int', 9000);
fetchedOrderObj.addToIndex('OrderDate_bin', '2013-11-03');
client.storeValue({ value: fetchedOrderObj }, function (err, rslt) {
async_cb(err, rslt);
});
});
client.fetchValue({ bucket: customersBucket, key: key }, function (err, rslt) {
checkErr(err);
var fetchedCustomerObj = rslt.values.shift();
var fetchedCustomer = fetchedCustomerObj.value.toString("utf8");
client.fetchValue({ bucket: orderSummariesBucket, key: key}, function (err, rslt) {
checkErr(err);
var fetchedOrderSummaryObj = rslt.values.shift();
var fetchedOrderSummary = fetchedOrderSummaryObj.value.toString("utf8");
logger.info("Customer 1: %s", fetchedCustomer);
logger.info("OrderSummary 1: %s", fetchedOrderSummary);
add_index_data();
});
});
r.read_one_random_record = function(next, old_key) {
var key = old_key;
if (typeof old_key == "undefined") {
key = "key" + Math.floor((Math.random() * max_key) + 1);
}
client.fetchValue({ bucket: "storage_test", key: key }, function(err) {
count++;
if (err) {
process.nextTick(function() { r.read_one_random_record(next, key); });
return;
}
next();
});
};
var client = new Riak.Client(nodes, function(err, c) {
if (err) {
console.error('Failed connect. Error: ' + err);
} else {
console.log('Connected.');
// client.ping((err)=>{
client.ping(function(err, rslt) {
if (err) {
console.error('Failed ping: ' + err);
} else {
// On success, ping returns true
console.log('Succeeded ping: ' + rslt);
}
});
// const listBuckets = new Riak.ListBuckets.Builder()
// .withCallback((err, response, data) => {
// console.log('hi');
// if (err) {
// console.error('Failed getting list of buckets. Error: ' + err + ', data:' + data);
// } else {
// console.log(response.buckets);
// console.log(response.done);
// }
// })
// .build();
// client.execute(listBuckets);
// c.stop((err, rslt) => {
// if (err) {
// console.error('Failed close client');
// } else {
// console.log('Closes');
// }
// })
}
});
function (async_cb) {
client.storeValue({ bucket: ordersBucket, key: order.id, value: order }, async_cb);
});
function (async_cb) {
client.storeValue({ bucket: customersBucket, key: customer.id, value: customer }, async_cb);
},
r.teardown = function() {
client.shutdown(function() {});
};
/**
* Stop the client
*/
stop() {
this._isConnected = false;
this.client.stop();
}