Skip to content

Commit aabb8f6

Browse files
committed
Add support for count
1 parent 6e983e9 commit aabb8f6

File tree

2 files changed

+13
-0
lines changed

2 files changed

+13
-0
lines changed

lib/activerecord_adbc_adapter/result.rb

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -92,6 +92,15 @@ def indexed_rows
9292
@indexed_rows ||= to_a
9393
end
9494

95+
def cast_values(type_overrides = {})
96+
# TODO: type_overrides support
97+
if fields.size == 1
98+
rows.map(&:first)
99+
else
100+
rows
101+
end
102+
end
103+
95104
def to_arrow
96105
@table
97106
end

test/test_model.rb

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,10 @@ def test_all
2323
User.select(:id).all)
2424
end
2525

26+
def test_count
27+
assert_equal(3, User.count)
28+
end
29+
2630
sub_test_case(".ingest") do
2731
def id_array
2832
Arrow::Int64Array.new([4, 5, 6])

0 commit comments

Comments
 (0)