Merge pull request #327 from eugeneius/lazy_load_hook

Use a lazy load hook to configure Active Record
attr-encrypted · Nov 13, 2018 · 3ea0c45 · 3ea0c45
2 parents ee82b3b + 333775b
commit 3ea0c45
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/lib/attr_encrypted/adapters/active_record.rb b/lib/attr_encrypted/adapters/active_record.rb
@@ -136,6 +136,8 @@ def method_missing_with_attr_encrypted(method, *args, &block)
     end
   end
 
-  ActiveRecord::Base.extend AttrEncrypted
-  ActiveRecord::Base.extend AttrEncrypted::Adapters::ActiveRecord
+  ActiveSupport.on_load(:active_record) do
+    extend AttrEncrypted
+    extend AttrEncrypted::Adapters::ActiveRecord
+  end
 end