istox-resque-scheduler 1.0.0.pre

data/lib/resque/scheduler/cli.rb

@@ -0,0 +1,147 @@
+# vim:fileencoding=utf-8
+
+require 'optparse'
+
+module Resque
+  module Scheduler
+    CLI_OPTIONS_ENV_MAPPING = {
+      app_name: 'APP_NAME',
+      background: 'BACKGROUND',
+      dynamic: 'DYNAMIC_SCHEDULE',
+      env: 'RAILS_ENV',
+      initializer_path: 'INITIALIZER_PATH',
+      logfile: 'LOGFILE',
+      logformat: 'LOGFORMAT',
+      quiet: 'QUIET',
+      pidfile: 'PIDFILE',
+      poll_sleep_amount: 'RESQUE_SCHEDULER_INTERVAL',
+      verbose: 'VERBOSE'
+    }.freeze
+
+    class Cli
+      BANNER = <<-EOF.gsub(/ {6}/, '')
+      Usage: resque-scheduler [options]
+
+      Runs a resque scheduler process directly (rather than via rake).
+
+      EOF
+      OPTIONS = [
+        {
+          args: ['-n', '--app-name [APP_NAME]',
+                 'Application name for procline'],
+          callback: ->(options) { ->(n) { options[:app_name] = n } }
+        },
+        {
+          args: ['-B', '--background', 'Run in the background [BACKGROUND]'],
+          callback: ->(options) { ->(b) { options[:background] = b } }
+        },
+        {
+          args: ['-D', '--dynamic-schedule',
+                 'Enable dynamic scheduling [DYNAMIC_SCHEDULE]'],
+          callback: ->(options) { ->(d) { options[:dynamic] = d } }
+        },
+        {
+          args: ['-E', '--environment [RAILS_ENV]', 'Environment name'],
+          callback: ->(options) { ->(e) { options[:env] = e } }
+        },
+        {
+          args: ['-I', '--initializer-path [INITIALIZER_PATH]',
+                 'Path to optional initializer ruby file'],
+          callback: ->(options) { ->(i) { options[:initializer_path] = i } }
+        },
+        {
+          args: ['-i', '--interval [RESQUE_SCHEDULER_INTERVAL]',
+                 'Interval for checking if a scheduled job must run'],
+          callback: ->(options) { ->(i) { options[:poll_sleep_amount] = i } }
+        },
+        {
+          args: ['-l', '--logfile [LOGFILE]', 'Log file name'],
+          callback: ->(options) { ->(l) { options[:logfile] = l } }
+        },
+        {
+          args: ['-F', '--logformat [LOGFORMAT]', 'Log output format'],
+          callback: ->(options) { ->(f) { options[:logformat] = f } }
+        },
+        {
+          args: ['-P', '--pidfile [PIDFILE]', 'PID file name'],
+          callback: ->(options) { ->(p) { options[:pidfile] = p } }
+        },
+        {
+          args: ['-q', '--quiet', 'Run with minimal output [QUIET]'],
+          callback: ->(options) { ->(q) { options[:quiet] = q } }
+        },
+        {
+          args: ['-v', '--verbose', 'Run with verbose output [VERBOSE]'],
+          callback: ->(options) { ->(v) { options[:verbose] = v } }
+        }
+      ].freeze
+
+      def self.run!(argv = ARGV, env = ENV)
+        new(argv, env).run!
+      end
+
+      def initialize(argv = ARGV, env = ENV)
+        @argv = argv
+        @env = env
+      end
+
+      def run!
+        pre_run
+        run_forever
+      end
+
+      def pre_run
+        parse_options
+        pre_setup
+        setup_env
+      end
+
+      def parse_options
+        option_parser.parse!(argv.dup)
+      end
+
+      def pre_setup
+        if options[:initializer_path]
+          load options[:initializer_path].to_s.strip
+        else
+          false
+        end
+      end
+
+      def setup_env
+        require_relative 'env'
+        runtime_env.setup
+      end
+
+      def run_forever
+        Resque::Scheduler.run
+      end
+
+      private
+
+      attr_reader :argv, :env
+
+      def runtime_env
+        @runtime_env ||= Resque::Scheduler::Env.new(options)
+      end
+
+      def option_parser
+        OptionParser.new do |opts|
+          opts.banner = BANNER
+          opts.version = Resque::Scheduler::VERSION
+          OPTIONS.each do |opt|
+            opts.on(*opt[:args], &opt[:callback].call(options))
+          end
+        end
+      end
+
+      def options
+        @options ||= {}.tap do |o|
+          CLI_OPTIONS_ENV_MAPPING.each do |key, envvar|
+            o[key] = env[envvar] if env.include?(envvar)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/resque/scheduler/configuration.rb

