'use strict'; var _ = require('underscore'); var RedisPool = require('redis-mpool'); var RedisDistlockLocker = require('./provider/redis-distlock'); var debug = require('../util/debug')('leader-locker'); var LOCK = { TTL: 5000 }; function Locker(locker, ttl) { this.locker = locker; this.ttl = (Number.isFinite(ttl) && ttl > 0) ? ttl : LOCK.TTL; this.renewInterval = this.ttl / 5; this.intervalIds = {}; } 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) { 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 redisPool = new RedisPool(_.extend({ name: 'batch-distlock' }, config.redisConfig)); var locker = new RedisDistlockLocker(redisPool); return new Locker(locker, config.ttl); };