first commit

2023-05-19 00:42:48 +08:00
commit 53de9c6c51
243 changed files with 39485 additions and 0 deletions
--- a/batch/leader/provider/redis-distlock.js
+++ b/batch/leader/provider/redis-distlock.js
@@ -0,0 +1,111 @@
+'use strict';
+
+var REDIS_DISTLOCK = {
+    DB: 5,
+    PREFIX: 'batch:locks:'
+};
+
+var Redlock = require('redlock');
+var debug = require('../../util/debug')('leader:redis-distlock');
+
+function RedisDistlockLocker(redisPool) {
+    this.pool = redisPool;
+    this.redlock = new Redlock([{}], {
+        // see http://redis.io/topics/distlock
+        driftFactor: 0.01, // time in ms
+        // the max number of times Redlock will attempt to lock a resource before failing
+        retryCount: 3,
+        // the time in ms between attempts
+        retryDelay: 100
+    });
+    this._locks = {};
+}
+
+module.exports = RedisDistlockLocker;
+module.exports.type = 'redis-distlock';
+
+function resourceId(resource) {
+    return REDIS_DISTLOCK.PREFIX + resource;
+}
+
+RedisDistlockLocker.prototype.lock = function(resource, ttl, callback) {
+    var self = this;
+    debug('RedisDistlockLocker.lock(%s, %d)', resource, ttl);
+    var lockId = resourceId(resource);
+
+    var lock = this._getLock(lockId);
+    function acquireCallback(err, _lock) {
+        if (err) {
+            return callback(err);
+        }
+        self._setLock(lockId, _lock);
+        return callback(null, _lock);
+    }
+    if (lock) {
+        return this._tryExtend(lock, ttl, function(err, _lock) {
+            if (err) {
+                return self._tryAcquire(lockId, ttl, acquireCallback);
+            }
+
+            return callback(null, _lock);
+        });
+    } else {
+        return this._tryAcquire(lockId, ttl, acquireCallback);
+    }
+};
+
+RedisDistlockLocker.prototype.unlock = function(resource, callback) {
+    var self = this;
+    var lock = this._getLock(resourceId(resource));
+    if (lock) {
+        this.pool.acquire(REDIS_DISTLOCK.DB, function (err, client) {
+            if (err) {
+                return callback(err);
+            }
+            self.redlock.servers = [client];
+            return self.redlock.unlock(lock, function(err) {
+                self.pool.release(REDIS_DISTLOCK.DB, client);
+                return callback(err);
+            });
+        });
+    }
+};
+
+RedisDistlockLocker.prototype._getLock = function(resource) {
+    if (this._locks.hasOwnProperty(resource)) {
+        return this._locks[resource];
+    }
+    return null;
+};
+
+RedisDistlockLocker.prototype._setLock = function(resource, lock) {
+    this._locks[resource] = lock;
+};
+
+RedisDistlockLocker.prototype._tryExtend = function(lock, ttl, callback) {
+    var self = this;
+    this.pool.acquire(REDIS_DISTLOCK.DB, function (err, client) {
+        if (err) {
+            return callback(err);
+        }
+        self.redlock.servers = [client];
+        return lock.extend(ttl, function(err, _lock) {
+            self.pool.release(REDIS_DISTLOCK.DB, client);
+            return callback(err, _lock);
+        });
+    });
+};
+
+RedisDistlockLocker.prototype._tryAcquire = function(resource, ttl, callback) {
+    var self = this;
+    this.pool.acquire(REDIS_DISTLOCK.DB, function (err, client) {
+        if (err) {
+            return callback(err);
+        }
+        self.redlock.servers = [client];
+        return self.redlock.lock(resource, ttl, function(err, _lock) {
+            self.pool.release(REDIS_DISTLOCK.DB, client);
+            return callback(err, _lock);
+        });
+    });
+};