Contents

# frozen_string_literal: true

module Esse
  class Repository
    module ClassMethods
      def import(**kwargs)
        index.import(repo_name, **kwargs)
      end

      def update_documents_attribute(name, *ids_or_doc_headers, **kwargs)
        batch = documents_for_lazy_attribute(name, *ids_or_doc_headers)
        return if batch.empty?

        index.bulk(**kwargs, update: batch)
      end

      def documents_for_lazy_attribute(name, *ids_or_doc_headers)
        retrieve_lazy_attribute_values(name, *ids_or_doc_headers).map do |doc_header, datum|
          doc_header.to_doc(name => datum)
        end
      end

      def retrieve_lazy_attribute_values(name, *ids_or_doc_headers)
        unless lazy_document_attribute?(name)
          raise ArgumentError, <<~MSG
            The attribute `#{name}` is not defined as a lazy document attribute.

            Define the attribute as a lazy document attribute using the `lazy_document_attribute` method.
          MSG
        end

        docs = LazyDocumentHeader.coerce_each(ids_or_doc_headers)
        return [] if docs.empty?

        result = fetch_lazy_document_attribute(name).call(docs)
        return [] unless result.is_a?(Hash)

        result.each_with_object({}) do |(key, value), memo|
          if key.is_a?(LazyDocumentHeader) && (doc = docs.find { |d| d == key || d.id == key.id })
            memo[doc] = value
          elsif (doc = docs.find { |d| d.id == key })
            memo[doc] = value
          end
        end
      end
    end

    extend ClassMethods
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
esse-0.3.1	lib/esse/repository/documents.rb