[Node.js] 基于Redis实现分布式锁

分布式锁出应用场景有很多,比如库存扣减什么的,不再多说。

我第一次接触这个是做数据同步服务。同步服务可能有多个实例,使用node.js开发,多个同步服务竟争为多个客户数据进行同步。

 

在实现分布式锁的时候,有两个特别重要的点要注意:

1. 具有失效机制,防止死锁(加锁的实例不排除中途释放)

2. 高可用高性能的锁获取与释放

 

下面直接上代码:

...

enum LockState {
  wait,  // 等待中,空闲中
  execute // 正在执行
}

// 分布式部分的主要代码
async execute(data: xxx) {
  // 获取当前实例的标识作为 owner,主要用来区分多个实例时,加锁的服务是否为当前服务
  let owner = this.getOwner(); 
  try {
    // 获取锁
    let lock = await this.getLock(data.id);
    // 如果锁正在执行,并且所有者不是本实例,则退出
    // 这种情况说明是其它实例正在处理这个data
    // 在我的项目中,实际情况是,另一个同步服务正在给这个客户同步数据,所以我们这个服务实例,就不需要再给它同步了,直接跳过,处理后继的客户
    if (item.state == LockState.execute && item.owner != owner)
      return;
      
    lock.owner = owner;
    lock.state = LockState.execute;
    // 加锁
    awiat this.setLock(data.id, lock);
    
    // 等待500毫秒
    await Utils.sleep(500);
    // 重新取一下同步状态, 防止并发冲突
    let lock = await this.getLock(data.id);
    // 如果所有者不为当前服务或状态不是执行中,则退出
    if (lock.owner != data.owner || lock.state != LockState.execute) {
      // 注意了,这里为什么要等个500ms,再来获取锁,判断锁的状态呢?
      // 那是因为,由于网络延时,我们上锁的时候,可能会有另一个实例也来给这个data上锁,我们不一定就成功了。
      // 但我们也不能马上去获取锁看是否成功,还是因为网络延时的原因。
      // 所以这里等待一小儿,则可以避免此情况的出现。
      return;
    }
      
    // 开一个定时器, 每隔5秒更新锁, 防止丢失
    // 为什么要加这个呢? 很简单,我们锁内的代码执行一次,不一定60秒就完成了,
    // 比如我们第一次给某个客户同步数据,可能要好几分钟
    // 那为什么又不能一直锁着,不让它超时呢?
    // 那是因为可能会造成死锁,比如我们的实例意外中止,半天起不起来的话,那这个data就会一直锁着
    let timer: Timer = setInterval(async () => {
        if (lock.state == LockState.execute) {
            await this.setLock(item.id, item.state);
        }
    }, 5000);  
    
    try {
      // 此处才真正开始执行对这个数据的处理
      await this.exec(data); 
    } finally {
      // 清除定时器
      if (timer) clearInterval(timer);
      // 同步完成,释放锁
      lock.state = LockState.wait;
      lock.lastSync = Utils.getTimeInMillis();  // 记录一个同步时间,比如我们可以按此来排序,优先处理等得最久的客户数据
      await this.setLock(data.id, lock);
    }
  } catch (e) {
    Logger.error(e);
  }
}

// 获取锁
async getLock(key: string): Promise<Lock> {
  let v = await cache.get(key);
  return (new Lock()).load(v);
}

// 设置锁
async setLock(key: string, state: Lock) {
  // 往redis中写入一个数据,有效期 60 秒
  await cache.setEx(key, 60, state.toString());
}

// 返回当前服务实例标识,用于和其它服务作区分
getOwner(): string {
  // 比如这里用本机的ip地址加上缓存标识作为owner标识
  return `lock:${AppUtils.getIPAddress()}:${CacheKeys.flag}`;
}


export class Utils {

    /**
     * 等待指定的时间
     * @param ms
     */
    static async sleep(ms: number) {
        return new Promise((resolve) => {
            setTimeout(() => {
                resolve('');
            }, ms)
        });
    }
    
    /**
     * 获取当前时间戳 (与java相同)
     */
    static getTimeInMillis(): number {
        return new Date().getTime() - AppUtils._time1970;
    }

}
...

 

posted @ 2021-03-09 14:54  我爱我家喵喵  阅读(740)  评论(0编辑  收藏  举报