基于redis实现的锁(用于控制nodejs的并发)

in 编程
关注公众号【好便宜】( ID:haopianyi222 ),领红包啦~
阿里云,国内最大的云服务商,注册就送数千元优惠券:https://t.cn/AiQe5A0g
腾讯云,良心云,价格优惠: https://t.cn/AieHwwKl
搬瓦工,CN2 GIA 优质线路,搭梯子、海外建站推荐: https://t.cn/AieHwfX9

场景:使用node+mongo搭建web应用时,常常遇到并发问题导致创建一条数据的时候会同时创建多条。

通过redis set 方法 配合 ‘NX’ 选项即可实现锁机制。

主要的方法有三个:

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 查看源码

关注公众号【好便宜】( ID:haopianyi222 ),领红包啦~
阿里云,国内最大的云服务商,注册就送数千元优惠券:https://t.cn/AiQe5A0g
腾讯云,良心云,价格优惠: https://t.cn/AieHwwKl
搬瓦工,CN2 GIA 优质线路,搭梯子、海外建站推荐: https://t.cn/AieHwfX9
扫一扫关注公众号添加购物返利助手,领红包
Comments are closed.

推荐使用阿里云服务器

超多优惠券

服务器最低一折,一年不到100!

朕已阅去看看