class ArelExtensions::Visitors::ArelExtensions::Visitors::Arel::Visitors::PostgreSQL

Constants

DATE_FORMAT_DIRECTIVES
DATE_MAPPING
NUMBER_COMMA_MAPPING

Public Instance Methods

old_visit_Arel_Attributes_Attribute(o, collector)
old_visit_Arel_Nodes_As(o, collector)
Alias for: visit_Arel_Nodes_As
old_visit_Arel_Nodes_SelectStatement(o, collector)
old_visit_Arel_Nodes_TableAlias(o, collector)
visit_Aggregate_For_AggregateFunction(o, collector) click to toggle source
# File lib/arel_extensions/visitors/postgresql.rb, line 95
def visit_Aggregate_For_AggregateFunction o, collector
  if !o.order.blank? || !o.group.blank?
    collector << " OVER ("
    if !o.group.blank?
      collector << " PARTITION BY "
      o.group.each_with_index do |group, i|
        collector << COMMA if i != 0
        visit group, collector
      end
    end
    if !o.order.blank?
      collector << " ORDER BY "
      o.order.each_with_index do |order, i|
        collector << COMMA if i != 0
        visit order, collector
      end
    end
    collector << ")"
  end
  collector
end
visit_ArelExtensions_Nodes_AiIMatches(o, collector) click to toggle source
# File lib/arel_extensions/visitors/postgresql.rb, line 201
def visit_ArelExtensions_Nodes_AiIMatches o, collector
  collector = visit o.left.ai_collate, collector
  collector << ' ILIKE '
  collector = visit o.right.ai_collate, collector
  if o.escape
    collector << ' ESCAPE '
    visit o.escape, collector
  else
    collector
  end
end
visit_ArelExtensions_Nodes_AiMatches(o, collector) click to toggle source
# File lib/arel_extensions/visitors/postgresql.rb, line 189
def visit_ArelExtensions_Nodes_AiMatches o, collector
  collector = visit o.left.ai_collate, collector
  collector << ' LIKE '
  collector = visit o.right.ai_collate, collector
  if o.escape
    collector << ' ESCAPE '
    visit o.escape, collector
  else
    collector
  end
end
visit_ArelExtensions_Nodes_Cast(o, collector) click to toggle source
# File lib/arel_extensions/visitors/postgresql.rb, line 355
def visit_ArelExtensions_Nodes_Cast o, collector
  as_attr = case o.as_attr
  when :string
    Arel::Nodes::SqlLiteral.new('varchar')
  when :text, :ntext
    Arel::Nodes::SqlLiteral.new('text')
  when :time
    Arel::Nodes::SqlLiteral.new('time')
  when :int
    collector << "CAST(CAST("
    collector = visit o.left, collector
    collector << " AS numeric) AS int)"
    return collector
  when :number, :decimal, :float
    Arel::Nodes::SqlLiteral.new('numeric')
  when :datetime
    Arel::Nodes::SqlLiteral.new('timestamp')
  when :date
    Arel::Nodes::SqlLiteral.new('date')
  when :binary
    Arel::Nodes::SqlLiteral.new('binary')
  else
    Arel::Nodes::SqlLiteral.new(o.as_attr.to_s)
  end
  collector << "CAST("
  collector = visit o.left, collector
  collector << " AS "
  collector = visit as_attr, collector
  collector << ")"
  collector
end
visit_ArelExtensions_Nodes_Collate(o, collector) click to toggle source
# File lib/arel_extensions/visitors/postgresql.rb, line 225
def visit_ArelExtensions_Nodes_Collate o, collector
  if o.ai
    collector << "unaccent("
    collector = visit o.expressions.first, collector
    collector << ")"
  elsif o.ci
    collector = visit o.expressions.first, collector
  else
    collector = visit o.expressions.first, collector
  end
  collector