@@ -0,0 +1,73 @@
+# vim:fileencoding=utf-8
+
+module Resque
+  module Scheduler
+    module Configuration
+      # Allows for block-style configuration
+      def configure
+        yield self
+      end
+
+      # Used in `#load_schedule_job`
+      attr_writer :env
+
+      def env
+        return @env if @env
+        @env ||= Rails.env if defined?(Rails) && Rails.respond_to?(:env)
+        @env ||= ENV['RAILS_ENV']
+        @env
+      end
+
+      # If true, logs more stuff...
+      attr_writer :verbose
+
+      def verbose
+        @verbose ||= !!ENV['VERBOSE']
+      end
+
+      # If set, produces no output
+      attr_writer :quiet
+
+      def quiet
+        @quiet ||= !!ENV['QUIET']
+      end
+
+      # If set, will write messages to the file
+      attr_writer :logfile
+
+      def logfile
+        @logfile ||= ENV['LOGFILE']
+      end
+
+      # Sets whether to log in 'text' or 'json'
+      attr_writer :logformat
+
+      def logformat
+        @logformat ||= ENV['LOGFORMAT']
+      end
+
+      # If set, will try to update the schedule in the loop
+      attr_writer :dynamic
+
+      def dynamic
+        @dynamic ||= !!ENV['DYNAMIC_SCHEDULE']
+      end
+
+      # If set, will append the app name to procline
+      attr_writer :app_name
+
+      def app_name
+        @app_name ||= ENV['APP_NAME']
+      end
+
+      # Amount of time in seconds to sleep between polls of the delayed
+      # queue.  Defaults to 5
+      attr_writer :poll_sleep_amount
+
+      def poll_sleep_amount
+        @poll_sleep_amount ||=
+          Float(ENV.fetch('RESQUE_SCHEDULER_INTERVAL', '5'))
+      end
+    end
+  end
+end

data/lib/resque/scheduler/delaying_extensions.rb

