ah-resque-ui/src/actions/ah-resque-ui.ts

A resque administration website for actionhero

import * as os from "os";
import { api, task, Action } from "actionhero";

// A helper class

abstract class ResqueAction extends Action {
  constructor() {
    super();
    this.middleware = ["ah-resque-ui-proxy-middleware"];
    this.logLevel = "debug";
    this.toDocument = false;
  }
}

export class ResqueRedisInfo extends ResqueAction {
  constructor() {
    super();
    this.name = "resque:redisInfo";
    this.description = "I return the results of redis info";
    this.inputs = {};
  }

  async run() {
    const redisInfo = await api.resque.queue.connection.redis.info();
    if (redisInfo) {
      return { redisInfo: redisInfo.split(os.EOL) };
    }
  }
}

export class ResqueResqueDetails extends ResqueAction {
  constructor() {
    super();
    this.name = "resque:resqueDetails";
    this.description = "I return the results of api.tasks.details";
    this.inputs = {};
  }

  async run() {
    return { resqueDetails: await task.details() };
  }
}

export class ResqueLoadWorkerQueues extends ResqueAction {
  constructor() {
    super();
    this.name = "resque:loadWorkerQueues";
    this.description = "I return the results of api.tasks.workers";
    this.inputs = {};
  }

  async run() {
    return { workerQueues: await task.workers() };
  }
}

export class ResqueForceCleanWorker extends ResqueAction {
  constructor() {
    super();
    this.name = "resque:forceCleanWorker";
    this.description = "I remove a worker from resque";
    this.inputs = {
      workerName: { required: true },
    };
  }

  async run({ params }) {
    return {
      generatedErrorPayload: await api.resque.queue.forceCleanWorker(
        params.workerName
      ),
    };
  }
}

export class ResqueFailedCount extends ResqueAction {
  constructor() {
    super();
    this.name = "resque:resqueFailedCount";
    this.description = "I return a count of failed jobs";
    this.inputs = {};
  }

  async run() {
    return { failedCount: await task.failedCount() };
  }
}

export class ResqueQueued extends ResqueAction {
  constructor() {
    super();
    this.name = "resque:queued";
    this.description = "I list enqueued jobs";
    this.inputs = {
      queue: {
        required: true,
      },
      start: {
        required: true,
        formatter: parseInt,
        default: 0,
      },
      stop: {
        required: true,
        formatter: parseInt,
        default: 99,
      },
    };
  }

  async run({ params }) {
    return {
      queueLength: await api.resque.queue.length(params.queue),
      jobs: await task.queued(params.queue, params.start, params.stop),
    };
  }
}

export class ResqueDelQueue extends ResqueAction {
  constructor() {
    super();
    this.name = "resque:delQueue";
    this.description = "I delete a queue";
    this.inputs = {
      queue: { required: true },
    };
  }

  async run({ params }) {
    return { deleted: await task.delQueue(params.queue) };
  }
}

export class ResqueResqueFailed extends ResqueAction {
  constructor() {
    super();
    this.name = "resque:resqueFailed";
    this.description = "I return failed jobs";
    this.inputs = {
      start: {
        required: true,
        formatter: parseInt,
        default: 0,
      },
      stop: {
        required: true,
        formatter: parseInt,
        default: 99,
      },
    };
  }

  async run({ params }) {
    return { failed: await task.failed(params.start, params.stop) };
  }
}

export class ResqueRemoveFailed extends ResqueAction {
  constructor() {
    super();
    this.name = "resque:removeFailed";
    this.description = "I remove a failed job";
    this.inputs = {
      id: {
        required: true,
        formatter: parseInt,
      },
    };
  }

  async run({ params }) {
    const failed = await task.failed(params.id, params.id);
    if (!failed) {
      throw Error("failed job not found");
    }
    return { removed: await task.removeFailed(failed[0]) };
  }
}

export class ResqueRemoveAllFailed extends ResqueAction {
  constructor() {
    super();
    this.name = "resque:removeAllFailed";
    this.description = "I remove all failed jobs";
    this.inputs = {};
  }

