lib/flip/database_strategy.rb in flip-1.0.1 vs lib/flip/database_strategy.rb in flip-1.1.0

@@ -33,10 +33,14 @@
     end
 
     private
 
     def feature(definition)
-      @klass.where(key: definition.key.to_s).first
+      if @klass.respond_to?(:use_feature_cache) && @klass.use_feature_cache
+        @klass.feature_cache[definition.key.to_s]
+      else
+        @klass.where(key: definition.key.to_s).first
+      end
     end
 
   end
 end