Contents

module Elastics
  class Result
    module ActiveModel

      include Elastics::Result::Scope

      # extend if the context include a Elastics::ActiveModel
      def self.should_extend?(result)
        result.variables[:context] && result.variables[:context].include?(Elastics::ActiveModel)
      end

      def get_docs(should_freeze=false)
        # super is from elastics-scopes
        docs = super()
        return docs if variables[:raw_result]
        raw_result = self
        if docs.is_a?(Array)
          res = docs.map {|doc| build_object(doc,should_freeze)}
          res.extend(Struct::Paginable).setup(raw_result['hits']['total'], variables)
          class << res; self end.class_eval do
            define_method(:raw_result){ raw_result }
            define_method(:method_missing) do |meth, *args, &block|
              raw_result.respond_to?(meth) ? raw_result.send(meth, *args, &block) : super(meth, *args, &block)
            end
          end
          res
        else
          build_object docs, should_freeze
        end
      end

    private

      def build_object(doc, should_freeze)
        attrs      = (doc['_source']||{}).merge(doc['fields']||{})
        object     = variables[:context].new attrs
        raw_result = self
        object.instance_eval do
          class << self; self end.class_eval do
            define_method(:raw_result){ raw_result }
            define_method(:raw_document){ doc }
            define_method(:respond_to?) do |*args|
              doc.respond_to?(*args) || super(*args)
            end
            define_method(:method_missing) do |meth, *args, &block|
              doc.respond_to?(meth) ? doc.send(meth, *args, &block) : super(meth, *args, &block)
            end
          end
          @_id        = doc['_id']
          @_version   = doc['_version']
          @highlight  = doc['highlight']
          # load the elastics proxy before freezing
          elastics
          self.freeze if should_freeze
        end
        object
      end

    end
  end
end

Version data entries

6 entries across 6 versions & 1 rubygems

Version	Path
elastics-models-1.3.3	lib/elastics/result/active_model.rb
elastics-models-1.3.2	lib/elastics/result/active_model.rb
elastics-models-1.3.1	lib/elastics/result/active_model.rb
elastics-models-1.3.0	lib/elastics/result/active_model.rb
elastics-models-1.2.1	lib/elastics/result/active_model.rb
elastics-models-1.2.0	lib/elastics/result/active_model.rb