  async run() {
    let removed = 0;
    const failed = await task.failed(0, 0);
    if (failed && failed.length > 0) {
      const failedJob = failed[0];
      removed += await task.removeFailed(failedJob);
      await this.run();
    }
    return { removed };
  }
}

export class ResqueRetryAndRemoveFailed extends ResqueAction {
  constructor() {
    super();
    this.name = "resque:retryAndRemoveFailed";
    this.description = "I retry a failed job";
    this.inputs = {
      id: {
        required: true,
        formatter: parseInt,
      },
    };
  }

  async run({ params }) {
    const failed = await task.failed(params.id, params.id);
    if (!failed) {
      throw new Error("failed job not found");
    }
    return { retried: task.retryAndRemoveFailed(failed[0]) };
  }
}

export class ResqueRetryAndRemoveAllFailed extends ResqueAction {
  constructor() {
    super();
    this.name = "resque:retryAndRemoveAllFailed";
    this.description = "I retry all failed jobs";
    this.inputs = {};
  }

  async run() {
    let retried = 0;
    const failed = await task.failed(0, 0);
    if (failed && failed.length > 0) {
      const failedJob = failed[0];
      retried += (await task.retryAndRemoveFailed(failedJob)) ? 1 : 0;
      await this.run();
    }
    return { retried };
  }
}

export class ResqueLocks extends ResqueAction {
  constructor() {
    super();
    this.name = "resque:locks";
    this.description = "I return all locks";
    this.inputs = {};
  }

  async run() {
    return { locks: await task.locks() };
  }
}

export class ResqueDelLock extends ResqueAction {
  constructor() {
    super();
    this.name = "resque:delLock";
    this.description = "I delete a lock";
    this.inputs = {
      lock: { required: true },
    };
  }

  async run({ params }) {
    return { count: await task.delLock(params.lock) };
  }
}

export class ResqueDelayedJobs extends ResqueAction {
  constructor() {
    super();
    this.name = "resque:delayedjobs";
    this.description = "I return paginated lists of delayedjobs";
    this.inputs = {
      start: {
        required: true,
        formatter: parseInt,
        default: 0,
      },
      stop: {
        required: true,
        formatter: parseInt,
        default: 99,
      },
    };
  }

  async run({ params }) {
    const timestamps = [];
    const delayedjobs = {};
    let timestampsCount = 0;

    const allTimestamps = await task.timestamps();
    if (allTimestamps.length > 0) {
      timestampsCount = allTimestamps.length;

      for (let i = 0; i < allTimestamps.length; i++) {
        if (i >= params.start && i <= params.stop) {
          timestamps.push(allTimestamps[i]);
        }
      }

      for (const j in timestamps) {
        const timestamp = timestamps[j];
        delayedjobs[timestamp] = await task.delayedAt(timestamp);
      }
    }

    return { delayedjobs, timestampsCount };
  }
}

export class ResqueDelDelayed extends ResqueAction {
  constructor() {
    super();
    this.name = "resque:delDelayed";
    this.description = "I delete a delayed job";
    this.inputs = {
      timestamp: {
        required: true,
        formatter: parseInt,
      },
      count: {
        required: true,
        formatter: parseInt,
      },
    };
  }

  async run({ params }) {
    const delayed = await task.delayedAt(params.timestamp);
    if (delayed.tasks.length === 0 || !delayed.tasks[params.count]) {
      throw new Error("delayed job not found");
    }

    const job = delayed.tasks[params.count];
    return {
      timestamps: await task.delDelayed(job.queue, job.class, job.args),
    };
  }
}

export class ResqueRunDelayed extends ResqueAction {
  constructor() {
    super();
    this.name = "resque:runDelayed";
    this.description = "I run a delayed job now";
    this.inputs = {
      timestamp: {
        required: true,
        formatter: parseInt,
      },
      count: {
        required: true,
        formatter: parseInt,
      },
    };
  }

  async run({ params }) {
    const delayed = await task.delayedAt(params.timestamp);
    if (delayed.tasks.length === 0 || !delayed.tasks[params.count]) {
      throw new Error("delayed job not found");
    }

    const job = delayed.tasks[params.count];
    await task.delDelayed(job.queue, job.class, job.args);
    await task.enqueue(job.class, job.args, job.queue);
  }
}