@@ -0,0 +1,356 @@
+# vim:fileencoding=utf-8
+require 'resque'
+require_relative 'plugin'
+require_relative '../scheduler'
+
+module Resque
+  module Scheduler
+    module DelayingExtensions
+      # This method is nearly identical to +enqueue+ only it also
+      # takes a timestamp which will be used to schedule the job
+      # for queueing.  Until timestamp is in the past, the job will
+      # sit in the schedule list.
+      def enqueue_at(timestamp, klass, *args)
+        validate(klass)
+        enqueue_at_with_queue(
+          queue_from_class(klass), timestamp, klass, *args
+        )
+      end
+
+      # Identical to +enqueue_at+, except you can also specify
+      # a queue in which the job will be placed after the
+      # timestamp has passed. It respects Resque.inline option, by
+      # creating the job right away instead of adding to the queue.
+      def enqueue_at_with_queue(queue, timestamp, klass, *args)
+        return false unless plugin.run_before_schedule_hooks(klass, *args)
+
+        if Resque.inline? || timestamp.to_i <= Time.now.to_i
+          # Just create the job and let resque perform it right away with
+          # inline.  If the class is a custom job class, call self#scheduled
+          # on it. This allows you to do things like
+          # Resque.enqueue_at(timestamp, CustomJobClass, :opt1 => val1).
+          # Otherwise, pass off to Resque.
+          if klass.respond_to?(:scheduled)
+            klass.scheduled(queue, klass.to_s, *args)
+          else
+            Resque.enqueue_to(queue, klass, *args)
+          end
+        else
+          delayed_push(timestamp, job_to_hash_with_queue(queue, klass, args))
+        end
+
+        plugin.run_after_schedule_hooks(klass, *args)
+      end
+
+      # Identical to enqueue_at but takes number_of_seconds_from_now
+      # instead of a timestamp.
+      def enqueue_in(number_of_seconds_from_now, klass, *args)
+        unless number_of_seconds_from_now.is_a?(Numeric)
+          raise ArgumentError, 'Please supply a numeric number of seconds'
+        end
+        enqueue_at(Time.now + number_of_seconds_from_now, klass, *args)
+      end
+
+      # Identical to +enqueue_in+, except you can also specify
+      # a queue in which the job will be placed after the
+      # number of seconds has passed.
+      def enqueue_in_with_queue(queue, number_of_seconds_from_now,
+                                klass, *args)
+        unless number_of_seconds_from_now.is_a?(Numeric)
+          raise ArgumentError, 'Please supply a numeric number of seconds'
+        end
+        enqueue_at_with_queue(queue, Time.now + number_of_seconds_from_now,
+                              klass, *args)
+      end
+
+      # Used internally to stuff the item into the schedule sorted list.
+      # +timestamp+ can be either in seconds or a datetime object. The
+      # insertion time complexity is O(log(n)). Returns true if it's
+      # the first job to be scheduled at that time, else false.
+      def delayed_push(timestamp, item)
+        # First add this item to the list for this timestamp
+        redis.rpush("delayed:#{timestamp.to_i}", encode(item))
+
+        # Store the timestamps at with this item occurs
+        redis.sadd("timestamps:#{encode(item)}", "delayed:#{timestamp.to_i}")
+
+        # Now, add this timestamp to the zsets.  The score and the value are
+        # the same since we'll be querying by timestamp, and we don't have
+        # anything else to store.
+        redis.zadd :delayed_queue_schedule, timestamp.to_i, timestamp.to_i
+      end
+
+      # Returns an array of timestamps based on start and count
+      def delayed_queue_peek(start, count)
+        result = redis.zrange(:delayed_queue_schedule, start,
+                              start + count - 1)
+        Array(result).map(&:to_i)
+      end
+
+      # Returns the size of the delayed queue schedule
+      # this does not represent the number of items in the queue to be scheduled
+      def delayed_queue_schedule_size
+        redis.zcard :delayed_queue_schedule
+      end
+
+      # Returns the number of jobs for a given timestamp in the delayed queue
+      # schedule
+      def delayed_timestamp_size(timestamp)
+        redis.llen("delayed:#{timestamp.to_i}").to_i
+      end
+
+      # Returns an array of delayed items for the given timestamp
+      def delayed_timestamp_peek(timestamp, start, count)
+        if 1 == count
+          r = list_range "delayed:#{timestamp.to_i}", start, count
+          r.nil? ? [] : [r]
+        else
+          list_range "delayed:#{timestamp.to_i}", start, count
+        end
+      end
+
+      # Returns the next delayed queue timestamp
+      # (don't call directly)
+      def next_delayed_timestamp(at_time = nil)
+        search_first_delayed_timestamp_in_range(nil, at_time || Time.now)
+      end
+
+      # Returns the next item to be processed for a given timestamp, nil if
+      # done. (don't call directly)
+      # +timestamp+ can either be in seconds or a datetime
+      def next_item_for_timestamp(timestamp)
+        key = "delayed:#{timestamp.to_i}"
+
+        encoded_item = redis.lpop(key)
+        redis.srem("timestamps:#{encoded_item}", key)
+        item = decode(encoded_item)
+
+        # If the list is empty, remove it.
+        clean_up_timestamp(key, timestamp)
+        item
+      end
+
+      # Clears all jobs created with enqueue_at or enqueue_in
+      def reset_delayed_queue
+        Array(redis.zrange(:delayed_queue_schedule, 0, -1)).each do |item|
+          key = "delayed:#{item}"
+          items = redis.lrange(key, 0, -1)
+          redis.pipelined do
+            items.each { |ts_item| redis.del("timestamps:#{ts_item}") }
+          end
+          redis.del key
+        end
+
+        redis.del :delayed_queue_schedule
+      end
+
+      # Given an encoded item, remove it from the delayed_queue
+      def remove_delayed(klass, *args)
+        search = encode(job_to_hash(klass, args))
+        remove_delayed_job(search)
+      end
+
+      def remove_delayed_in_queue(klass, queue, *args)
+        search = encode(job_to_hash_with_queue(queue, klass, args))
+        remove_delayed_job(search)
+      end
+
+      # Given an encoded item, enqueue it now
+      def enqueue_delayed(klass, *args)
+        hash = job_to_hash(klass, args)
+        remove_delayed(klass, *args).times do
+          Resque::Scheduler.enqueue_from_config(hash)
+        end
+      end
+
+      def enqueue_delayed_with_queue(klass, queue, *args)
+        hash = job_to_hash_with_queue(queue, klass, args)
+        remove_delayed_in_queue(klass, queue, *args).times do
+          Resque::Scheduler.enqueue_from_config(hash)
+        end
+      end
+
+      # Given a block, remove jobs that return true from a block
+      #
+      # This allows for removal of delayed jobs that have arguments matching
+      # certain criteria
+      def remove_delayed_selection(klass = nil)
+        raise ArgumentError, 'Please supply a block' unless block_given?
+
+        found_jobs = find_delayed_selection(klass) { |args| yield(args) }
+        found_jobs.reduce(0) do |sum, encoded_job|
+          sum + remove_delayed_job(encoded_job)
+        end
+      end
+
+      # Given a block, enqueue jobs now that return true from a block
+      #
+      # This allows for enqueuing of delayed jobs that have arguments matching
+      # certain criteria
+      def enqueue_delayed_selection(klass = nil)
+        raise ArgumentError, 'Please supply a block' unless block_given?
+
+        found_jobs = find_delayed_selection(klass) { |args| yield(args) }
+        found_jobs.reduce(0) do |sum, encoded_job|
+          decoded_job = decode(encoded_job)
+          klass = Util.constantize(decoded_job['class'])
+          queue = decoded_job['queue']
+
+          if queue
+            jobs_queued = enqueue_delayed_with_queue(klass, queue, *decoded_job['args'])
+          else
+            jobs_queued = enqueue_delayed(klass, *decoded_job['args'])
+          end
+
+          jobs_queued + sum
+        end
+      end
+
+      # Given a block, find jobs that return true from a block
+      #
+      # This allows for finding of delayed jobs that have arguments matching
+      # certain criteria
+      def find_delayed_selection(klass = nil, &block)
+        raise ArgumentError, 'Please supply a block' unless block_given?
+
+        timestamps = redis.zrange(:delayed_queue_schedule, 0, -1)
+
+        # Beyond 100 there's almost no improvement in speed
+        found = timestamps.each_slice(100).map do |ts_group|
+          jobs = redis.pipelined do |r|
+            ts_group.each do |ts|
+              r.lrange("delayed:#{ts}", 0, -1)
+            end
+          end
+
+          jobs.flatten.select do |payload|
+            payload_matches_selection?(decode(payload), klass, &block)
+          end
+        end
+
+        found.flatten
+      end
+
+      # Given a timestamp and job (klass + args) it removes all instances and
+      # returns the count of jobs removed.
+      #
+      # O(N) where N is the number of jobs scheduled to fire at the given
+      # timestamp
+      def remove_delayed_job_from_timestamp(timestamp, klass, *args)
+        return 0 if Resque.inline?
+
+        key = "delayed:#{timestamp.to_i}"
+        encoded_job = encode(job_to_hash(klass, args))
+
+        redis.srem("timestamps:#{encoded_job}", key)
+        count = redis.lrem(key, 0, encoded_job)
+        clean_up_timestamp(key, timestamp)
+
+        count
+      end
+
+      def count_all_scheduled_jobs
+        total_jobs = 0
+        Array(redis.zrange(:delayed_queue_schedule, 0, -1)).each do |ts|
+          total_jobs += redis.llen("delayed:#{ts}").to_i
+        end
+        total_jobs
+      end
+
+      # Discover if a job has been delayed.
+      # Examples
+      #   Resque.delayed?(MyJob)
+      #   Resque.delayed?(MyJob, id: 1)
+      # Returns true if the job has been delayed
+      def delayed?(klass, *args)
+        !scheduled_at(klass, *args).empty?
+      end
+
+      # Returns delayed jobs schedule timestamp for +klass+, +args+.
+      def scheduled_at(klass, *args)
+        search = encode(job_to_hash(klass, args))
+        redis.smembers("timestamps:#{search}").map do |key|
+          key.tr('delayed:', '').to_i
+        end
+      end
+
+      def last_enqueued_at(job_name, date)
+        redis.hset('delayed:last_enqueued_at', job_name, date)
+      end
+
+      def get_last_enqueued_at(job_name)
+        redis.hget('delayed:last_enqueued_at', job_name)
+      end
+
+      private
+
+      def job_to_hash(klass, args)
+        { class: klass.to_s, args: args, queue: queue_from_class(klass) }
+      end
+
+      def job_to_hash_with_queue(queue, klass, args)
+        { class: klass.to_s, args: args, queue: queue }
+      end
+
+      # Removes a job from the queue, but not modify the timestamp schedule. This method
+      # will not effect the output of `delayed_queue_schedule_size`
+      def remove_delayed_job(encoded_job)
+        return 0 if Resque.inline?
+
+        timestamps = redis.smembers("timestamps:#{encoded_job}")
+
+        replies = redis.pipelined do
+          timestamps.each do |key|
+            redis.lrem(key, 0, encoded_job)
+            redis.srem("timestamps:#{encoded_job}", key)
+          end
+        end
+
+        # timestamp key is not removed from the schedule, this is done later
+        # by the scheduler loop
+
+        return 0 if replies.nil? || replies.empty?
+        replies.each_slice(2).map(&:first).inject(:+)
+      end
+
+      def clean_up_timestamp(key, timestamp)
+        # Use a watch here to ensure nobody adds jobs to this delayed
+        # queue while we're removing it.
+        redis.watch(key) do
+          if redis.llen(key).to_i == 0
+            # If the list is empty, remove it.
+            redis.multi do
+              redis.del(key)
+              redis.zrem(:delayed_queue_schedule, timestamp.to_i)
+            end
+          else
+            redis.redis.unwatch
+          end
+        end
+      end
+
+      def search_first_delayed_timestamp_in_range(start_at, stop_at)
+        start_at = start_at.nil? ? '-inf' : start_at.to_i
+        stop_at = stop_at.nil? ? '+inf' : stop_at.to_i
+
+        items = redis.zrangebyscore(
+          :delayed_queue_schedule, start_at, stop_at,
+          limit: [0, 1]
+        )
+        timestamp = items.nil? ? nil : Array(items).first
+        timestamp.to_i unless timestamp.nil?
+      end
+
+      def payload_matches_selection?(decoded_payload, klass)
+        return false if decoded_payload.nil?
+        job_class = decoded_payload['class']
+        relevant_class = (klass.nil? || klass.to_s == job_class)
+        relevant_class && yield(decoded_payload['args'])
+      end
+
+      def plugin
+        Resque::Scheduler::Plugin
+      end
+    end
+  end
+end