lib/flipper/adapters/mongo.rb in flipper-mongo-0.2.2 vs lib/flipper/adapters/mongo.rb in flipper-mongo-0.3.0

@@ -15,42 +15,48 @@
       def read(key)
         find_one key
       end
 
       def write(key, value)
-        update key, {'$set' => {'v' => value}}
+        update key, {'$set' => {'v' => value.to_s}}
       end
 
       def delete(key)
-        @collection.remove criteria(key)
+        remove key
       end
 
       def set_members(key)
         (find_one(key) || Set.new).to_set
       end
 
       def set_add(key, value)
-        update key, {'$addToSet' => {'v' => value}}
+        update key, {'$addToSet' => {'v' => value.to_s}}
       end
 
       def set_delete(key, value)
-        update key, {'$pull' => {'v' => value}}
+        update key, {'$pull' => {'v' => value.to_s}}
       end
 
       private
 
       def find_one(key)
-        if (doc = @collection.find_one(criteria(key)))
+        doc = @collection.find_one(criteria(key))
+
+        unless doc.nil?
           doc['v']
         end
       end
 
       def update(key, updates)
         @collection.update criteria(key), updates, @update_options
       end
 
+      def remove(key)
+        @collection.remove criteria(key)
+      end
+
       def criteria(key)
-        {:_id => key}
+        {:_id => key.to_s}
       end
     end
   end
 end