Fix client RequestQueue
- Reference documented queue parameters - Add freeze/thaw methods
This commit is contained in:
parent
3b101a7b2f
commit
acfaa69a07
|
@ -97,20 +97,26 @@ function RequestQueue(opts) {
|
|||
this.timeout = (opts.timeout > 0) ? opts.timeout : 0;
|
||||
this._queue = [];
|
||||
this._timer = null;
|
||||
this._frozen = false;
|
||||
}
|
||||
|
||||
/**
|
||||
* Insert request into queue.
|
||||
*
|
||||
*/
|
||||
RequestQueue.prototype.queue = function queue(message, expect, emitter, cb) {
|
||||
if (this._queue.length >= this.maxLength) {
|
||||
RequestQueue.prototype.enqueue = function enqueue(msg, expect, emitter, cb) {
|
||||
if (this._queue.length >= this.size || this._frozen) {
|
||||
return false;
|
||||
}
|
||||
this._queue.push([message, expect, emitter, cb]);
|
||||
if (this.maxAge > 0) {
|
||||
var self = this;
|
||||
this._queue.push([msg, expect, emitter, cb]);
|
||||
if (this.timeout > 0) {
|
||||
if (this._timer !== null) {
|
||||
this._timer = setTimeout(this.purge.bind(this), this.maxAge);
|
||||
this._timer = setTimeout(function () {
|
||||
// If queue times out, don't allow new entries until thawed
|
||||
self.freeze();
|
||||
self.purge();
|
||||
}, this.timeout);
|
||||
}
|
||||
}
|
||||
return true;
|
||||
|
@ -140,6 +146,21 @@ RequestQueue.prototype.purge = function purge() {
|
|||
});
|
||||
};
|
||||
|
||||
/**
|
||||
* Freeze queue, refusing any new entries.
|
||||
*/
|
||||
RequestQueue.prototype.freeze = function freeze() {
|
||||
this._frozen = true;
|
||||
};
|
||||
|
||||
/**
|
||||
* Thaw queue, allowing new entries again.
|
||||
*/
|
||||
RequestQueue.prototype.thaw = function thaw() {
|
||||
this._frozen = false;
|
||||
};
|
||||
|
||||
|
||||
///--- API
|
||||
|
||||
/**
|
||||
|
@ -183,11 +204,12 @@ function Client(options) {
|
|||
}
|
||||
|
||||
this.queuing = (options.queuing !== undefined) ? options.queuing : true;
|
||||
if (this.queuing) {
|
||||
this._queue = new RequestQueue({
|
||||
this.queue = new RequestQueue({
|
||||
size: parseInt((options.queueSize || 0), 10),
|
||||
timeout: parseInt((options.queueTimeout || 0), 10)
|
||||
});
|
||||
if (!this.queuing) {
|
||||
this.queue.freeze();
|
||||
}
|
||||
|
||||
this.socket = null;
|
||||
|
@ -677,14 +699,13 @@ Client.prototype.unbind = function unbind(callback) {
|
|||
*/
|
||||
Client.prototype.destroy = function destroy() {
|
||||
this.destroyed = true;
|
||||
if (this._queue) {
|
||||
this.queue.freeze();
|
||||
// Purge any queued requests which are now meaningless
|
||||
this._queue.flush(function (msg, expect, emitter, cb) {
|
||||
this.queue.flush(function (msg, expect, emitter, cb) {
|
||||
if (typeof (cb) === 'function') {
|
||||
cb(new Error('client destroyed'));
|
||||
}
|
||||
});
|
||||
}
|
||||
this.unbind();
|
||||
this.emit('destroy');
|
||||
};
|
||||
|
@ -930,10 +951,8 @@ Client.prototype._connect = function _connect() {
|
|||
* Flush queued requests out to the socket.
|
||||
*/
|
||||
Client.prototype._flushQueue = function _flushQueue() {
|
||||
if (this._queue) {
|
||||
// Pull items we're about to process out of the queue.
|
||||
this._queue.flush(this._send.bind(this));
|
||||
}
|
||||
this.queue.flush(this._send.bind(this));
|
||||
};
|
||||
|
||||
/**
|
||||
|
@ -1048,14 +1067,14 @@ Client.prototype._send = function _send(message,
|
|||
return this._sendSocket(message, expect, emitter, callback);
|
||||
}
|
||||
if (!this.socket || !this.connected) {
|
||||
if (this._queue) {
|
||||
if (!this._queue.queue(message, expect, emitter, callback)) {
|
||||
callback(new Error('request queue full'));
|
||||
if (!this.queue.enqueue(message, expect, emitter, callback)) {
|
||||
callback(new Error('unable to enqueue request'));
|
||||
}
|
||||
// Initiate reconnect if needed
|
||||
return this._connect();
|
||||
if (this.reconnect) {
|
||||
this._connect();
|
||||
}
|
||||
return callback(new ConnectionError('no socket'));
|
||||
return false;
|
||||
} else {
|
||||
this._flushQueue();
|
||||
return this._sendSocket(message, expect, emitter, callback);
|
||||
|
|
Loading…
Reference in New Issue