Module: ActiveRecord::ConnectionAdapters::CipherStashPG::Quoting

Included in:
ActiveRecord::ConnectionAdapters::CipherStashPGAdapter
Defined in:
lib/active_record/connection_adapters/6.1/cipherstash_pg/quoting.rb,
lib/active_record/connection_adapters/7.0/cipherstash_pg/quoting.rb,
lib/active_record/connection_adapters/7.1/cipherstash_pg/quoting.rb

Defined Under Namespace

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 



28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
 
# File 'lib/active_record/connection_adapters/7.0/cipherstash_pg/quoting.rb', line 28

def check_int_in_range(value)
  if value.to_int > 9223372036854775807 || value.to_int < -9223372036854775808
    exception = "      Provided value outside of the range of a signed 64bit integer.\n\n      PostgreSQL will treat the column type in question as a numeric.\n      This may result in a slow sequential scan due to a comparison\n      being performed between an integer or bigint value and a numeric value.\n\n      To allow for this potentially unwanted behavior, set\n      ActiveRecord.raise_int_wider_than_64bit to false.\n    ERROR\n    raise IntegerOutOf64BitRange.new exception\n  end\nend\n"

#column_name_matcher ⇒ Object 



81
82
83
 
# File 'lib/active_record/connection_adapters/6.1/cipherstash_pg/quoting.rb', line 81

def column_name_matcher
  COLUMN_NAME
end

#column_name_with_order_matcher ⇒ Object 



85
86
87
 
# File 'lib/active_record/connection_adapters/6.1/cipherstash_pg/quoting.rb', line 85

def column_name_with_order_matcher
  COLUMN_NAME_WITH_ORDER
end

#escape_bytea(value) ⇒ Object

Escapes binary strings for bytea input to the database.



8
9
10
 
# File 'lib/active_record/connection_adapters/6.1/cipherstash_pg/quoting.rb', line 8

def escape_bytea(value)
  @connection.escape_bytea(value) if value
end

#lookup_cast_type_from_column(column) ⇒ Object

:nodoc:



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

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

#quote(value) ⇒ Object

:nodoc:



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
69
70
71
 
# File 'lib/active_record/connection_adapters/7.0/cipherstash_pg/quoting.rb', line 44

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.



46
47
48
 
# File 'lib/active_record/connection_adapters/6.1/cipherstash_pg/quoting.rb', line 46

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

#quote_default_expression(value, column) ⇒ Object

:nodoc:



64
65
66
67
68
69
70
71
72
73
74
75
 
# File 'lib/active_record/connection_adapters/6.1/cipherstash_pg/quoting.rb', line 64

def quote_default_expression(value, column) # :nodoc:
  if value.is_a?(Proc)
    value.call
  elsif column.type == :uuid && value.is_a?(String) && /\(\)/.match?(value)
    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.



37
38
39
 
# File 'lib/active_record/connection_adapters/6.1/cipherstash_pg/quoting.rb', line 37

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

#quote_string(s) ⇒ Object

Quotes strings for use in SQL input.



20
21
22
 
# File 'lib/active_record/connection_adapters/6.1/cipherstash_pg/quoting.rb', line 20

def quote_string(s) #:nodoc:
  @connection.escape(s)
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”



32
33
34
 
# File 'lib/active_record/connection_adapters/6.1/cipherstash_pg/quoting.rb', line 32

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 



41
42
43
 
# File 'lib/active_record/connection_adapters/6.1/cipherstash_pg/quoting.rb', line 41

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

#quoted_binary(value) ⇒ Object

:nodoc:



60
61
62
 
# File 'lib/active_record/connection_adapters/6.1/cipherstash_pg/quoting.rb', line 60

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

#quoted_date(value) ⇒ Object

Quote date/time values for use in SQL input.



51
52
53
54
55
56
57
58
 
# File 'lib/active_record/connection_adapters/6.1/cipherstash_pg/quoting.rb', line 51

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:



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

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.



15
16
17
 
# File 'lib/active_record/connection_adapters/6.1/cipherstash_pg/quoting.rb', line 15

def unescape_bytea(value)
  @connection.unescape_bytea(value) if value
end