oira666_tg 1.0.13 → 1.0.14

package/index.js

@@ -4,6 +4,7 @@ const { redirect_handler, apply_handler_to_cron_task } = require('./middleware/a
 const { find_in_handlers } = require('./utils/find_in_handlers');
 const { send_empty_message } = require('./utils/send_empty_message');
 const { remove_html_tags } = require('./utils/remove_html_tags');
+const { exclusive_locker } = require('./utils/exclusive_locker');
 const { error_handler, notifyAdmins, sendErrorToDevelopers, error_403_handler, sendErrorToUser } = require('./middleware/error_handler');
 
 const { TelegramSession } = require('./source/classes/TelegramSession');
@@ -36,3 +37,4 @@ module.exports.WaitMessage = WaitMessage;
 
 // utils
 module.exports.remove_html_tags = remove_html_tags;
+module.exports.exclusive_locker = exclusive_locker;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "oira666_tg",
-  "version": "1.0.13",
+  "version": "1.0.14",
   "description": "tg framework for oira666",
   "main": "index.js",
   "scripts": {

package/utils/exclusive_locker.js

@@ -0,0 +1,41 @@
+const tracked_functions = {};
+
+const exclusive_locker = (max_counter = 30, callback) => {
+  const key = String(callback);
+  if (!tracked_functions[key]) {
+    tracked_functions[key] = {
+      is_locked: false,
+      counter: 0,
+    };
+  }
+
+  const locked = tracked_functions[key];
+
+  return async (...args) => {
+    if (locked.is_locked) {
+      locked.counter++;
+      if (locked.counter > max_counter) {
+        locked.is_locked = false;
+        locked.counter = 0;
+      }
+      return;
+    }
+
+    locked.is_locked = true;
+    locked.counter = 0;
+
+    try {
+      await callback(...args);
+    }
+    catch (err) {
+      locked.is_locked = false;
+      locked.counter = 0;
+      throw err;
+    }
+
+    locked.is_locked = false;
+    locked.counter = 0;
+  }
+};
+
+module.exports.exclusive_locker = exclusive_locker;