75 lines
2.2 KiB
JavaScript
75 lines
2.2 KiB
JavaScript
'use strict';
|
|
|
|
var RedisDistlockLocker = require('./provider/redis-distlock');
|
|
var debug = require('../util/debug')('leader-locker');
|
|
var EventEmitter = require('events').EventEmitter;
|
|
var util = require('util');
|
|
|
|
var LOCK = {
|
|
TTL: 5000
|
|
};
|
|
|
|
function Locker (locker, ttl) {
|
|
EventEmitter.call(this);
|
|
this.locker = locker;
|
|
this.ttl = (Number.isFinite(ttl) && ttl > 0) ? ttl : LOCK.TTL;
|
|
this.renewInterval = this.ttl / 5;
|
|
this.intervalIds = {};
|
|
}
|
|
util.inherits(Locker, EventEmitter);
|
|
|
|
module.exports = Locker;
|
|
|
|
Locker.prototype.lock = function (resource, callback) {
|
|
var self = this;
|
|
debug('Locker.lock(%s, %d)', resource, this.ttl);
|
|
this.locker.lock(resource, this.ttl, function (err, lock) {
|
|
if (!err) {
|
|
self.startRenewal(resource);
|
|
}
|
|
return callback(err, lock);
|
|
});
|
|
};
|
|
|
|
Locker.prototype.unlock = function (resource, callback) {
|
|
var self = this;
|
|
debug('Locker.unlock(%s)', resource);
|
|
this.locker.unlock(resource, function (err) {
|
|
self.stopRenewal(resource);
|
|
return callback(err);
|
|
});
|
|
};
|
|
|
|
Locker.prototype.startRenewal = function (resource) {
|
|
var self = this;
|
|
if (!this.intervalIds.hasOwnProperty(resource)) {
|
|
this.intervalIds[resource] = setInterval(function () {
|
|
debug('Trying to extend lock resource=%s', resource);
|
|
self.locker.lock(resource, self.ttl, function (err, _lock) {
|
|
if (err) {
|
|
self.emit('error', err, resource);
|
|
return self.stopRenewal(resource);
|
|
}
|
|
if (_lock) {
|
|
debug('Extended lock resource=%s', resource);
|
|
}
|
|
});
|
|
}, this.renewInterval);
|
|
}
|
|
};
|
|
|
|
Locker.prototype.stopRenewal = function (resource) {
|
|
if (this.intervalIds.hasOwnProperty(resource)) {
|
|
clearInterval(this.intervalIds[resource]);
|
|
delete this.intervalIds[resource];
|
|
}
|
|
};
|
|
|
|
module.exports.create = function createLocker (type, config) {
|
|
if (type !== 'redis-distlock') {
|
|
throw new Error('Invalid type Locker type. Valid types are: "redis-distlock"');
|
|
}
|
|
var locker = new RedisDistlockLocker(config.pool);
|
|
return new Locker(locker, config.ttl);
|
|
};
|