Class: Arel::Visitors::SQLServer

Inherits:
ToSql
  • Object
show all
Includes:
ArJdbc::MsSQL::LimitHelpers::SqlServerReplaceLimitOffset
Defined in:
lib/arel/visitors/mssql.rb

Direct Known Subclasses

SQLServer2000

Instance Method Summary collapse

Methods included from ArJdbc::MsSQL::LimitHelpers::SqlServerReplaceLimitOffset

replace_limit_offset!

Instance Method Details

#select_count?(o) ⇒ Boolean

Returns:

  • (Boolean)


6
7
8
9
10
# File 'lib/arel/visitors/mssql.rb', line 6

def select_count? o
  sel = o.cores.length == 1 && o.cores.first
  projections = sel.projections.length == 1 && sel.projections
  Arel::Nodes::Count === projections.first
end

#visit_Arel_Nodes_SelectStatement(o) ⇒ Object

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



14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
# File 'lib/arel/visitors/mssql.rb', line 14

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, o.limit.to_i, o.offset && o.offset.value.to_i, order)
    sql = "SELECT COUNT(*) AS count_id FROM (#{sql}) AS subquery" if subquery
  elsif order
    sql << " #{order}"
  else
    sql = super
  end
  sql
end