elastic_record 0.2.0 → 0.2.1

data/lib/elastic_record/relation.rb

@@ -35,7 +35,10 @@ module ElasticRecord
     end
 
     def to_a
-      @records ||= klass.find(to_ids)
+      @records ||= begin
+        scope = select_values.any? ? klass.select(select_values) : klass
+        scope.find(to_ids)
+      end
     end
 
     def to_ids

data/lib/elastic_record/relation/search_methods.rb

@@ -60,6 +60,19 @@ module ElasticRecord
       clone.offset! value
     end
 
+    def select!(value)
+      self.select_values += Array.wrap(value)
+      self
+    end
+
+    def select(value, &block)
+      if block_given?
+        to_a.select(&block)
+      else
+        clone.select! value
+      end
+    end
+
     def facet!(name_or_facet, options = {})
       if name_or_facet.is_a?(String)
         self.facet_values += [arelastic.facet[name_or_facet].terms(name_or_facet, options)]

data/lib/elastic_record/relation/value_methods.rb

@@ -1,6 +1,6 @@
 module ElasticRecord
   class Relation
-    MULTI_VALUE_METHODS  = [:extending, :filter, :facet, :order]
+    MULTI_VALUE_METHODS  = [:extending, :facet, :filter, :order, :select]
     SINGLE_VALUE_METHODS = [:query, :limit, :offset]
   end
 end

data/test/elastic_record/relation/search_methods_test.rb

@@ -128,6 +128,20 @@ class ElasticRecord::Relation::SearchMethodsTest < MiniTest::Spec
     assert_equal expected, relation.as_elastic['sort']
   end
 
+  def test_select
+    selectable_klass = Widget.anon do
+      def self.select(values)
+        @latest_select = values
+        self
+      end
+    end
+
+    relation = selectable_klass.elastic_relation.select 'foo'
+    relation.to_a
+
+    assert_equal ['foo'], selectable_klass.instance_variable_get('@latest_select')
+  end
+
   def test_extending_with_block
     relation.extending! do
       def foo

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: elastic_record
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.2.1
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-08-09 00:00:00.000000000 Z
+date: 2012-08-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: arelastic