end
visit_ArelExtensions_Nodes_Concat(o, collector) click to toggle source
# File lib/arel_extensions/visitors/postgresql.rb, line 70
def visit_ArelExtensions_Nodes_Concat o, collector
  collector << '('
  o.expressions.each_with_index { |arg, i|
    collector = visit arg, collector
    collector << ' || ' unless i == o.expressions.length - 1
  }
  collector << ")"
  collector
end
visit_ArelExtensions_Nodes_DateAdd(o, collector) click to toggle source
# File lib/arel_extensions/visitors/postgresql.rb, line 238
def visit_ArelExtensions_Nodes_DateAdd o, collector
  collector = visit o.left, collector
  collector << ' + ' # (o.right.value >= 0 ? ' + ' : ' - ')
  collector = visit o.postgresql_value(o.right), collector
  collector
end
visit_ArelExtensions_Nodes_DateDiff(o, collector) click to toggle source
# File lib/arel_extensions/visitors/postgresql.rb, line 245
def visit_ArelExtensions_Nodes_DateDiff o, collector
  if o.right_node_type == :ruby_date || o.right_node_type == :ruby_time || o.right_node_type == :date || o.right_node_type == :datetime || o.right_node_type == :time
    collector << if o.left_node_type == :ruby_time || o.left_node_type == :datetime || o.left_node_type == :time
                    "DATEDIFF('second', "
                else
                    "DATEDIFF('day', "
                end
    collector = visit o.right, collector
    collector << (o.right_node_type == :date ? '::date' : '::timestamp')
    collector << COMMA
    collector = visit o.left, collector
    collector << (o.left_node_type == :date ? '::date' : '::timestamp')
    collector << ")"
  else
    collector << '('
    collector = visit o.left, collector
    collector << ' - ('
    if o.right.is_a?(ArelExtensions::Nodes::Duration)
      o.right.with_interval = true
    end
    collector = visit o.right, collector
    collector << '))'
  end
  collector
end
visit_ArelExtensions_Nodes_Duration(o, collector) click to toggle source
# File lib/arel_extensions/visitors/postgresql.rb, line 271
def visit_ArelExtensions_Nodes_Duration o, collector
  if o.with_interval
    interval = case o.left
      when  'd','m','y'
        'DAY'
      when 'h','mn','s'
        'SECOND'
      when /i\z/
        collector << "("
        collector = visit o.right, collector
        collector << ")"
        collector << " * (INTERVAL '1' #{DATE_MAPPING[o.left[0..-2]]})"
        return collector
      end
  end
  collector << "EXTRACT(#{DATE_MAPPING[o.left]} FROM "
  collector = visit o.right, collector
  collector << ")"
  collector << " * (INTERVAL '1' #{interval})" if interval && o.with_interval
  collector
end
visit_ArelExtensions_Nodes_Format(o, collector) click to toggle source
# File lib/arel_extensions/visitors/postgresql.rb, line 169
def visit_ArelExtensions_Nodes_Format o, collector
  fmt = ArelExtensions::Visitors::strftime_to_format(o.iso_format, DATE_FORMAT_DIRECTIVES)
  collector << "TO_CHAR("
  collector = visit o.left, collector
  collector << COMMA
  collector = visit Arel::Nodes.build_quoted(fmt), collector
  collector << ")"
  collector
