Merge pull request #131 from fatkodima/avoid-premature-ar-load

Do not prematurely load `ActiveRecord::Base`
ledermann · Nov 27, 2022 · d19c4e5 · d19c4e5
2 parents 19d5b94 + 25a9cfd
commit d19c4e5
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/lib/unread.rb b/lib/unread.rb
@@ -1,18 +1,21 @@
 require 'active_record'
 
 require 'unread/base'
-require 'unread/read_mark'
 require 'unread/readable'
 require 'unread/reader'
 require 'unread/readable_scopes'
 require 'unread/reader_scopes'
 require 'unread/garbage_collector'
 require 'unread/version'
 
-ActiveRecord::Base.send :include, Unread
-
 Unread::MIGRATION_BASE_CLASS = if ActiveRecord::VERSION::MAJOR >= 5
   ActiveRecord::Migration[5.0]
 else
   ActiveRecord::Migration
 end
+
+ActiveSupport.on_load(:active_record) do
+  require 'unread/read_mark'
+
+  include Unread
+end