您的位置:首页 > 数据库 > Redis

基于redis实现的锁

2016-05-26 11:45 519 查看
主要的方法有三个:

lock(加锁)


unlock(解锁)


extend(更改锁过期时间)


主要源码: 

加锁:

Redlock.prototype.lock = function(resource,ttl,callback){
var self = this;
var _val = _random();
return new Promise(function(resolve, reject) {
var lock = new Lock(self, resource, _val);
self.client.SET(resource,_val,'EX',ttl,'NX',function(err,data){
if(!err && data === 'OK'){
return resolve(lock);
}else{
var err = err || 'resource: '+resource+' is locked.';
self.emit('lockError', err);
return reject(err);
}
})
}).asCallback(callback);
}

解锁: 

Redlock.prototype.unlock = function(lock,callback){
var self = this;
return new Promise(function(resolve, reject) {
if(typeof lock !== 'object') return reject('unlockError:lock is not object.');
self.client.GET(lock.resource,function(err,data){
if(!err && data === lock.value){
self.client.DEL(lock.resource,function(err){
if(err){
self.emit('unlockError',err);
return reject('unlockError:'+err);
}else{
return resolve(lock);
}
})
}else{
self.emit('unlockError',err || 'resource:' + lock.resource + ' unable to unlock.')
return reject('unlockError:' + err || ('resource:' + lock.resource + ' unable to unlock.'));
}
})
}).asCallback(callback);
}

扩展锁: 

Redlock.prototype.extend = function(lock,ttl,callback){
var self = this;
return new Promise(function(resolve, reject){
if(typeof lock !== 'object') return reject('unlockError:lock is not object.');
self.client.GET(lock.resource,function(err,data){
if(!err && data === lock.value){
self.client.expire(lock.resource,ttl,function(err,data){
if(!err && data === 1){
return resolve(lock);
}else{
self.emit('extendError', err);
return reject('extendError:'+err);
}
});
}else{
self.emit('extendError', err);
return reject('extendError:'+err);
}
})
}).asCallback(callback);
}

用法 

var client = require('redis').createClient('port','host');
var redlock = new RedLock(client);
var lock;

`callback`:

//lock
redlock.lock('test-resource-lock',3,function(err,lockInstance){
lock = lockInstance;
done(err);
});
//unlock
redlock.unlock(lock,function(err,data){
done(err);
});

`promise`:

redlock.lock('test-resource-lock-promise',3).done(
function(lock){
//todo
redlock.unlock(lock);
},
function(){
}
)


PS:支持callback and promise.

实现比较简单,100行内,有兴趣的可以在 https://github.com/navyxie/redlock-node href="https://github.com/navyxie/redlock-node" target=_blank> 查看源码
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  Redis Mongodb Nodejs 并发