Module: ActiveRecord::ConnectionAdapters::PostgreSQL::Quoting

Defined in:
activerecord/lib/active_record/connection_adapters/postgresql/quoting.rb

Defined Under Namespace

Classes: IntegerOutOf64BitRange

Instance Method Summary collapse

Instance Method Details

#check_int_in_range(value) ⇒ Object 



25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
 
# File 'activerecord/lib/active_record/connection_adapters/postgresql/quoting.rb', line 25

def check_int_in_range(value)
  if value.to_int > 9223372036854775807 || value.to_int < -9223372036854775808
    exception = <<~ERROR
      Provided value outside of the range of a signed 64bit integer.

      PostgreSQL will treat the column type in question as a numeric.
      This may result in a slow sequential scan due to a comparison
      being performed between an integer or bigint value and a numeric value.

      To allow for this potentially unwanted behavior, set
      ActiveRecord.raise_int_wider_than_64bit to false.
    ERROR
    raise IntegerOutOf64BitRange.new exception
  end
end

#column_name_matcher ⇒ Object 



152
153
154
 
# File 'activerecord/lib/active_record/connection_adapters/postgresql/quoting.rb', line 152

def column_name_matcher
  COLUMN_NAME
end

#column_name_with_order_matcher ⇒ Object 



156
157
158
 
# File 'activerecord/lib/active_record/connection_adapters/postgresql/quoting.rb', line 156

def column_name_with_order_matcher
  COLUMN_NAME_WITH_ORDER
end

#escape_bytea(value) ⇒ Object

Escapes binary strings for bytea input to the database.



14
15
16
 
# File 'activerecord/lib/active_record/connection_adapters/postgresql/quoting.rb', line 14

def escape_bytea(value)
  valid_raw_connection.escape_bytea(value) if value
end

#lookup_cast_type_from_column(column) ⇒ Object

:nodoc:



148
149
150
 
# File 'activerecord/lib/active_record/connection_adapters/postgresql/quoting.rb', line 148

def lookup_cast_type_from_column(column) # :nodoc:
  type_map.lookup(column.oid, column.fmod, column.sql_type)
end

#quote(value) ⇒ Object

:nodoc:



41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
 
# File 'activerecord/lib/active_record/connection_adapters/postgresql/quoting.rb', line 41

def quote(value) # :nodoc:
  if ActiveRecord.raise_int_wider_than_64bit && value.is_a?(Integer)
    check_int_in_range(value)
  end

  case value
  when OID::Xml::Data
    "xml '#{quote_string(value.to_s)}'"
  when OID::Bit::Data
    if value.binary?
      "B'#{value}'"
    elsif value.hex?
      "X'#{value}'"
    end
  when Numeric
    if value.finite?
      super
    else
      "'#{value}'"
    end
  when OID::Array::Data
    quote(encode_array(value))
  when Range
    quote(encode_range(value))
  else
    super
  end
end

#quote_column_name(name) ⇒ Object

Quotes column names for use in SQL queries.



99
100
101
 
# File 'activerecord/lib/active_record/connection_adapters/postgresql/quoting.rb', line 99

def quote_column_name(name) # :nodoc:
  self.class.quoted_column_names[name] ||= PG::Connection.quote_ident(super).freeze
end

#quote_default_expression(value, column) ⇒ Object

:nodoc:



117
118
119
120
121
122
123
124
125
126
127
128
 
# File 'activerecord/lib/active_record/connection_adapters/postgresql/quoting.rb', line 117

def quote_default_expression(value, column) # :nodoc:
  if value.is_a?(Proc)
    value.call
  elsif column.type == :uuid && value.is_a?(String) && value.include?("()")
    value # Does not quote function default values for UUID columns
  elsif column.respond_to?(:array?)
    type = lookup_cast_type_from_column(column)
    quote(type.serialize(value))
  else
    super
  end
end

#quote_schema_name(name) ⇒ Object

Quotes schema names for use in SQL queries.



90
91
92
 
# File 'activerecord/lib/active_record/connection_adapters/postgresql/quoting.rb', line 90

def quote_schema_name(name)
  PG::Connection.quote_ident(name)
end

#quote_string(s) ⇒ Object

Quotes strings for use in SQL input.



71
72
73
74
75
 
# File 'activerecord/lib/active_record/connection_adapters/postgresql/quoting.rb', line 71

def quote_string(s) # :nodoc:
  with_raw_connection(allow_retry: true, uses_transaction: false) do |connection|
    connection.escape(s)
  end
end

#quote_table_name(name) ⇒ Object

Checks the following cases:

  • table_name

  • “table.name”

  • schema_name.table_name

  • schema_name.“table.name”

  • “schema.name”.table_name

  • “schema.name”.“table.name”



85
86
87
 
# File 'activerecord/lib/active_record/connection_adapters/postgresql/quoting.rb', line 85

def quote_table_name(name) # :nodoc:
  self.class.quoted_table_names[name] ||= Utils.extract_schema_qualified_name(name.to_s).quoted.freeze
end

#quote_table_name_for_assignment(table, attr) ⇒ Object 



94
95
96
 
# File 'activerecord/lib/active_record/connection_adapters/postgresql/quoting.rb', line 94

def quote_table_name_for_assignment(table, attr)
  quote_column_name(attr)
end

#quoted_binary(value) ⇒ Object

:nodoc:



113
114
115
 
# File 'activerecord/lib/active_record/connection_adapters/postgresql/quoting.rb', line 113

def quoted_binary(value) # :nodoc:
  "'#{escape_bytea(value.to_s)}'"
end

#quoted_date(value) ⇒ Object

Quote date/time values for use in SQL input.



104
105
106
107
108
109
110
111
 
# File 'activerecord/lib/active_record/connection_adapters/postgresql/quoting.rb', line 104

def quoted_date(value) # :nodoc:
  if value.year <= 0
    bce_year = format("%04d", -value.year + 1)
    super.sub(/^-?\d+/, bce_year) + " BC"
  else
    super
  end
end

#type_cast(value) ⇒ Object

:nodoc:



130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
 
# File 'activerecord/lib/active_record/connection_adapters/postgresql/quoting.rb', line 130

def type_cast(value) # :nodoc:
  case value
  when Type::Binary::Data
    # Return a bind param hash with format as binary.
    # See https://deveiate.org/code/pg/PG/Connection.html#method-i-exec_prepared-doc
    # for more information
    { value: value.to_s, format: 1 }
  when OID::Xml::Data, OID::Bit::Data
    value.to_s
  when OID::Array::Data
    encode_array(value)
  when Range
    encode_range(value)
  else
    super
  end
end

#unescape_bytea(value) ⇒ Object

Unescapes bytea output from a database to the binary string it represents. NOTE: This is NOT an inverse of escape_bytea! This is only to be used on escaped binary output from database drive.



21
22
23
 
# File 'activerecord/lib/active_record/connection_adapters/postgresql/quoting.rb', line 21

def unescape_bytea(value)
  valid_raw_connection.unescape_bytea(value) if value
end