Add ids method as in AR

Author: nsauk

lib/active_graph/node/query/query_proxy_methods.rb

@@ -270,6 +270,11 @@ def first_and_last(func, target)
           end.first
         end
 
+        # @return [Array] An array of primary key values
+        def ids
+          pluck(association_id_key)
+        end
+
         # @return [String] The primary key of a the current QueryProxy's model or target class
         def association_id_key
           self.association.nil? ? model.primary_key : self.association.target_class.primary_key

lib/active_graph/node/query_methods.rb

@@ -49,6 +49,13 @@ def find_each(options = {})
         end
       end
 
+      # Returns an array of all the IDs (primary key values) of the model's nodes.
+      # ActiveRecord-compatible shortcut for `pluck(primary_key)`.
+      # @return [Array] An array of primary key values
+      def ids
+        self.all.ids
+      end
+
       private
 
       def exists_query_start(condition)

spec/e2e/query_proxy_methods_spec.rb

@@ -444,6 +444,34 @@ def destroy_called
     end
   end
 
+  describe 'ids' do
+    before(:each) do
+      [Student, Lesson].each(&:delete_all)
+
+      @john = Student.create(name: 'John')
+      @history = Lesson.create(name: 'history')
+      @math2 = Lesson.create(name: 'math2')
+      @john.lessons << @history
+      @john.lessons << @math2
+    end
+
+    it 'returns the ids of nodes on the class' do
+      expect(Lesson.ids).to match_array([@history.id, @math2.id])
+    end
+
+    it 'returns the ids of nodes on a query proxy association' do
+      expect(@john.lessons.ids).to match_array([@history.id, @math2.id])
+    end
+
+    it 'returns the ids of nodes on a where chain' do
+      expect(Lesson.where(name: 'history').ids).to eq([@history.id])
+    end
+
+    it 'returns an empty array when there are no matches' do
+      expect(Lesson.where(name: 'nope').ids).to eq([])
+    end
+  end
+
   describe 'distinct' do
     let(:frank) { Student.create(name: 'Frank') }
     let(:bill) { Teacher.create(name: 'Bill') }