RailsでSTI親クラスのリレーションが子クラスに引き継がれない

Question

前提・実現したいこと

RailsでSTIを用いた開発を行っています。
そこで、子クラスのロードがされない問題が発生したため https://guides.rubyonrails.org/autoloading_and_reloading_constants.html#single-table-inheritance を参考に、STI子クラスの自動ロードをするコードを書きました。

発生している問題・エラーメッセージ

rails consoleにて次のコードを実行する

>> user = User.create!(name: "Hoge")
=> User id=1

>> tweet = user.normal_tweets.create!(text: "Fuga")
=> NormalTweet id=1

>> tweet.author
=> User id=1

rails consoleを再起動し、次のコードを実行する

>> Tweet.find(1).author
=> C:/Ruby30-x64/lib/ruby/gems/3.0.0/gems/activerecord-7.0.1/lib/active_record/associations.rb:301:in `association': Association named 'author' was not found on NormalTweet; perhaps you misspelled it? (ActiveRecord::AssociationNotFoundError)

このように、エラーが発生します。※Tweetモデルからinclude StiPreloadを削除した場合この問題は起きません。

>> Tweet.find(1).author_id
=> 1

もちろん、外部キーであるauthor_id自体は存在しています。

該当のソースコード

class User < ApplicationRecord
  has_many :normal_tweets, dependent: :destroy, class_name: "NormalTweet", foreign_key: :author_id
end

require "sti_preload"

class Tweet < ApplicationRecord
  include StiPreload

  belongs_to :author, class_name: "User"
end

class NormalTweet < Tweet
  has_many :replies, class_name: "Reply", foreign_key: :reply_to_id, dependent: :nullify
end

class Reply < NormalTweet
  belongs_to :reply_to, class_name: "NormalTweet", optional: true
end

module StiPreload
  unless Rails.application.config.eager_load
    extend ActiveSupport::Concern

    included do
      cattr_accessor :preloaded, instance_accessor: false
      cattr_accessor :preloading, instance_accessor: false
    end

    class_methods do
      def descendants
        preload_sti unless preloaded
        super
      end

      def preload_sti
        return if self.preloading
        self.preloading = true

        types_in_db = \
          base_class.
          unscoped.
          select(inheritance_column).
          distinct.
          pluck(inheritance_column).
          compact

        types_in_db.each do |type|
          logger.debug("Preloading STI type #{type}")
          type.safe_constantize # 重要: safeでないconstantizeを使うと例外が発生する
        end

        self.preloading = false
        self.preloaded = true
      end
    end
  end
end

https://github.com/PJZ9n/RailsSTIProblem

試したこと

StiPreloadモジュールを使わず、手動でロードしてやると問題なく動作します。

>> [Tweet, NormalTweet, Reply]

補足情報（FW/ツールのバージョンなど）

$ rails -v Rails 7.0.1
$ ruby -v ruby 3.0.3p157 (2021-11-24 revision 3fb7d2cadc) [x64-mingw32]

Answer 1

変更前は

class Tweet < ApplicationRecord
  include StiPreload

  belongs_to :author, class_name: "User"
end

となっていて、belongs_toを呼び出す前にincludeが呼び出されてしまい、そのタイミングではbelongs_toのリレーションが設定されていないためエラーになっていたようでした。

正常に動くコードは以下の通りです。

class Tweet < ApplicationRecord
  belongs_to :author, class_name: "User"

  include StiPreload
end

回答していただいた https://ruby-jp.slack.com/ の方々どうもありがとうございました！