pluck_each 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: e38a2e5de6b397dab4950becd1de9a80b703dd4e
-  data.tar.gz: 0f3798ecff31d3f35ece1f6ecedcea9ce657a233
+  metadata.gz: a808f819a95611a329a5a724924b41ff62e6bbb7
+  data.tar.gz: d2c9cc635521e5ffd1fbc5e8f09c9dcf23d3a2c1
 SHA512:
-  metadata.gz: 4cfd3aff95b389605210a1721da6361de08ef10ed15f70e269be0f846f48811afa7f4791fa559bdc13def4da414a0545e61a64fb2b1b118c144971740d9f753f
-  data.tar.gz: d3009c05a01c391bda8294158abab7fe8d45ad432fefa64304cd99a50f4c0f11d675d68d9c9ad3b050709a4e955eb7fca651bb8f1f6948ff9220d87d0a11178b
+  metadata.gz: 0c72e5d7bef5021a2d9563fea25c94f9abbdd78f7c6ec2b497134a3d66816df7e902fe8b7efe9ae4817819ad379bbcf44cc4b16d0ba9c54bdac7e2031967b38f
+  data.tar.gz: d332a5c55fa5a6d969172d908e13f50e1f5c810d95cd64a146ebdf326201aeb406b180e5627cfa624fa722598e6393638271434befe8dee501521b71373ee451

data/lib/pluck_each/version.rb

@@ -1,3 +1,3 @@
 module PluckEach
-  VERSION = "0.1.0"
+  VERSION = "0.1.1"
 end

data/lib/pluck_each.rb

@@ -20,6 +20,7 @@ module ActiveRecord
 
       # Ensure the primary key is selected so we can use it as an offset
       id_in_columns_requested = string_column_names.include?(primary_key)
+      id_only_field_requested = id_in_columns_requested && string_column_names.size == 1
       string_column_names.unshift(primary_key) unless id_in_columns_requested
       id_position_in_response = string_column_names.index(primary_key)
 
@@ -33,12 +34,16 @@ module ActiveRecord
         batch = batch_relation.pluck(*string_column_names)
         break if batch.empty?
 
-        primary_key_offset = batch.last.at(id_position_in_response)
+        primary_key_offset = batch.last
 
-        unless id_in_columns_requested
-          batch.collect! do |record|
-            record.delete_at(id_position_in_response)
-            record
+        if !id_only_field_requested
+          primary_key_offset = primary_key_offset.at(id_position_in_response)
+
+          unless id_in_columns_requested
+            batch.collect! do |record|
+              record.delete_at(id_position_in_response)
+              record
+            end
           end
         end
 

data/spec/pluck_each_spec.rb

@@ -30,6 +30,16 @@ describe PluckEach do
       User.create(:first_name => '5', :last_name => '5')
     end
 
+    it 'plucks :id when only field requested' do
+      values = []
+      User.all.pluck_each(:id) do |id|
+        values << id
+      end
+
+      values.sort!
+      values.must_equal User.all.pluck(:id).sort
+    end
+
     it 'plucks only the fields requested' do
       values = []
       User.all.pluck_each(:first_name) do |first_name|

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: pluck_each
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - Brandon Dewitt