support batch getFileSize calls
This commit is contained in:
40
rpc.js
40
rpc.js
@@ -212,14 +212,45 @@ var getChannelList = function (store, Sessions, publicKey, cb) {
|
||||
|
||||
var getFileSize = function (store, channel, cb) {
|
||||
if (!isValidChannel(channel)) { return void cb('INVALID_CHAN'); }
|
||||
if (typeof(store.getChannelSize) !== 'function') {
|
||||
return cb('GET_CHANNEL_SIZE_UNSUPPORTED');
|
||||
}
|
||||
|
||||
// TODO don't blow up if their store doesn't have this API
|
||||
return void store.getChannelSize(channel, function (e, size) {
|
||||
if (e) { return void cb(e.code); }
|
||||
cb(void 0, size);
|
||||
});
|
||||
};
|
||||
|
||||
var getMultipleFileSize = function (store, channels, cb) {
|
||||
if (!isValidChannel(channel)) { }
|
||||
|
||||
if (!Array.isArray(channels)) { return cb('INVALID_LIST'); }
|
||||
if (typeof(store.getChannelSize) !== 'function') {
|
||||
return cb('GET_CHANNEL_SIZE_UNSUPPORTED');
|
||||
}
|
||||
|
||||
var i = channels.length;
|
||||
var counts = {};
|
||||
|
||||
var done = function () {
|
||||
i--;
|
||||
if (i === 0) { return cb(void 0, counts); }
|
||||
};
|
||||
|
||||
channels.forEach(function (channel) {
|
||||
store.getChannelSize(channel, function (e, size) {
|
||||
if (e) {
|
||||
counts[channel] = -1;
|
||||
return done();
|
||||
}
|
||||
|
||||
counts[channel] = size;
|
||||
done();
|
||||
});
|
||||
});
|
||||
};
|
||||
|
||||
var getTotalSize = function (pinStore, messageStore, Sessions, publicKey, cb) {
|
||||
var bytes = 0;
|
||||
|
||||
@@ -430,6 +461,13 @@ RPC.create = function (config, cb) {
|
||||
});
|
||||
case 'GET_FILE_SIZE':
|
||||
return void getFileSize(ctx.store, msg[1], Respond);
|
||||
case 'GET_MULTIPLE_FILE_SIZE':
|
||||
return void getMultipleFileSize(ctx.store, msg[1], function (e, dict) {
|
||||
if (e) { return void Respond(e); }
|
||||
Respond(void 0, dict);
|
||||
});
|
||||
return void Respond('NOT_IMPLEMENTED');
|
||||
break;
|
||||
default:
|
||||
return void Respond('UNSUPPORTED_RPC_CALL', msg);
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user