lib/flipper/adapters/mongo.rb in flipper-mongo-0.10.1 vs lib/flipper/adapters/mongo.rb in flipper-mongo-0.10.2

@@ -44,24 +44,20 @@
 
       # Public: Gets the values for all gates for a given feature.
       #
       # Returns a Hash of Flipper::Gate#key => value.
       def get(feature)
-        result = {}
         doc = find(feature.key)
+        result_for_feature(feature, doc)
+      end
 
-        feature.gates.each do |gate|
-          result[gate.key] = case gate.data_type
-          when :boolean, :integer
-            doc[gate.key.to_s]
-          when :set
-            doc.fetch(gate.key.to_s) { Set.new }.to_set
-          else
-            unsupported_data_type gate.data_type
-          end
+      def get_multi(features)
+        docs = find_many(features.map(&:key))
+        result = {}
+        features.each do |feature|
+          result[feature.key] = result_for_feature(feature, docs[feature.key])
         end
-
         result
       end
 
       # Public: Enables a gate for a given thing.
       #
@@ -114,26 +110,45 @@
         raise "#{data_type} is not supported by this adapter"
       end
 
       # Private
       def find(key)
-        @collection.find(criteria(key)).limit(1).first || {}
+        @collection.find({:_id => key.to_s}).limit(1).first || {}
       end
 
+      def find_many(keys)
+        docs = @collection.find({_id: {'$in' => keys}}).to_a
+        result = Hash.new { |hash, key| hash[key] = {} }
+        docs.each do |doc|
+          result[doc["_id"]] = doc
+        end
+        result
+      end
+
       # Private
       def update(key, updates)
         options = {:upsert => true}
-        @collection.find(criteria(key)).update_one(updates, options)
+        @collection.find({:_id => key.to_s}).update_one(updates, options)
       end
 
       # Private
       def delete(key)
-        @collection.find(criteria(key)).delete_one
+        @collection.find({:_id => key.to_s}).delete_one
       end
 
-      # Private
-      def criteria(key)
-        {:_id => key.to_s}
+      def result_for_feature(feature, doc)
+        result = {}
+        feature.gates.each do |gate|
+          result[gate.key] = case gate.data_type
+          when :boolean, :integer
+            doc[gate.key.to_s]
+          when :set
+            doc.fetch(gate.key.to_s) { Set.new }.to_set
+          else
+            unsupported_data_type gate.data_type
+          end
+        end
+        result
       end
     end
   end
 end