dymk/has_aux_table
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

id column rewriting

Browse Source
This commit is contained in:
Dylan Knutson
2025-07-29 16:18:37 +00:00
parent 94c2fb9593
commit 3016a17480
6 changed files with 207 additions and 101 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
spec/has_aux_table_spec.rb
View File

@@ -235,7 +235,7 @@ RSpec.describe HasAuxTable do
it "allows saving the model with auxiliary columns" do
car = Car.create!(name: "Honda Civic")
queries =
SpecHelper.capture_queries do
capture_queries do
car.fuel_type = "hybrid"
car.engine_size = 1.8
car.save!
@@ -424,7 +424,7 @@ RSpec.describe HasAuxTable do
it "loads single model with auxiliary data in one query using find" do
queries =
SpecHelper.capture_queries do
capture_queries do
car = Car.find(@car1.id)
# Access auxiliary attributes to ensure they're loaded
car.fuel_type
@@ -436,7 +436,7 @@ RSpec.describe HasAuxTable do
it "loads single model with auxiliary data in one query using find_by" do
queries =
SpecHelper.capture_queries do
capture_queries do
car = Car.find_by(name: "Toyota Prius")
# Access auxiliary attributes to ensure they're loaded
car.fuel_type
@@ -448,7 +448,7 @@ RSpec.describe HasAuxTable do
it "loads multiple models with auxiliary data in one query using where" do
queries =
SpecHelper.capture_queries do
capture_queries do
cars = Car.where(fuel_type: %w[hybrid electric])
# Access auxiliary attributes for all cars
cars.each do |car|
@@ -473,7 +473,7 @@ RSpec.describe HasAuxTable do
cars = nil
queries =
SpecHelper.capture_queries do
capture_queries do
cars = Car.where(fuel_type: "gasoline")
# Access auxiliary attributes for all cars - should not trigger additional queries
cars.each do |car|
@@ -489,7 +489,7 @@ RSpec.describe HasAuxTable do
it "uses single query when ordering by auxiliary columns" do
queries =
SpecHelper.capture_queries do
capture_queries do
cars = Car.where(engine_size: 1.0..3.0).order(:engine_size)
# Access all attributes
cars.each do |car|
@@ -504,7 +504,7 @@ RSpec.describe HasAuxTable do
it "uses single query for complex auxiliary column queries" do
queries =
SpecHelper.capture_queries do
capture_queries do
cars =
Car.where(fuel_type: "hybrid").or(Car.where(engine_size: 0.0))
# Access all attributes
@@ -520,7 +520,7 @@ RSpec.describe HasAuxTable do
it "uses single query when finding by auxiliary columns" do
queries =
SpecHelper.capture_queries do
capture_queries do
car = Car.find_by(fuel_type: "hybrid", name: "Toyota Prius")
# Access all attributes
car.name
@@ -537,7 +537,7 @@ RSpec.describe HasAuxTable do
# Now count queries when accessing auxiliary attributes
queries =
SpecHelper.capture_queries do
capture_queries do
car.fuel_type
car.engine_size
car.fuel_type? # presence check
@@ -550,7 +550,7 @@ RSpec.describe HasAuxTable do
it "handles mixed queries with main and auxiliary columns in single query" do
queries =
SpecHelper.capture_queries do
capture_queries do
cars = Car.where(name: "Toyota Prius", fuel_type: "hybrid")
cars.each do |car|
car.name
@@ -564,7 +564,7 @@ RSpec.describe HasAuxTable do
it "uses single query for range queries on auxiliary columns" do
queries =
SpecHelper.capture_queries do
capture_queries do
cars = Car.where(engine_size: 0.0..1.9)
cars.each do |car|
car.name
@@ -578,7 +578,7 @@ RSpec.describe HasAuxTable do
it "maintains single query performance with limit and offset" do
queries =
SpecHelper.capture_queries do
capture_queries do
car = Car.where(fuel_type: %w[hybrid electric]).limit(1).first
# Access auxiliary attributes
car.fuel_type
@@ -983,7 +983,7 @@ RSpec.describe HasAuxTable do
end
it "reloads with one query" do
queries = SpecHelper.capture_queries { @car.reload }
queries = capture_queries { @car.reload }
expect(queries.length).to eq(1)
end
@@ -1027,16 +1027,12 @@ RSpec.describe HasAuxTable do
expect(Car.count).to eq(1)
expect(Boat.count).to eq(1)
expect(
SpecHelper.capture_queries { car = Vehicle.find(car.id) }.length
).to eq(1)
expect(capture_queries { car = Vehicle.find(car.id) }.length).to eq(1)
expect(car.fuel_type).to eq("gasoline")
expect(car.engine_size).to eq(2.0)
expect(car.name).to eq("Honda Civic")
expect(
SpecHelper.capture_queries { boat = Vehicle.find(boat.id) }.length
).to eq(1)
expect(capture_queries { boat = Vehicle.find(boat.id) }.length).to eq(1)
expect(boat.only_freshwater).to eq(true)
end

