Ruby on Rails のポリモーフィック関連時に発行されるクエリの効率について

Question

前提・実現したいこと

ポリモーフィック関連付け を用いてアプリケーションを作成しております。
この時、発行されるクエリが非効率なものではないか、と思うのですが、より適切な方法があるかアドバイスを頂けないでしょうか。

サンプルアプリケーション

会社(Companyクラス)は社員（Employeeクラス）を複数もち、
社員は正社員(Permanentクラス), 契約社員(Temporaryクラス)の種類があり、給与計算(salaryメソッド)のロジックが異なります。
異なるロジックを隠蔽して扱うためにポリモーフィック関連付けを利用できないかと考えました。

# 会社
class Company < ApplicationRecord
  has_many :employees
end

# 社員
class Employee < ApplicationRecord
  belongs_to :company
  belongs_to :employable, polymorphic: true

  # 基本給(社員種別を問わず一緒)
  def base_salary
    100
  end

  # 給与計算
  def salary
    employable.salary
  end
end

# 正社員
class Permanent < ApplicationRecord
  has_one :employee, as: :employable, dependent: :destroy

  # 給与計算
  def salary
    # 基本給 + 100円とする
    employee.base_salary + 100
  end
end

# 契約社員
class Temporary < ApplicationRecord
  has_one :employee, as: :employable, dependent: :destroy

  # 給与計算
  def salary
    # 基本給 + 80円とする
    employee.base_salary + 80
  end
end

以上のソースコードでEmployee.salary で社員の種別を意識せず給与計算を行うことができるようになりました。

しかし、発行されるSQLを確認したところ、

irb(main):001:0> Employee.first.salary
   (2.0ms)  SELECT sqlite_version(*)
  Employee Load (0.5ms)  SELECT "employees".* FROM "employees" ORDER BY "employees"."id" ASC LIMIT ?  [["LIMIT", 1]]
  Temporary Load (0.6ms)  SELECT "temporaries".* FROM "temporaries" WHERE "temporaries"."id" = ? LIMIT ?  [["id", 1], ["LIMIT", 1]]
  Employee Load (0.6ms)  SELECT "employees".* FROM "employees" WHERE "employees"."employable_id" = ? AND "employees"."employable_type" = ? LIMIT ?  [["employable_id", 1], ["employable_type", "Temporary"], ["LIMIT", 1]]
=> 180

Employeeが二度取得されてしまいました。

全社員を給与計算を行った場合も同様です。

Company.first.employees.map {|x| x.salary}
  Company Load (0.3ms)  SELECT "companies".* FROM "companies" ORDER BY "companies"."id" ASC LIMIT ?  [["LIMIT", 1]]
  Employee Load (0.3ms)  SELECT "employees".* FROM "employees" WHERE "employees"."company_id" = ?  [["company_id", 1]]
  Temporary Load (0.2ms)  SELECT "temporaries".* FROM "temporaries" WHERE "temporaries"."id" = ? LIMIT ?  [["id", 1], ["LIMIT", 1]]
  Employee Load (0.2ms)  SELECT "employees".* FROM "employees" WHERE "employees"."employable_id" = ? AND "employees"."employable_type" = ? LIMIT ?  [["employable_id", 1], ["employable_type", "Temporary"], ["LIMIT", 1]]
  Temporary Load (0.1ms)  SELECT "temporaries".* FROM "temporaries" WHERE "temporaries"."id" = ? LIMIT ?  [["id", 2], ["LIMIT", 1]]
  Employee Load (0.9ms)  SELECT "employees".* FROM "employees" WHERE "employees"."employable_id" = ? AND "employees"."employable_type" = ? LIMIT ?  [["employable_id", 2], ["employable_type", "Temporary"], ["LIMIT", 1]]
  Permanent Load (0.1ms)  SELECT "permanents".* FROM "permanents" WHERE "permanents"."id" = ? LIMIT ?  [["id", 1], ["LIMIT", 1]]
  Employee Load (0.1ms)  SELECT "employees".* FROM "employees" WHERE "employees"."employable_id" = ? AND "employees"."employable_type" = ? LIMIT ?  [["employable_id", 1], ["employable_type", "Permanent"], ["LIMIT", 1]]
  Permanent Load (0.1ms)  SELECT "permanents".* FROM "permanents" WHERE "permanents"."id" = ? LIMIT ?  [["id", 2], ["LIMIT", 1]]
  Employee Load (0.1ms)  SELECT "employees".* FROM "employees" WHERE "employees"."employable_id" = ? AND "employees"."employable_type" = ? LIMIT ?  [["employable_id", 2], ["employable_type", "Permanent"], ["LIMIT", 1]]
