Contents

require 'forwardable'
require_relative 'strategy'
require_relative 'middleware'
require_relative './util'
require_relative '../upperkut'

module Upperkut
  module Worker
    def self.included(base)
      base.extend(ClassMethods)
    end

    module ClassMethods
      extend Forwardable

      def_delegators :setup, :strategy, :middlewares
      def_delegators :strategy, :push_items, :size, :latency, :clear, :redis

      def push_items(items)
        strategy.push_items(items)
      end

      def fetch_items
        strategy.fetch_items(setup.batch_size)
      end

      def setup_upperkut
        yield(setup) if block_given?
      end

      def setup
        @config ||=
          begin
            config = Upperkut::Configuration.default.clone
            config.strategy = Upperkut::Strategy.new(self)
            config
          end
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
upperkut-0.5.1	lib/upperkut/worker.rb