module Nazrin
  class DataAccessor
    class Mongoid < Nazrin::DataAccessor
      def load_all(ids)
        documents_table = {}

        relation = options.reduce(model) do |rel, send_args|
          rel.send(*send_args.compact)
        end

        relation.where('_id' => { '$in' => ids }).each do |document|
          documents_table[document._id.to_s] = document
        end

        documents_table.values_at(*ids.map(&:to_s)).compact
      end

      def data_from_response(res)
        res.data.hits.hit.map(&:id)
      end
    end
  end
end