polling_gist5.js

class Poll {
  /**
   * @typedef options
   * @property {number} interval Interval between each poll in `seconds`
   * @property {number} maxPollCount Max number of times to poll
   *
   * @param {options} options
   * @param {(Object) => Promise<boolean>} pollingFunction
   * @param {QueueClientInterface} queueClient
   */
  constructor(
    { interval, maxPollCount },
    pollingFunction,
    queueClient,
    logger = console
  ) {
    this.logger = logger;
    this.interval = interval;
    this.queueClient = queueClient;
    this.maxPollCount = maxPollCount;
    this.pollingFunction = pollingFunction;
  }

  /**
   * @param {(dequeuedData: object) => Promise<boolean>} pollingFunction if this function returns `true`, polling is continued
   */
  async fetchAndValidate(dequeuedData) {
    // the polling function calls the external service to check the status
    const pollContinue = await this.pollingFunction(dequeuedData);
    // We are saving the polls left count with the poll data when
    // it is being pushed to the queue so we can check if polling
    // can be done or not, this count gets altered everytime poll
    // happens
    const pollsLeft = dequeuedData.pollCount - 1;
    if (pollContinue && pollsLeft > 0) {
      // Here we push the data back to the queue with the
      // updated pollCount
      this.queueClient.push(this.interval, {
        ...dequeuedData,
        pollCount: pollsLeft,
      });
    } else {
      // if poll limit is reached or the polling function returns
      // false, we stop the polling, this can be logged if you want
      // or not
      this.logger.warn({
        MQ: {
          msg: "polling limit reached for the following batch",
          details: dequeuedData,
        },
      });
    }
  }
}