end
visit_ArelExtensions_Nodes_FormattedNumber(o, collector) click to toggle source
# File lib/arel_extensions/visitors/postgresql.rb, line 387
def visit_ArelExtensions_Nodes_FormattedNumber o, collector
  col = o.left.coalesce(0)
  comma = o.precision == 0 ? '' : (NUMBER_COMMA_MAPPING[o.locale][0] || '.')
  thousand_separator = NUMBER_COMMA_MAPPING[o.locale][1] || (NUMBER_COMMA_MAPPING[o.locale] ? '' : 'G')
  nines_after = (1..o.precision).map{'9'}.join('')
  nines_before = ("999#{thousand_separator}"*4+"990")

  sign = ArelExtensions::Nodes::Case.new.when(col<0).
            then('-').
            else(o.flags.include?('+') ? '+' : (o.flags.include?(' ') ? ' ' : ''))
  sign_length = ArelExtensions::Nodes::Length.new([sign])

  number =
    if o.scientific_notation
      ArelExtensions::Nodes::Concat.new([
          Arel::Nodes::NamedFunction.new('TRIM',[
            Arel::Nodes::NamedFunction.new('TO_CHAR',[
              col.abs/Arel::Nodes.build_quoted(10).pow(col.abs.log10.floor),
              Arel::Nodes.build_quoted('FM'+nines_before+'"'+comma+'"V'+nines_after)
            ])]),
          o.type,
          Arel::Nodes::NamedFunction.new('TRIM',[
            Arel::Nodes::NamedFunction.new('TO_CHAR',[
              col.abs.log10.floor,
              Arel::Nodes.build_quoted('FM'+nines_before)
            ])])
        ])
    else
      Arel::Nodes::NamedFunction.new('TRIM',[
        Arel::Nodes::NamedFunction.new('TO_CHAR',[
          Arel::Nodes.build_quoted(col.abs),
          Arel::Nodes.build_quoted('FM'+nines_before+'"'+comma+'"V'+nines_after)
        ])])
    end

  repeated_char = (o.width == 0) ? Arel::Nodes.build_quoted('') : ArelExtensions::Nodes::Case.new().
    when(Arel::Nodes.build_quoted(o.width).abs-(number.length+sign_length)>0).
    then(Arel::Nodes.build_quoted(
        o.flags.include?('-') ? ' ' : (o.flags.include?('0') ? '0' : ' ')
      ).repeat(Arel::Nodes.build_quoted(o.width).abs-(number.length+sign_length))
    ).
    else('')
  before = (!o.flags.include?('0'))&&(!o.flags.include?('-')) ? repeated_char : ''
  middle = (o.flags.include?('0'))&&(!o.flags.include?('-'))  ? repeated_char : ''
  after  = o.flags.include?('-') ? repeated_char : ''
  full_number = ArelExtensions::Nodes::Concat.new([
      before,
      sign,
      middle,
      number,
      after
    ])
  collector = visit ArelExtensions::Nodes::Concat.new([Arel::Nodes.build_quoted(o.prefix),full_number,Arel::Nodes.build_quoted(o.suffix)]), collector
  collector
end
visit_ArelExtensions_Nodes_GroupConcat(o, collector) click to toggle source
# File lib/arel_extensions/visitors/postgresql.rb, line 117
def visit_ArelExtensions_Nodes_GroupConcat o, collector
  collector << "array_to_string(array_agg("
  collector = visit o.left, collector
  if o.order && !o.order.blank?
    collector << " ORDER BY"
    o.order.each_with_index do |order, i|
      collector << COMMA if i != 0
      collector << " "
      visit order, collector
    end
  end
  collector << ")"
  o.order = nil
  visit_Aggregate_For_AggregateFunction o, collector
  collector << COMMA
  collector =
    if o.separator && o.separator != 'NULL'
      visit o.separator, collector
    else
      visit Arel::Nodes.build_quoted(','), collector
    end
  collector << ")"
  collector
end
visit_ArelExtensions_Nodes_IsNotNull(o, collector) click to toggle source
# File lib/arel_extensions/visitors/postgresql.rb, line 334
def visit_ArelExtensions_Nodes_IsNotNull o, collector
    collector = visit o.expr, collector
    collector << ' IS NOT NULL'
    collector
end
visit_ArelExtensions_Nodes_IsNull(o, collector) click to toggle source
# File lib/arel_extensions/visitors/postgresql.rb, line 328
def visit_ArelExtensions_Nodes_IsNull o, collector
  collector = visit o.expr, collector
  collector << ' IS NULL'
  collector
