Sha256: b1a4059302226fcd9643ca033b7c297499bf9a260c98ed8c0986c80b5d6af6b6
Contents?: true
Size: 1.5 KB
Versions: 11
Compression:
Stored size: 1.5 KB
Contents
require 'arel/visitors/compat'
module Arel
module Visitors
class SQLServer < Arel::Visitors::ToSql
include ArJdbc::MsSQL::LimitHelpers::SqlServerReplaceLimitOffset
def select_count? o
sel = o.cores.length == 1 && o.cores.first
projections = sel && sel.projections.length == 1 && sel.projections
projections && Arel::Nodes::Count === projections.first
end
# Need to mimic the subquery logic in ARel 1.x for select count with limit
# See arel/engines/sql/compilers/mssql_compiler.rb for details
def visit_Arel_Nodes_SelectStatement o
order = "ORDER BY #{o.orders.map { |x| visit x }.join(', ')}" unless o.orders.empty?
if o.limit
if select_count?(o)
subquery = true
sql = o.cores.map do |x|
x = x.dup
x.projections = [Arel::Nodes::SqlLiteral.new("*")]
visit_Arel_Nodes_SelectCore x
end.join
else
sql = o.cores.map { |x| visit_Arel_Nodes_SelectCore x }.join
end
order ||= "ORDER BY #{@connection.determine_order_clause(sql)}"
replace_limit_offset!(sql, limit_for(o.limit).to_i, o.offset && o.offset.value.to_i, order)
sql = "SELECT COUNT(*) AS count_id FROM (#{sql}) AS subquery" if subquery
else
sql = super
end
sql
end
end
class SQLServer2000 < SQLServer
include ArJdbc::MsSQL::LimitHelpers::SqlServer2000ReplaceLimitOffset
end
end
end
Version data entries
11 entries across 11 versions & 4 rubygems