1. const schedule = require('node-schedule');
  2. const fs = require('fs');
  3. const path = require('path');
  5. class FileScheduler {
  6. constructor(rateLimit = 5, intervalSeconds = 60) {
  7. this.rateLimit = rateLimit; // Max files to run per interval
  8. this.intervalSeconds = intervalSeconds; // Interval between scheduling checks
  9. this.scheduledFiles = [];
  10. }
  12. /**
  13. * Schedules a file for execution with rate limiting.
  14. * @param {string} filePath - The path to the file to execute.
  15. */
  16. scheduleFile(filePath) {
  17. // Check if the file is already scheduled
  18. if (this.scheduledFiles.includes(filePath)) {
  19. console.log(`File ${filePath} already scheduled.`);
  20. return;
  21. }
  23. // Check rate limit
  24. if (this.scheduledFiles.length >= this.rateLimit) {
  25. console.log(`Rate limit reached. Cannot schedule ${filePath}.`);
  26. return;
  27. }
  29. // Schedule the file
  30. const job = schedule.scheduleJob(this.intervalSeconds, async () => {
  31. try {
  32. console.log(`Executing ${filePath} at ${new Date().toLocaleTimeString()}`);
  33. await this.executeFile(filePath);
  34. } catch (error) {
  35. console.error(`Error executing ${filePath}:`, error);
  36. }
  37. });
  39. this.scheduledFiles.push(filePath);
  40. console.log(`Scheduled ${filePath} for execution.`);
  41. }
  43. /**
  44. * Executes a file.
  45. * @param {string} filePath - The path to the file to execute.
  46. * @returns {Promise} - A promise that resolves when the file execution is complete.
  47. */
  48. async executeFile(filePath) {
  49. try {
  50. // Read file content
  51. const fileContent = fs.readFileSync(filePath, 'utf8');
  53. // Execute the file content (e.g., using eval, but be cautious!)
  54. // Consider using a safer execution method like vm (Node.js built-in) if possible.
  55. eval(fileContent);
  58. // Simulate some work
  59. await new Promise(resolve => setTimeout(resolve, 1000));
  60. } catch (error) {
  61. throw error; // Re-throw the error for the scheduler to handle
  62. }
  63. }
  65. /**
  66. * Clears all scheduled files.
  67. */
  68. clearSchedule() {
  69. this.scheduledFiles = [];
  70. this.scheduledJobs = schedule.getRules().filter(rule => rule.job.name === 'fileExecutor').map(rule => rule.job.id);
  71. schedule.cancelJobs(this.scheduledJobs);
  72. console.log('Schedule cleared.');
  73. }
  74. }
  76. module.exports = FileScheduler;

Add your comment