end
visit_ArelExtensions_Nodes_Json(o,collector) click to toggle source
# File lib/arel_extensions/visitors/postgresql.rb, line 487
def visit_ArelExtensions_Nodes_Json o,collector
  case o.dict
  when Array
    collector << 'to_jsonb(array['
    o.dict.each.with_index do |v,i|
      if i != 0
        collector << Arel::Visitors::MySQL::COMMA
      end
      collector = visit v, collector
    end
    collector << '])'
  when Hash
    collector << 'jsonb_build_object('
    o.dict.each.with_index do |(k,v),i|
      if i != 0
        collector << Arel::Visitors::MySQL::COMMA
      end
      collector = visit k, collector
      collector << Arel::Visitors::MySQL::COMMA
      collector = visit v, collector
    end
    collector << ')'
  when String,Numeric,TrueClass,FalseClass
    collector = visit Arel::Nodes.build_quoted("#{o.dict}"), collector
    collector << '::jsonb'
  when NilClass
    collector  << %Q['null'::jsonb]
  when Arel::Attributes::Attribute
    collector = visit o.dict.cast(:jsonb), collector
  else
    collector = visit o.dict, collector
    collector << '::jsonb'
  end
  collector
end
visit_ArelExtensions_Nodes_JsonGet(o,collector) click to toggle source
# File lib/arel_extensions/visitors/postgresql.rb, line 533
def visit_ArelExtensions_Nodes_JsonGet o,collector
  collector = visit o.dict, collector
  collector << ' -> '
  collector = visit o.key, collector
  collector
end
visit_ArelExtensions_Nodes_JsonGroup(o, collector) click to toggle source
# File lib/arel_extensions/visitors/postgresql.rb, line 554
def visit_ArelExtensions_Nodes_JsonGroup o, collector
  if o.as_array
    collector << 'jsonb_agg('
    collector = visit o.dict, collector
    collector << ')'
  else
    case o.dict
    when Hash
      o.dict.each.with_index do |(k,v),i|
        if i != 0
          collector << ' || '
        end
        collector << 'jsonb_object_agg('
        collector = visit k, collector
        collector << Arel::Visitors::MySQL::COMMA
        collector = visit v, collector
        collector << ')'
      end
    else
      collector << 'jsonb_object_agg('
      collector = visit o.dict, collector
      collector << ')'
    end
  end
  collector
end
visit_ArelExtensions_Nodes_JsonMerge(o,collector) click to toggle source
# File lib/arel_extensions/visitors/postgresql.rb, line 523
def visit_ArelExtensions_Nodes_JsonMerge o,collector
  o.expressions.each.with_index do |v,i|
    if i != 0
      collector << ' || '
    end
    collector = visit v, collector
  end
  collector
end
visit_ArelExtensions_Nodes_JsonSet(o,collector) click to toggle source
# File lib/arel_extensions/visitors/postgresql.rb, line 540
def visit_ArelExtensions_Nodes_JsonSet o,collector
  collector << 'jsonb_set('
  collector = visit o.dict, collector
  collector << Arel::Visitors::MySQL::COMMA
  collector << 'array['
  collector = visit o.key, collector
  collector << ']'
  collector << Arel::Visitors::MySQL::COMMA
  collector = visit o.value, collector
  collector << Arel::Visitors::MySQL::COMMA
  collector << 'true)'
  collector
end
visit_ArelExtensions_Nodes_Locate(o, collector) click to toggle source
# File lib/arel_extensions/visitors/postgresql.rb, line 293
def visit_ArelExtensions_Nodes_Locate o, collector
  collector << "POSITION("
  collector = visit o.right, collector
  collector << " IN "
  collector = visit o.left, collector
  collector << ")"
  collector
end
visit_ArelExtensions_Nodes_Log10(o, collector) click to toggle source
# File lib/arel_extensions/visitors/postgresql.rb, line 43
def visit_ArelExtensions_Nodes_Log10 o, collector
  collector << "LOG("
  o.expressions.each_with_index { |arg, i|
    collector << Arel::Visitors::ToSql::COMMA if i != 0
    collector = visit arg, collector
  }
  collector << ")"
  collector
