change Bullet's feature enable logic

flyerhzm · Feb 24, 2025 · 99817ee · 99817ee
1 parent 86654e7
commit 99817ee
Showing 1 changed file with 20 additions and 4 deletions.
diff --git a/lib/bullet.rb b/lib/bullet.rb
@@ -64,7 +64,11 @@ def raise=(should_raise)
     ].freeze
 
     def enable=(enable)
-      @enable = @n_plus_one_query_enable = @unused_eager_loading_enable = @counter_cache_enable = enable
+      @enable = enable
+
+      @n_plus_one_query_enable ||= true
+      @unused_eager_loading_enable ||= true
+      @counter_cache_enable ||= true
 
       if enable?
         reset_safelist
@@ -90,15 +94,27 @@ def app_root
     end
 
     def n_plus_one_query_enable?
-      enable? && !!@n_plus_one_query_enable
+      enable? && default_n_plus_one_query_enable?
     end
 
     def unused_eager_loading_enable?
-      enable? && !!@unused_eager_loading_enable
+      enable? && default_unused_eager_loading_enable?
     end
 
     def counter_cache_enable?
-      enable? && !!@counter_cache_enable
+      enable? && default_counter_cache_enable?
+    end
+
+    def default_n_plus_one_query_enable?
+      !!@n_plus_one_query_enable
+    end
+
+    def default_unused_eager_loading_enable?
+      !!@unused_eager_loading_enable
+    end
+
+    def default_counter_cache_enable?
+      !!@counter_cache_enable
     end
 
     def stacktrace_includes