Contents

# frozen_string_literal: true

module BulkLoader
  class LoaderNotFoundError < StandardError; end

  class ClassAttribute
    def initialize
      @loader_of = {}

      @no_autoload_of = {}
    end

    def include?(name)
      @loader_of.include?(name)
    end

    def autoload?(name)
      !@no_autoload_of[name]
    end

    def each(&block)
      @loader_of.each(&block)
    end

    def load(loader_names, attributes, *args)
      attrs = convert_attributes(attributes)

      loader_names = [loader_names] unless loader_names.is_a?(Array)
      loader_names.each do |name|
        raise LoaderNotFoundError, "bulk_loader :#{name} is not defined" unless @loader_of[name]

        @loader_of[name].load(attrs.map { |attr| attr.lazy(name) }, *args)
      end
    end

    def define_loader(name, loader, autoload: true)
      @loader_of[name] = loader
      @no_autoload_of[name] = true unless autoload
      define_singleton_method(name) { loader }
    end

    def inspect
      "#<BulkLoader::ClassAttribute:#{object_id} #{loader_inspect}>"
    end

    def dup
      new_object = self.class.new
      each do |key, loader|
        new_object.define_loader(key, loader)
      end
      new_object
    end

    private

    def loader_inspect
      @loader_of.map { |name, _| name }.join(' ')
    end

    def convert_attributes(attributes)
      attributes.map(&method(:convert_attribute))
    end

    def convert_attribute(attr)
      if attr.respond_to?(:bulk_loader)
        attr.bulk_loader
      elsif attr.respond_to?(:lazy)
        attr
      else
        raise 'attributes should be BulkLoader::Attribute or BulkLoader::DSL included class!!'
      end
    end
  end
end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
bulk_loader-2.0.0	lib/bulk_loader/class_attribute.rb
bulk_loader-1.3.1	lib/bulk_loader/class_attribute.rb
bulk_loader-1.3.0	lib/bulk_loader/class_attribute.rb
bulk_loader-1.2.0	lib/bulk_loader/class_attribute.rb