end
visit_ArelExtensions_Nodes_Ltrim(o, collector) click to toggle source
# File lib/arel_extensions/visitors/postgresql.rb, line 151
def visit_ArelExtensions_Nodes_Ltrim o, collector
    collector << 'TRIM(LEADING '
    collector = visit o.right, collector
    collector << " FROM "
    collector = visit o.left, collector
    collector << ")"
    collector
end
visit_ArelExtensions_Nodes_Power(o, collector) click to toggle source
# File lib/arel_extensions/visitors/postgresql.rb, line 33
def visit_ArelExtensions_Nodes_Power o, collector
  collector << "POWER("
  o.expressions.each_with_index { |arg, i|
    collector << Arel::Visitors::ToSql::COMMA if i != 0
    collector = visit arg, collector
  }
  collector << ")"
  collector
end
visit_ArelExtensions_Nodes_Rand(o, collector) click to toggle source
# File lib/arel_extensions/visitors/postgresql.rb, line 22
def visit_ArelExtensions_Nodes_Rand o, collector
  collector << "RANDOM("
  if (o.left != nil && o.right != nil)
    collector = visit o.left, collector
    collector << COMMA
    collector = isit o.right, collector
  end
  collector << ")"
  collector
end
visit_ArelExtensions_Nodes_RegexpReplace(o, collector) click to toggle source
# File lib/arel_extensions/visitors/postgresql.rb, line 312
def visit_ArelExtensions_Nodes_RegexpReplace o, collector
  collector << "REGEXP_REPLACE("
  visit o.left, collector
  collector << Arel::Visitors::ToSql::COMMA
  tab = o.pattern.inspect+ 'g' # Make it always global
  pattern = tab.split('/')[1..-2].join('/')
  flags = tab.split('/')[-1]
  visit Arel::Nodes.build_quoted(pattern), collector
  collector << Arel::Visitors::ToSql::COMMA
  visit o.substitute, collector
  collector << Arel::Visitors::ToSql::COMMA
  visit Arel::Nodes.build_quoted(flags+"g"), collector
  collector << ")"
  collector
end
visit_ArelExtensions_Nodes_Repeat(o, collector) click to toggle source
# File lib/arel_extensions/visitors/postgresql.rb, line 179
def visit_ArelExtensions_Nodes_Repeat o, collector
  collector << "REPEAT("
  o.expressions.each_with_index { |arg, i|
    collector << Arel::Visitors::ToSql::COMMA if i != 0
    collector = visit arg, collector
  }
  collector << ")"
  collector
end
visit_ArelExtensions_Nodes_Rtrim(o, collector) click to toggle source
# File lib/arel_extensions/visitors/postgresql.rb, line 160
def visit_ArelExtensions_Nodes_Rtrim o, collector
  collector << 'TRIM(TRAILING '
  collector = visit o.right, collector
  collector << " FROM "
  collector = visit o.left, collector
  collector << ")"
  collector
end
visit_ArelExtensions_Nodes_SMatches(o, collector) click to toggle source
# File lib/arel_extensions/visitors/postgresql.rb, line 213
def visit_ArelExtensions_Nodes_SMatches o, collector
  collector = visit o.left, collector
  collector << ' LIKE '
  collector = visit o.right, collector
  if o.escape
    collector << ' ESCAPE '
    visit o.escape, collector
  else
    collector
  end
end
visit_ArelExtensions_Nodes_Std(o, collector) click to toggle source
# File lib/arel_extensions/visitors/postgresql.rb, line 471
def visit_ArelExtensions_Nodes_Std o, collector
  collector << (o.unbiased_estimator ? "STDDEV_SAMP(" : "STDDEV_POP(")
  visit o.left, collector
  collector << ")"
  visit_Aggregate_For_AggregateFunction o, collector
  collector
end
visit_ArelExtensions_Nodes_Substring(o, collector) click to toggle source
# File lib/arel_extensions/visitors/postgresql.rb, line 302
def visit_ArelExtensions_Nodes_Substring o, collector
  collector << "SUBSTR("
  o.expressions.each_with_index { |arg, i|
    collector << COMMA if i != 0
    collector = visit arg, collector
  }
  collector << ")"
  collector