165
spec/loading_optimizations_spec.rb
View File

@@ -17,84 +17,105 @@ RSpec.describe "loading optimizations" do
shared_examples "queries only the aux table" do
it "queries only the aux table" do
expect(@queries.length).to eq(1)
expect(@queries.first).not_to include("JOIN")
expect(@queries.first).to match(/\bvehicles_car_aux\b/)
expect(@queries.first).not_to match(/\bvehicles\b/)
expect(queries.length).to eq(1)
expect(queries.first).not_to include("JOIN")
expect(queries.first).to match(/\bvehicles_car_aux\b/)
expect(queries.first).not_to match(/\bvehicles\b/)
end
end
shared_examples "queries both tables" do
it "queries both tables" do
expect(@queries.length).to eq(1)
expect(@queries.first).to include("JOIN")
expect(@queries.first).to match(/\bvehicles\b/)
expect(@queries.first).to match(/\bvehicles_car_aux\b/)
expect(queries.length).to eq(1)
expect(queries.first).to include("JOIN")
expect(queries.first).to match(/\bvehicles\b/)
expect(queries.first).to match(/\bvehicles_car_aux\b/)
end
end
describe "pluck" do
context "aux columns are referenced" do
before do
@queries =
SpecHelper.capture_queries do
expect(Car.pluck(:fuel_type)).to eq(%w[gasoline hybrid electric])
end
let_and_capture(:queries) { Car.pluck(:fuel_type) }
it "returns the correct result" do
expect(queries.result).to eq(%w[gasoline hybrid electric])
end
it_behaves_like "queries only the aux table"
end
context "aux columns are chained on a where clause" do
before do
@queries =
SpecHelper.capture_queries do
expect(Car.where(engine_size: 1.4..1.9).pluck(:fuel_type)).to eq(
%w[hybrid electric]
)
end
let_and_capture(:queries) do
Car.where(engine_size: 1.4..1.9).pluck(:fuel_type)
end
it "returns the correct result" do
expect(queries.result).to eq(%w[hybrid electric])
end
it_behaves_like "queries only the aux table"
it "applies the BETWEEN clause" do
expect(@queries.first).to include("BETWEEN")
expect(queries.first).to include("BETWEEN")
end
end
context "main table columns are referenced" do
before do
@queries =
SpecHelper.capture_queries do
expect(Car.where(name: "Toyota Camry").pluck(:fuel_type)).to eq(
%w[gasoline]
)
end
let(:queries) do
capture_queries { Car.where(name: "Toyota Camry").pluck(:fuel_type) }
end
it "returns the correct result" do
expect(queries.result).to eq(%w[gasoline])
end
it_behaves_like "queries both tables"
end
context "main table columns are chained on a where clause" do
before do
@queries =
SpecHelper.capture_queries do
expect(Car.where(name: "Toyota Camry").pluck(:fuel_type)).to eq(
%w[gasoline]
)
end
let(:queries) do
capture_queries { Car.where(name: "Toyota Camry").pluck(:fuel_type) }
end
it "returns the correct result" do
expect(queries.result).to eq(%w[gasoline])
end
it_behaves_like "queries both tables"
end
context "multiple columns" do
before do
@queries =
SpecHelper.capture_queries do
expect(Car.pluck(:fuel_type, :engine_size)).to eq(
[["gasoline", 2.0], ["hybrid", 1.5], ["electric", 1.8]]
)
end
let(:queries) do
capture_queries { Car.pluck(:fuel_type, :engine_size) }
end
it "returns the correct result" do
expect(queries.result).to eq(
[["gasoline", 2.0], ["hybrid", 1.5], ["electric", 1.8]]
)
end
it_behaves_like "queries only the aux table"
end
context "multiple columns with a where clause" do
let(:queries) do
capture_queries do
Car.where(name: "Toyota Camry").pluck(:fuel_type, :engine_size)
end
end
it "returns the correct result" do
expect(queries.result).to eq([["gasoline", 2.0]])
end
it_behaves_like "queries both tables"
end
context "querying the id column" do
let_and_capture(:queries) { Car.pluck(:id) }
it "renames the base_table_id to id" do
expect(queries.first).to include("\"base_table_id\" AS \"id\"")
end
it_behaves_like "queries only the aux table"
@@ -103,50 +124,56 @@ RSpec.describe "loading optimizations" do
describe "maximum" do
context "aux columns are referenced" do
before do
@queries =
SpecHelper.capture_queries do
expect(Car.maximum(:engine_size)).to eq(2.0)
end
end
let_and_capture(:queries) { Car.maximum(:engine_size) }
it_behaves_like "queries only the aux table"
end
context "aux columns are chained on a where clause" do
before do
@queries =
SpecHelper.capture_queries do
expect(
Car.where(engine_size: 1.4..1.9).maximum(:engine_size)
).to eq(1.8)
end
let_and_capture(:queries) do
Car.where(engine_size: 1.4..1.9).maximum(:engine_size)
end
it "returns the correct result" do
expect(queries.result).to eq(1.8)
end
it_behaves_like "queries only the aux table"
end
context "main table columns are referenced" do
before do
@queries =
SpecHelper.capture_queries do
expect(
Car.where(name: "Toyota Camry").maximum(:engine_size)
).to eq(2.0)
end
let_and_capture(:queries) do
Car.where(name: "Toyota Camry").maximum(:engine_size)
end
it "returns the correct result" do
expect(queries.result).to eq(2.0)
end
it_behaves_like "queries both tables"
end
context "id column is referenced" do
let_and_capture(:queries) { Car.maximum(:id) }
it "renames the base_table_id to id" do
expect(queries.first).to include('"base_table_id"')
end
it "has the right result" do
expect(queries.result).to eq(3)
end
it_behaves_like "queries only the aux table"
end
context "main table columns are chained on a where clause" do
before do
@queries =
SpecHelper.capture_queries do
expect(
Car.where(name: "Toyota Camry").maximum(:engine_size)
).to eq(2.0)
end
let_and_capture(:queries) do
Car.where(name: "Toyota Camry").maximum(:engine_size)
end
it "returns the correct result" do
expect(queries.result).to eq(2.0)
end
it_behaves_like "queries both tables"

