// Subclass of Bot.
// Polls content services at specified intervals.

var Bot = require('../bot');
var util = require('util');
var logger = require('../../logger');

// options.source - The source to pull from.
// options.contentService - The contentService to control.
var PullBot = function(options) {
  this.interval = options.contentService.interval || 120000;
  Bot.call(this, options);
};

util.inherits(PullBot, Bot);

// Start polling content service
PullBot.prototype.start = function() {
  if (this.enabled) return;
  PullBot.super_.prototype.start.apply(this);
  this._fetching = false;
  this.poll();
};

// Clear interval to stop polling
PullBot.prototype.stop = function() {
  clearTimeout(this.timeoutObject);
  PullBot.super_.prototype.stop.apply(this);
};

// Poll content service at specified intervals
PullBot.prototype.poll = function() {
  if (!this.enabled) return;
  var self = this;
  this.fetch(function(err, lastReportDate, lastReportDateSavedSearch) {
    if (err) {
      logger.warning(err);
    }
    // Set timeout to do the next poll.
    self.timeoutObject = setTimeout(function() { self.poll(); }, self._interval());

    // If fetch returned a lastReportDate, store it.
    if (lastReportDate && lastReportDate != -Infinity && self.source.lastReportDate !== lastReportDate) {
      self.source.lastReportDate = lastReportDate;
      self.source.save();
    }
    if (lastReportDateSavedSearch && lastReportDateSavedSearch != -Infinity && self.source.lastReportDateSavedSearch !== lastReportDateSavedSearch) {
      self.source.lastReportDateSavedSearch = lastReportDateSavedSearch;
      self.source.save();
    }
  });
};

// Fetch next batch of data from content service
PullBot.prototype.fetch = function(callback) {
  var self = this;

  if (!this.enabled || this._fetching) return callback();

  var maxCount = this.queue.capacity - this.queue.count;
  if (maxCount === 0) return callback();

  self._fetching = true;

  this.contentService.fetch({ maxCount: maxCount }, function(err, lastReportDate, lastReportDateSavedSearch) {
    self._fetching = false;
    callback(err, lastReportDate, lastReportDateSavedSearch);
  });
};

// Use random jitter to avoid clustered requests
PullBot.prototype._interval = function() {
  var diff = this.interval / 10;
  return this.interval - diff + Math.floor(Math.random() * 2 * diff + 1);
};

module.exports = PullBot;

@@ -914,16 +914,22 @@
 s.fetch(
+async 
 function
@@ -1360,32 +1360,38 @@
 portDate;%0A      
+await 
 self.source.save
@@ -1625,24 +1625,30 @@
 arch;%0A      
+await 
 self.source.

Make save() async to fix ParallelSaveError (#346)