end
visit_ArelExtensions_Nodes_Sum(o, collector) click to toggle source
# File lib/arel_extensions/visitors/postgresql.rb, line 340
def visit_ArelExtensions_Nodes_Sum o, collector
  collector << "sum("
  collector = visit o.expr, collector
  collector << ")"
  visit_Aggregate_For_AggregateFunction o, collector
  collector
end
visit_ArelExtensions_Nodes_Trim(o, collector) click to toggle source
# File lib/arel_extensions/visitors/postgresql.rb, line 142
def visit_ArelExtensions_Nodes_Trim o, collector
    collector << 'TRIM(BOTH '
    collector = visit o.right, collector
    collector << " FROM "
    collector = visit o.left, collector
    collector << ")"
    collector
end
visit_ArelExtensions_Nodes_Variance(o, collector) click to toggle source
# File lib/arel_extensions/visitors/postgresql.rb, line 479
def visit_ArelExtensions_Nodes_Variance o, collector
  collector << (o.unbiased_estimator ? "VAR_SAMP(" : "VAR_POP(")
  visit o.left, collector
  collector << ")"
  visit_Aggregate_For_AggregateFunction o, collector
  collector
end
visit_ArelExtensions_Nodes_Wday(o, collector) click to toggle source
# File lib/arel_extensions/visitors/postgresql.rb, line 348
def visit_ArelExtensions_Nodes_Wday o, collector
  collector << "EXRTACT(DOW, "
  collector = visit o.date, collector
  collector << ')'
  collector
end
visit_Arel_Attributes_Attribute(o, collector) click to toggle source
# File lib/arel_extensions/visitors/postgresql.rb, line 463
def visit_Arel_Attributes_Attribute o, collector
  join_name = o.relation.table_alias || o.relation.name
  if join_name.length > 63
    join_name = Arel.shorten(join_name)
  end
  collector << "#{quote_table_name join_name}.#{quote_column_name o.name}"
end
visit_Arel_Nodes_As(o, collector) click to toggle source
# File lib/arel_extensions/visitors/postgresql.rb, line 81
def visit_Arel_Nodes_As o, collector
  if o.left.is_a?(Arel::Nodes::Binary)
    collector << '('
    collector = visit o.left, collector
    collector << ')'
  else
    collector = visit o.left, collector
  end
  collector << " AS \""
  collector = visit o.right, collector
  collector << "\""
  collector
end
Also aliased as: old_visit_Arel_Nodes_As
visit_Arel_Nodes_NotRegexp(o, collector) click to toggle source
# File lib/arel_extensions/visitors/postgresql.rb, line 63
def visit_Arel_Nodes_NotRegexp o, collector
  collector = visit o.left, collector
  collector << " !~ "
  collector = visit o.right, collector
  collector
end
visit_Arel_Nodes_Regexp(o, collector) click to toggle source
# File lib/arel_extensions/visitors/postgresql.rb, line 55
def visit_Arel_Nodes_Regexp o, collector
  collector = visit o.left, collector
  collector << " ~ "
  collector = visit o.right, collector
  collector
end
visit_Arel_Nodes_SelectStatement(o, collector) click to toggle source
# File lib/arel_extensions/visitors/postgresql.rb, line 445
def visit_Arel_Nodes_SelectStatement o, collector

  if !(collector.value.blank? || (collector.value.is_a?(Array) && collector.value[0].blank?)) && o.limit.blank? && o.offset.blank?
    o = o.dup
    o.orders = []
  end
  old_visit_Arel_Nodes_SelectStatement(o,collector)
end
visit_Arel_Nodes_TableAlias(o, collector) click to toggle source
# File lib/arel_extensions/visitors/postgresql.rb, line 455
def visit_Arel_Nodes_TableAlias o, collector
  if o.name.length > 63
    o = Arel::Table.new(o.table_name).alias(Arel.shorten(o.name))
  end
  old_visit_Arel_Nodes_TableAlias(o,collector)
end