Contents

require 'json'
require 'upperkut/item'

module Upperkut
  module Util
    def to_underscore(object)
      klass_name = object.dup
      klass_name.gsub!(/::/, '_')
      klass_name.gsub!(/([A-Z\d]+)([A-Z][a-z])/, '\1_\2')
      klass_name.gsub!(/([a-z\d])([A-Z])/, '\1_\2')
      klass_name.tr!('-', '_')
      klass_name.downcase!
      klass_name
    end

    # Public:
    #  Normalize hash and hash arrays into a hash of Items.
    #  An Item object contains metadata, for example the timestamp from the moment it was enqueued,
    #   that we need to carry through multiple execution tries.
    #
    #  When the execution fails, we need to schedule the whole batch for retry, and scheduling
    #   an Item will make Upperkut understand that we're not dealing with a new batch,
    #   so metrics like latency will increase.
    def normalize_items(items)
      items = [items] unless items.is_a?(Array)

      items.map do |item|
        next item if item.is_a?(Item)

        Item.new(id: SecureRandom.uuid, body: item)
      end
    end

    def encode_json_items(items)
      items = [items] unless items.is_a?(Array)

      items.map do |item|
        JSON.generate(
          'id' => item.id,
          'body' => item.body,
          'enqueued_at' => item.enqueued_at
        )
      end
    end

    def decode_json_items(items)
      items.each_with_object([]) do |item_json, memo|
        next unless item_json

        hash = JSON.parse(item_json)
        id, body, enqueued_at = hash.values_at('id', 'body', 'enqueued_at')
        memo << Item.new(id: id, body: body, enqueued_at: enqueued_at)
      end
    end

    def retry_block(retries_limit = 3, base_sleep = 2)
      retries = 0

      begin
        yield
      rescue StandardError => err
        if retries < retries_limit
          retries += 1
          sleep_time = base_sleep**retries
          Kernel.sleep(sleep_time)
          retry
        end

        raise err
      end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
upperkut-1.0.4	lib/upperkut/util.rb
upperkut-1.0.2	lib/upperkut/util.rb