Cache de contador Rails para has_one :through no Rails 7 e 8

Estou com dificuldades para fazer um counter cache funcionar em um has_one :through e estou com um comportamento estranho. Quero ter um counter_cache no Companymodelo que armazene em cache o número de associados Location(through the BusinessUnitmodel).

Objetos:

class Company < ApplicationRecord
  has_many :business_units, class_name: :BusinessUnit, dependent: :destroy
  has_many :locations, through: :business_units, dependent: :destroy, counter_cache: :locations_count
end

class BusinessUnit < ApplicationRecord
  belongs_to :company, counter_cache: :locations_count
  has_many :locations, dependent: :destroy
end

class Location < ApplicationRecord
  belongs_to :business_unit
  has_one :company, through: :business_unit
end

Esquema:

ActiveRecord::Schema[7.1].define(version: 2025_04_03_203549) do
  enable_extension "plpgsql"

  create_table "business_units", force: :cascade do |t|
    t.string "name"
    t.bigint "company_id", null: false
    t.datetime "created_at", null: false
    t.datetime "updated_at", null: false
    t.index ["company_id"], name: "index_business_units_on_company_id"
  end

  create_table "companies", force: :cascade do |t|
    t.string "name"
    t.datetime "created_at", null: false
    t.datetime "updated_at", null: false
    t.integer "locations_count", default: 0, null: false
  end

  create_table "locations", force: :cascade do |t|
    t.string "name"
    t.bigint "business_unit_id", null: false
    t.datetime "created_at", null: false
    t.datetime "updated_at", null: false
    t.index ["business_unit_id"], name: "index_locations_on_business_unit_id"
  end

  add_foreign_key "business_units", "companies"
  add_foreign_key "locations", "business_units"
end

Meu entendimento com base nessas respostas do SO ( 1 , 2 ) é que isso deve funcionar. Não está bem documentado e é um pouco contraintuitivo como o Rails sabe automaticamente se referir à contagem de locais, mas, de fato, parece ter alguma mágica do Rails e chamar Company.reset_counters(1, :locations)corretamente define o contador para corresponder ao número de locais.

NO ENTANTO , o contador não atualiza automaticamente (para cima ou para baixo) quando locais são adicionados ou removidos, e se uma BusinessUnit com N locais for atualizada e atribuída a outra empresa, o contador só aumenta/diminui em 1 (em vez dos N locais esperados). Esse comportamento está presente no Rails 8.0.1 e 7.1.3 (que usei acidentalmente para construir meu MRE).

Há algo óbvio que estou esquecendo e que pode fazer o counter_cache funcionar corretamente?