基于redis實(shí)現(xiàn)的鎖（用于控制nodejs的并發(fā)）

pkhope 發(fā)布于2019-08-19 18:22 / 2747人閱讀

摘要：場(chǎng)景使用搭建應(yīng)用時(shí)，常常遇到并發(fā)問(wèn)題導(dǎo)致創(chuàng)建一條數(shù)據(jù)的時(shí)候會(huì)同時(shí)創(chuàng)建多條。通過(guò)方法配合選項(xiàng)即可實(shí)現(xiàn)鎖機(jī)制。主要的方法有三個(gè)加鎖解鎖更改鎖過(guò)期時(shí)間用法支持實(shí)現(xiàn)比較簡(jiǎn)單，行內(nèi)，有興趣的可以在這里查看源碼

場(chǎng)景：使用node+mongo搭建web應(yīng)用時(shí)，常常遇到并發(fā)問(wèn)題導(dǎo)致創(chuàng)建一條數(shù)據(jù)的時(shí)候會(huì)同時(shí)創(chuàng)建多條。

通過(guò)redis set 方法配合 ‘NX’ 選項(xiàng)即可實(shí)現(xiàn)鎖機(jī)制。

主要的方法有三個(gè)：

lock(加鎖)

unlock(解鎖)

extend(更改鎖過(guò)期時(shí)間)

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.

實(shí)現(xiàn)比較簡(jiǎn)單，100行內(nèi)，有興趣的可以在這里https://github.com/navyxie/redlock-node 查看源碼