lib/cachers/railtie.rb in cachers-4.0.0.0 vs lib/cachers/railtie.rb in cachers-4.1.0.0

@@ -1,10 +1,18 @@
 module Cachers
   class Railtie < Rails::Railtie
 
-    config.after_initialize do
-      Dir[Rails.root.join('app/cachers/*')].each do |file|
+    config.before_initialize do
+      Dir["#{Rails.root}/app/cachers/**/*_cacher.rb"].each do |file|
         load file
+      end
+    end
+
+    initializer 'cachers.active_record' do
+      ActiveSupport.on_load :active_record do
+        ::ActiveRecord::Base.include(
+          Cachers::Extensions::ActiveRecord::Base
+        )
       end
     end
 
     rake_tasks do
       load 'tasks/cachers.rake'