1. /**
  2. * Extends existing dataset logic with rate limiting.
  3. *
  4. * @param {function} originalFunction The original function to be rate limited.
  5. * @param {number} maxRequestsPerSecond The maximum number of requests allowed per second.
  6. * @returns {function} A rate-limited version of the original function.
  7. */
  8. function rateLimit(originalFunction, maxRequestsPerSecond) {
  9. let queue = []; // Queue to hold requests
  10. let running = 0; // Number of currently executing requests
  11. let lastExecutionTime = 0; // Time of the last execution
  13. /**
  14. * Rate-limited version of the original function.
  15. * @param {...any} args Arguments to pass to the original function.
  16. * @returns {Promise<any>} A promise that resolves with the result of the original function.
  17. */
  18. return async function (...args) {
  19. return new Promise((resolve, reject) => {
  20. const now = Date.now();
  22. // Check if we're exceeding the rate limit
  23. if (running >= maxRequestsPerSecond) {
  24. // Add the request to the queue
  25. queue.push({ args, resolve, reject });
  26. return;
  27. }
  29. // Start the request
  30. running++;
  31. const startTime = now;
  33. // Execute the original function
  34. originalFunction(...args)
  35. .then(result => {
  36. // Resolve the promise with the result
  37. resolve(result);
  38. })
  39. .catch(error => {
  40. // Reject the promise with the error
  41. reject(error);
  42. })
  43. .finally(() => {
  44. // Decrement the number of running requests
  45. running--;
  47. // Process the queue
  48. processQueue();
  49. });
  50. });
  51. };
  53. /**
  54. * Processes the request queue.
  55. */
  56. function processQueue() {
  57. while (queue.length > 0) {
  58. const { args, resolve, reject } = queue.shift();
  59. const now = Date.now();
  61. // Check if the rate limit has been exceeded
  62. if (now - lastExecutionTime >= 1000) {
  63. // Reset the rate limit
  64. running = 0;
  65. lastExecutionTime = now;
  66. }
  67. }
  68. }
  69. }

Add your comment