Skip to content

Instantly share code, notes, and snippets.

@public

public/gist:f3fd19bd542d795ff3b0

Created December 10, 2014 13:59
Show Gist options
  • Save public/f3fd19bd542d795ff3b0 to your computer and use it in GitHub Desktop.
Save public/f3fd19bd542d795ff3b0 to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. public created this gist Dec 10, 2014.
    11 changes: 11 additions & 0 deletions gistfile1.py
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,11 @@
    class PrefetchBatches(batches.Batches):
    def Strategy(self, task, app, consumer):
    handler = super(PrefetchBatches, self).Strategy(task, app, consumer)

    def task_handler(*args, **kwargs):
    if consumer.qos.value < self.flush_every:
    diff = self.flush_every - consumer.qos.value
    consumer.qos.increment_eventually(diff)
    return handler(*args, **kwargs)

    return task_handler