perf: enhanced security with AI suggestions

eggjs · Nov 16, 2024 · 176ec2c · 176ec2c
1 parent 2f46c2c
commit 176ec2c
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -276,7 +276,7 @@ config.schedule = {
       },
     },
   },
-  //the prefix for lockedKey for redis lock
+  // The final Redis key pattern will be: `${prefix}-${default}-${scheduleName}`
   default: 'default', // default schedule name,like project-name.
   prefix: 'schedule', // default schedule prefix
 };

diff --git a/lib/strategy/all.js b/lib/strategy/all.js
@@ -1,14 +1,15 @@
 'use strict';
 
 const Strategy = require('./timer');
+const path = require('path');
 
 module.exports = class AllStrategy extends Strategy {
   async handler() {
     let canBeLocked = true;
     const curConfig = this.agent?.config?.schedule;
     let lockedKey = '';
     if (curConfig?.cluster?.enable) {
-      lockedKey = this.key.replace(this.agent.baseDir, '');
+      lockedKey = path.relative(this.agent.baseDir, this.key);
       if (!(await this.agent.lockManager.tryAcquire(lockedKey))) {
         canBeLocked = false;
       }

diff --git a/lib/strategy/worker.js b/lib/strategy/worker.js
@@ -1,14 +1,15 @@
 'use strict';
 
 const Strategy = require('./timer');
+const path = require('path');
 
 module.exports = class AllStrategy extends Strategy {
   async handler() {
     let canBeLocked = true;
     const curConfig = this.agent?.config?.schedule;
     let lockedKey = '';
     if (curConfig?.cluster?.enable) {
-      lockedKey = this.key.replace(this.agent.baseDir, '');
+      lockedKey = path.relative(this.agent.baseDir, this.key);
       if (!(await this.agent.lockManager.tryAcquire(lockedKey))) {
         canBeLocked = false;
       }