Contents

require "batch_loader/executor"

class BatchLoader
  class ExecutorProxy
    attr_reader :block, :global_executor

    def initialize(&block)
      @block = block
      @block_hash_key = block.source_location
      @global_executor = BatchLoader::Executor.ensure_current
    end

    def add(item:)
      items << item
    end

    def list_items
      items.to_a
    end

    def delete_items
      global_executor.items_by_block[@block_hash_key] = Set.new
    end

    def load(item:, value:)
      loaded[item] = value
    end

    def loaded_value(item:)
      loaded[item]
    end

    def value_loaded?(item:)
      loaded.key?(item)
    end

    def unload_value(item:)
      loaded.delete(item)
    end

    private

    def items
      global_executor.items_by_block[@block_hash_key]
    end

    def loaded
      global_executor.loaded_values_by_block[@block_hash_key]
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
batch-loader-0.3.0	lib/batch_loader/executor_proxy.rb
batch-loader-0.2.0	lib/batch_loader/executor_proxy.rb