58
spec/spec_helper.rb
View File

@@ -54,6 +54,36 @@ RSpec.configure do |config|
raise ActiveRecord::Rollback
end
end
config.include(
Module.new do
extend T::Sig
sig do
type_parameters(:T)
.params(block: T.proc.returns(T.type_parameter(:T)))
.returns(SpecHelper::CaptureQueries[T.type_parameter(:T)])
end
def capture_queries(&block)
SpecHelper.capture_queries(&block)
end
end
)
config.extend(
Module.new do
extend T::Sig
sig do
type_parameters(:T)
.params(binding: Symbol, block: T.proc.returns(T.type_parameter(:T)))
.returns(T.type_parameter(:T))
end
def let_and_capture(binding, &block)
T.bind(self, RSpec::Core::MemoizedHelpers::ClassMethods)
let(binding) { SpecHelper.capture_queries(&block) }
end
end
)
end
module SpecHelper
@@ -61,8 +91,28 @@ module SpecHelper
extend T::Helpers
LOG_QUERIES = T.let(false, T::Boolean)
class CaptureQueries < Array
extend T::Sig
extend T::Generic
Elem = type_member { { fixed: String } }
Result = type_member
sig { params(queries: T::Array[String], result: Result).void }
def initialize(queries, result)
super(queries)
@result = result
end
sig { returns(Result) }
attr_reader :result
end
# Helper method to count queries
sig { params(block: T.proc.void).returns(T::Array[String]) }
sig do
type_parameters(:T)
.params(block: T.proc.returns(T.type_parameter(:T)))
.returns(CaptureQueries[T.type_parameter(:T)])
end
def self.capture_queries(&block)
queries = T.let([], T::Array[String])
query_callback =
@@ -70,6 +120,7 @@ module SpecHelper
queries << values[:sql]
end
result = T.let(nil, T.untyped)
ActiveSupport::Notifications.subscribed(
query_callback,
"sql.active_record"
@@ -79,10 +130,11 @@ module SpecHelper
ActiveRecord::Base.logger = Logger.new(STDOUT)
ActiveRecord::Base.logger.level = Logger::DEBUG
end
block.call
result = block.call
ensure
ActiveRecord::Base.logger = old_logger if LOG_QUERIES
end
queries
CaptureQueries.new(queries, result)
end
end