@clonegod/ttd-bsc-common 1.0.0

package/dist/utils/redis_lock.js

@@ -0,0 +1,100 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.RedisLock = void 0;
+const dist_1 = require("@clonegod/ttd-core/dist");
+class RedisLock {
+    constructor(lock_prefix) {
+        this.lock_prefix = lock_prefix;
+        this.redisClient = null;
+        this.lockMaxRetries = 10;
+        this.lockRetryDelayMs = 300;
+        this.lockExpireSeconds = 10;
+    }
+    getRedisClient() {
+        return __awaiter(this, void 0, void 0, function* () {
+            if (!this.redisClient) {
+                this.redisClient = yield (0, dist_1.getRedisCache)();
+            }
+            return this.redisClient;
+        });
+    }
+    getLockKey(lock_identifier) {
+        return `${this.lock_prefix}:lock:${lock_identifier}`;
+    }
+    acquireLock(lock_key_1, lock_value_1) {
+        return __awaiter(this, arguments, void 0, function* (lock_key, lock_value, expireSeconds = this.lockExpireSeconds) {
+            const redisClient = yield this.getRedisClient();
+            const result = yield redisClient.set(lock_key, lock_value, {
+                NX: true,
+                EX: expireSeconds
+            });
+            (0, dist_1.log_info)(`try acquireLock: lock_key=${lock_key}, lock_value=${lock_value}, expireSeconds=${expireSeconds}, result=${result}`);
+            const success = result === 'OK';
+            if (success) {
+                (0, dist_1.log_info)(`acquire lock success: lock_key=${lock_key}, lock_value=${lock_value}`);
+            }
+            return success;
+        });
+    }
+    releaseLock(lock_key, lock_value) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const redisClient = yield this.getRedisClient();
+            const script = `
+            if redis.call('get', KEYS[1]) == ARGV[1] then
+                return redis.call('del', KEYS[1])
+            else
+                return 0
+            end
+        `;
+            const result = yield redisClient.eval(script, {
+                keys: [lock_key],
+                arguments: [lock_value]
+            });
+            const success = Number(result) === 1;
+            if (success) {
+                (0, dist_1.log_info)(`release lock success: lock_key=${lock_key}, lock_value=${lock_value}`);
+            }
+            else {
+                (0, dist_1.log_info)(`release lock failed: lock_key=${lock_key}, lock_value=${lock_value}, maybe expired or locked by other process`);
+            }
+            return success;
+        });
+    }
+    withLock(lock_identifier, callback) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const lock_key = this.getLockKey(lock_identifier);
+            const lock_value = `${Date.now()}-${Math.random().toString(36).substring(2, 15)}`;
+            let retries = 0;
+            let acquired = false;
+            try {
+                while (retries < this.lockMaxRetries) {
+                    acquired = yield this.acquireLock(lock_key, lock_value);
+                    if (acquired)
+                        break;
+                    yield new Promise(resolve => setTimeout(resolve, this.lockRetryDelayMs));
+                    retries++;
+                }
+                if (!acquired) {
+                    throw new Error(`acquire lock failed: lock_key=${lock_key}, lock_value=${lock_value}, after ${this.lockMaxRetries} retries, maybe locked by other process`);
+                }
+                return yield callback();
+            }
+            finally {
+                if (acquired) {
+                    yield (0, dist_1.sleep)(parseInt(process.env.NONCE_LOCK_RELEASE_DELAY_MS || '1000'));
+                    yield this.releaseLock(lock_key, lock_value);
+                }
+            }
+        });
+    }
+}
+exports.RedisLock = RedisLock;

package/package.json

@@ -0,0 +1,30 @@
+{
+    "name": "@clonegod/ttd-bsc-common",
+    "version": "1.0.0",
+    "description": "BSC common library",
+    "license": "UNLICENSED",
+    "main": "dist/index.js",
+    "types": "dist/index.d.ts",
+    "files": [
+        "dist"
+    ],
+    "scripts": {
+        "clean": "rm -rf dist node_modules",
+        "build": "npx tsc --outDir ./dist",
+        "push": "npm run build && npm publish"
+    },
+    "dependencies": {
+        "@clonegod/ttd-core": "2.0.19",
+        "axios": "^1.8.4",
+        "dotenv": "^16.4.7",
+        "ethers": "^5.8.0"
+    },
+    "devDependencies": {
+        "@types/node": "^22.14.0",
+        "typescript": "^5.8.2"
+    },
+    "overrides": {},
+    "publishConfig": {
+        "access": "public"
+    }
+}