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

Extended by:
ActiveSupport::Concern
Included in:
ActiveRecord::ConnectionAdapters::PostgreSQLAdapter
Defined in:
lib/active_record/connection_adapters/postgresql/quoting.rb

Defined Under Namespace

Modules: ClassMethods Classes: IntegerOutOf64BitRange

Constant Summary collapse

QUOTED_COLUMN_NAMES =

:nodoc:

Concurrent::Map.new
QUOTED_TABLE_NAMES =

:nodoc:

Concurrent::Map.new

Instance Method Summary collapse

Instance Method Details

#check_int_in_range(value) ⇒ Object



81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
# File 'lib/active_record/connection_adapters/postgresql/quoting.rb', line 81

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

#escape_bytea(value) ⇒ Object

Escapes binary strings for bytea input to the database.



70
71
72
# File 'lib/active_record/connection_adapters/postgresql/quoting.rb', line 70

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

#lookup_cast_type_from_column(column) ⇒ Object

:nodoc:



189
190
191
192
# File 'lib/active_record/connection_adapters/postgresql/quoting.rb', line 189

def lookup_cast_type_from_column(column) # :nodoc:
  verify! if type_map.nil?
  type_map.lookup(column.oid, column.fmod, column.sql_type)
end

#quote(value) ⇒ Object

:nodoc:



97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
# File 'lib/active_record/connection_adapters/postgresql/quoting.rb', line 97

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_default_expression(value, column) ⇒ Object

:nodoc:



156
157
158
159
160
161
162
163
164
165
166
167
# File 'lib/active_record/connection_adapters/postgresql/quoting.rb', line 156

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(schema_name) ⇒ Object

Quotes schema names for use in SQL queries.



138
139
140
# File 'lib/active_record/connection_adapters/postgresql/quoting.rb', line 138

def quote_schema_name(schema_name)
  quote_column_name(schema_name)
end

#quote_string(s) ⇒ Object

Quotes strings for use in SQL input.



127
128
129
130
131
# File 'lib/active_record/connection_adapters/postgresql/quoting.rb', line 127

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

#quote_table_name_for_assignment(table, attr) ⇒ Object



133
134
135
# File 'lib/active_record/connection_adapters/postgresql/quoting.rb', line 133

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

#quoted_binary(value) ⇒ Object

:nodoc:



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

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

#quoted_date(value) ⇒ Object

Quote date/time values for use in SQL input.



143
144
145
146
147
148
149
150
# File 'lib/active_record/connection_adapters/postgresql/quoting.rb', line 143

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:



169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
# File 'lib/active_record/connection_adapters/postgresql/quoting.rb', line 169

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)
  when Rational
    value.to_f
  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.



77
78
79
# File 'lib/active_record/connection_adapters/postgresql/quoting.rb', line 77

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