=> [180, 180, 200, 200]

給与計算のロジック内で、Employee.base_salaryを利用するためなのですが、取得済みのEmployeeを利用させることはできないのでしょうか。

試したこと

ポリモーフィック関連付けを利用しない場合では、何度もクエリが発行されている様ではありませんでした。

# 契約社員管理者
class TemporaryEmployeeManager < ApplicationRecord
  has_many :temporaries

  # 何らかのロジックfoo
  def foo
    'foo'
  end
end

# 契約社員
class Temporary < ApplicationRecord
  belongs_to :temporary_employee_manager

  # 何らかのロジックfoo
  def foo
    # 管理者のfooに依存している
    temporary_employee_manager.foo + 'bar'
  end
end

irb(main):002:0> TemporaryEmployeeManager.first.temporaries.map { |x| x.foo }
  TemporaryEmployeeManager Load (0.6ms)  SELECT "temporary_employee_managers".* FROM "temporary_employee_managers" ORDER BY "temporary_employee_managers"."id" ASC LIMIT ?  [["LIMIT", 1]]
  Temporary Load (0.8ms)  SELECT "temporaries".* FROM "temporaries" WHERE "temporaries"."temporary_employee_manager_id" = ?  [["temporary_employee_manager_id", 1]]
=> ["foobar", "foobar"]

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

ruby '2.6.5'

# Bundle edge Rails instead: gem 'rails', github: 'rails/rails'
gem 'rails', '~> 6.0.3', '>= 6.0.3.2'

Comments

[winterboum]

シンプルな関連ではないので、 inverse_of を明示的に設定が必要なのではないでしょうか。
ただ、、、
belongs_to :employable, polymorphic: true　に対しなにをinverse_ofしたらよいものか、、

Answer 1

winterboumさん。コメントを下さりありがとうございます！

inverse_ofを指定し、生成されるクエリが変わることを確認できました。

class Employee < ApplicationRecord
  belongs_to :company
  belongs_to :employable, polymorphic: true, inverse_of: :employee

  def base_salary
    100
  end

  def salary
    employable.salary
  end
end

irb(main):001:0> Employee.first.salary
   (3.3ms)  SELECT sqlite_version(*)
  Employee Load (0.1ms)  SELECT "employees".* FROM "employees" ORDER BY "employees"."id" ASC LIMIT ?  [["LIMIT", 1]]
  Temporary Load (0.3ms)  SELECT "temporaries".* FROM "temporaries" WHERE "temporaries"."id" = ? LIMIT ?  [["id", 1], ["LIMIT", 1]]
=> 180
irb(main):002:0> Company.first.employees.map {|x| x.salary}
  Company Load (0.1ms)  SELECT "companies".* FROM "companies" ORDER BY "companies"."id" ASC LIMIT ?  [["LIMIT", 1]]
  Employee Load (0.2ms)  SELECT "employees".* FROM "employees" WHERE "employees"."company_id" = ?  [["company_id", 1]]
  Temporary Load (0.1ms)  SELECT "temporaries".* FROM "temporaries" WHERE "temporaries"."id" = ? LIMIT ?  [["id", 1], ["LIMIT", 1]]
  Temporary Load (0.1ms)  SELECT "temporaries".* FROM "temporaries" WHERE "temporaries"."id" = ? LIMIT ?  [["id", 2], ["LIMIT", 1]]
  Permanent Load (0.1ms)  SELECT "permanents".* FROM "permanents" WHERE "permanents"."id" = ? LIMIT ?  [["id", 1], ["LIMIT", 1]]
  Permanent Load (0.1ms)  SELECT "permanents".* FROM "permanents" WHERE "permanents"."id" = ? LIMIT ?  [["id", 2], ["LIMIT", 1]]
=> [180, 180, 200, 200]