populate-me 0.11.0 → 0.12.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: faa748585887ab4c87b9a96b0fcbc1cb819933840e5ed916a3455ea677c10ada
-  data.tar.gz: '0548e4f13d04c8f9a5375029fc11e6cfdd00b7759b2a6b5ba171151640ee5159'
+  metadata.gz: b2770ba89bcfa96ad56baafdf141c67891317044051f44b117a8e6b438f7c432
+  data.tar.gz: f076278f23d91e5c59c141b39698c34a9083c5861ba01f41a3692d587bc0b73b
 SHA512:
-  metadata.gz: 3244bbeb680e2d0ffea098719cc228623115666b498330651563672455c2495cfd3955270943f4b0020fc54780e876dd644a9025b0ff7f325eb016586f75cc8f
-  data.tar.gz: 3489e40c4e16e1b75c005d09937c535de3eef8d52edf8de9b1aa54a43d71a7a8304a7e80ebaaec2486da4b344b28752b1451f82e224f2014cc00d74a0e417118
+  metadata.gz: f952a4160015147d54bab9a504f189613521e33342561e95ae33710cb2c3929a9a9512a96d6b0ed2812571a3faa7c972f3b550556884e44c9c0cd7df19922ed0
+  data.tar.gz: cd10b6ec3d88432f6255803150c6c1b129b0ffab6a6736d92243238976cfbf295413ea5a1cd9251029edfd798cdead3ac98fcbf51cd3f415a0011702ada41c5b

data/lib/populate_me/document_mixins/admin_adapter.rb

@@ -76,11 +76,16 @@ module PopulateMe
         end
 
         def admin_get id
+          return self.admin_get_multiple(id) if id.is_a?(Array)
           self.cast do
-            documents.find{|doc| doc[id_string_key]==id }
+            documents.find{|doc| doc[id_string_key] == id }
           end
         end
 
+        def admin_get_multiple ids, o={sort: nil}
+          self.admin_find(o.merge(query: {id_string_key => {'$in' => ids.uniq.compact}}))
+        end
+
         def admin_find o={}
           o[:query] ||= {}
           docs = self.cast{documents}.find_all do |d|

data/lib/populate_me/mongo.rb

@@ -88,11 +88,19 @@ module PopulateMe
       end
 
       def admin_get theid
+        return self.admin_get_multiple(theid) if theid.is_a?(Array)
         theid = string_or_object_id theid
         self.cast{ collection.find({id_string_key => theid}).first }
       end
       alias_method :[], :admin_get
 
+      def admin_get_multiple theids, o={sort: nil}
+        theids = theids.uniq.compact.map{|theid| string_or_object_id(theid) }
+        self.admin_find(o.merge({
+          query: {id_string_key => {'$in' => theids} }
+        }))
+      end
+
       def admin_find o={}
         query = o.delete(:query) || {}
         o[:sort] ||= @current_sort

data/lib/populate_me/version.rb

@@ -1,4 +1,4 @@
 module PopulateMe
-  VERSION = '0.11.0'
+  VERSION = '0.12.0'
 end
 

data/test/test_mongo.rb

@@ -126,6 +126,19 @@ describe 'PopulateMe::Mongo' do
       assert_equal "regular", LowFish.admin_get(regular_fish_id).name
     end
 
+    it "Should get multiple items" do
+      # Order is not respected
+      alpha_id = LowFish.new(name: "alpha").perform_create
+      beta_id = LowFish.new(name: "beta").perform_create
+      gamma_id = LowFish.new(name: "gamma").perform_create
+      items = LowFish.admin_get([alpha_id, beta_id, nil, alpha_id])
+      assert items.is_a?(Array)
+      assert_equal 2, items.count
+      refute_nil items.find{|i| i.id == alpha_id}
+      refute_nil items.find{|i| i.id == beta_id}
+      assert_nil items.find{|i| i.id == gamma_id}
+    end
+
     it 'Should have the [] shortcut for admin_get' do
       LowFish.collection.insert_one(_id: 42, name: "H2G2")
       assert_equal LowFish[42], LowFish.admin_get(42)
@@ -138,7 +151,7 @@ describe 'PopulateMe::Mongo' do
       LowFish.collection.insert_one(_id: 40, name: "Bran")
       items = LowFish.admin_find
       assert items.is_a?(Array)
-      assert 4, items.count
+      assert_equal 4, items.count
       assert_equal 10, items[0].id
       items = LowFish.admin_find query: {name: 'Bran'}
       assert items.is_a?(Array)

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: populate-me
 version: !ruby/object:Gem::Version
-  version: 0.11.0
+  version: 0.12.0
 platform: ruby
 authors:
 - Mickael Riga
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2020-07-09 00:00:00.000000000 Z
+date: 2020-07-21 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: web-utils