virool-pivot/src/server/utils/requester/requester.ts

A web-based exploratory visualization UI for Druid.io

import { $, helper } from 'plywood';
import { druidRequesterFactory, DruidRequestDecorator } from 'plywood-druid-requester';
import { mySqlRequesterFactory } from 'plywood-mysql-requester';

export interface ProperDruidRequesterOptions {
  druidHost?: string;
  retry?: number;
  timeout?: number;
  verbose?: boolean;
  concurrentLimit?: number;
  requestDecorator?: DruidRequestDecorator;
}

export function properDruidRequesterFactory(options: ProperDruidRequesterOptions): Requester.PlywoodRequester<any> {
  var {
    druidHost,
    retry,
    timeout,
    verbose,
    concurrentLimit,
    requestDecorator
  } = options;

  var druidRequester = druidRequesterFactory({
    host: druidHost,
    timeout: timeout || 30000,
    requestDecorator
  });

  if (retry) {
    druidRequester = helper.retryRequesterFactory({
      requester: druidRequester,
      retry: retry,
      delay: 500,
      retryOnTimeout: false
    });
  }

  if (verbose) {
    druidRequester = helper.verboseRequesterFactory({
      requester: druidRequester
    });
  }

  if (concurrentLimit) {
    druidRequester = helper.concurrentLimitRequesterFactory({
      requester: druidRequester,
      concurrentLimit: concurrentLimit
    });
  }

  return druidRequester;
}