Highways4Good/api/node_modules/kafka-node/lib/consumerGroupHeartbeat.js

'use strict';

const HeartbeatTimeoutError = require('./errors/HeartbeatTimeoutError');
const logger = require('./logging')('kafka-node:ConsumerGroupHeartbeat');

module.exports = class Heartbeat {
  constructor (client, handler) {
    this.client = client;
    this.handler = handler;
    this.pending = true;
  }

  send (groupId, generationId, memberId) {
    this.client.sendHeartbeatRequest(groupId, generationId, memberId, (error) => {
      if (this.canceled) {
        logger.debug('heartbeat yielded after being canceled', error);
        return;
      }
      this.pending = false;
      this.handler(error);
    });
  }

  verifyResolved () {
    if (this.pending) {
      this.canceled = true;
      this.pending = false;
      this.handler(new HeartbeatTimeoutError('Heartbeat timed out'));
      return false;
    }
    return true;
  }
};