Class: ActiveRecord::ConnectionAdapters::SpannerAdapter

Inherits:
AbstractAdapter
  • Object
show all
Extended by:
TypeMapBuilder
Includes:
ActiveRecord::ConnectionAdapters::Spanner::DatabaseStatements, ActiveRecord::ConnectionAdapters::Spanner::Quoting, ActiveRecord::ConnectionAdapters::Spanner::SchemaStatements, TypeMapBuilder
Defined in:
lib/active_record/connection_adapters/spanner_adapter.rb

Defined Under Namespace

Modules: TypeMapBuilder

Constant Summary collapse

ADAPTER_NAME = 
"spanner".freeze
NATIVE_DATABASE_TYPES = 
{
  primary_key:  "INT64",
  parent_key:   "INT64",
  string:       { name: "STRING", limit: "MAX" },
  text:         { name: "STRING", limit: "MAX" },
  integer:      { name: "INT64" },
  bigint:       { name: "INT64" },
  float:        { name: "FLOAT64" },
  decimal:      { name: "NUMERIC" },
  numeric:      { name: "NUMERIC" },
  datetime:     { name: "TIMESTAMP" },
  time:         { name: "TIMESTAMP" },
  date:         { name: "DATE" },
  binary:       { name: "BYTES", limit: "MAX" },
  boolean:      { name: "BOOL" },
  json:         { name: "JSON" }
}.freeze
TYPE_MAP = 
Type::TypeMap.new.tap { |m| initialize_type_map m }

Constants included from ActiveRecord::ConnectionAdapters::Spanner::SchemaStatements

ActiveRecord::ConnectionAdapters::Spanner::SchemaStatements::VERSION_6_0_3, ActiveRecord::ConnectionAdapters::Spanner::SchemaStatements::VERSION_6_1_0

Constants included from ActiveRecord::ConnectionAdapters::Spanner::DatabaseStatements

ActiveRecord::ConnectionAdapters::Spanner::DatabaseStatements::COMMENT_REGEX, ActiveRecord::ConnectionAdapters::Spanner::DatabaseStatements::RequestOptions, ActiveRecord::ConnectionAdapters::Spanner::DatabaseStatements::VERSION_7_1_0

Constants included from ActiveRecord::ConnectionAdapters::Spanner::Quoting

ActiveRecord::ConnectionAdapters::Spanner::Quoting::QUOTED_COLUMN_NAMES, ActiveRecord::ConnectionAdapters::Spanner::Quoting::QUOTED_TABLE_NAMES

Class Method Summary collapse

Instance Method Summary collapse

Methods included from ActiveRecord::ConnectionAdapters::Spanner::SchemaStatements

#_add_foreign_key, #_remove_columns, #add_column, #add_foreign_key, #add_index, #add_reference, #assume_migrated_upto_version, #change_column, #change_column_default, #change_column_null, #check_constraints, #column_definitions, #create_join_table, #create_schema_dumper, #create_table, #current_database, #data_sources, #drop_table, #extract_schema_qualified_name, #fetch_type_metadata, #foreign_keys, #index_name_exists?, #indexes, #insert_versions_sql, #new_column_from_field, #primary_and_parent_keys, #primary_keys, #quoted_scope, #remove_column, #remove_columns, #remove_foreign_key, #remove_index, #rename_column, #rename_index, #rename_table, #table_exists?, #type_to_sql

Methods included from ActiveRecord::ConnectionAdapters::Spanner::DatabaseStatements

#_has_pk_binding, #append_request_tag_from_query_logs, #append_request_tag_from_query_logs_with_format, #begin_db_transaction, #begin_isolated_db_transaction, #commit_db_transaction, #exec_mutation, #exec_query, #exec_update, #execute, #execute_ddl, #execute_query_or_dml, #internal_exec_query, #internal_execute, #query, #rollback_db_transaction, #sql_for_insert, #transaction, #transaction_isolation_levels, #truncate, #update, #write_query?

Methods included from ActiveRecord::ConnectionAdapters::Spanner::Quoting

#_type_cast, #quote_column_name, #quote_string, #quote_table_name, #quoted_binary

Constructor Details

#initialize(connection, logger, connection_options, config) ⇒ SpannerAdapter

Returns a new instance of SpannerAdapter.



72
73
74
75
76
77
 
# File 'lib/active_record/connection_adapters/spanner_adapter.rb', line 72

def initialize connection, logger, connection_options, config
  @connection = connection
  @connection_options = connection_options
  super connection, logger, config
  @raw_connection ||= connection
end

Class Method Details

.database_exists?(config) ⇒ Boolean

Database

Returns:

  • (Boolean) 


89
90
91
92
93
94
95
 
# File 'lib/active_record/connection_adapters/spanner_adapter.rb', line 89

def self.database_exists? config
  connection = ActiveRecordSpannerAdapter::Connection.new config
  connection.connect!
  true
rescue ActiveRecord::NoDatabaseError
  false
end

Instance Method Details

#active? ⇒ Boolean

Connection management

Returns:

  • (Boolean) 


99
100
101
 
# File 'lib/active_record/connection_adapters/spanner_adapter.rb', line 99

def active?
  @connection.active?
end

#arel_visitor ⇒ Object 



191
192
193
 
# File 'lib/active_record/connection_adapters/spanner_adapter.rb', line 191

def arel_visitor
  Arel::Visitors::Spanner.new self
end

#build_insert_sql(insert) ⇒ Object 



195
196
197
198
199
200
201
202
203
204
205
206
 
# File 'lib/active_record/connection_adapters/spanner_adapter.rb', line 195

def build_insert_sql insert
  if current_spanner_transaction&.isolation == :buffered_mutations
    raise "ActiveRecordSpannerAdapter does not support insert_sql with buffered_mutations transaction."
  end

  if insert.skip_duplicates? || insert.update_duplicates?
    raise NotImplementedError, "CloudSpanner does not support skip_duplicates and update_duplicates."
  end

  values_list, = insert.values_list
  "INSERT #{insert.into} #{values_list}"
end

#current_spanner_transaction ⇒ Object 



121
122
123
 
# File 'lib/active_record/connection_adapters/spanner_adapter.rb', line 121

def current_spanner_transaction
  @connection.current_transaction
end

#disconnect! ⇒ Object 



103
104
105
106
 
# File 'lib/active_record/connection_adapters/spanner_adapter.rb', line 103

def disconnect!
  super
  @connection.disconnect!
end

#log(sql, name = "SQL", binds = [], type_casted_binds = [], statement_name = nil, *args) ⇒ Object

Overwrite the standard log method to be able to translate exceptions.



273
274
275
276
277
278
279
 
# File 'lib/active_record/connection_adapters/spanner_adapter.rb', line 273

def log sql, name = "SQL", binds = [], type_casted_binds = [], statement_name = nil, *args
  super
rescue ActiveRecord::StatementInvalid
  raise
rescue StandardError => e
  raise translate_exception_class(e, sql, binds)
end

#max_identifier_length ⇒ Object 



79
80
81
 
# File 'lib/active_record/connection_adapters/spanner_adapter.rb', line 79

def max_identifier_length
  128
end

#native_database_types ⇒ Object 



83
84
85
 
# File 'lib/active_record/connection_adapters/spanner_adapter.rb', line 83

def native_database_types
  NATIVE_DATABASE_TYPES
end

#next_sequence_value(_sequence_name) ⇒ Object

Generate next sequence number for primary key



183
184
185
 
# File 'lib/active_record/connection_adapters/spanner_adapter.rb', line 183

def next_sequence_value _sequence_name
  SecureRandom.uuid.gsub("-", "").hex & 0x7FFFFFFFFFFFFFFF
end

#prefetch_primary_key?(_table_name = nil) ⇒ Boolean

Returns:

  • (Boolean) 


170
171
172
 
# File 'lib/active_record/connection_adapters/spanner_adapter.rb', line 170

def prefetch_primary_key? _table_name = nil
  true
end

#reset! ⇒ Object Also known as: reconnect! 



108
109
110
111
 
# File 'lib/active_record/connection_adapters/spanner_adapter.rb', line 108

def reset!
  super
  @connection.reset!
end

#return_value_after_insert?(column) ⇒ Boolean

Returns:

  • (Boolean) 


187
188
189
 
# File 'lib/active_record/connection_adapters/spanner_adapter.rb', line 187

def return_value_after_insert? column
  column.auto_incremented_by_db? || column.primary_key?
end

#spanner_schema_cache ⇒ Object 



114
115
116
 
# File 'lib/active_record/connection_adapters/spanner_adapter.rb', line 114

def spanner_schema_cache
  @spanner_schema_cache ||= SpannerSchemaCache.new self
end

#supports_bulk_alter? ⇒ Boolean

Supported features

Returns:

  • (Boolean) 


127
128
129
 
# File 'lib/active_record/connection_adapters/spanner_adapter.rb', line 127

def supports_bulk_alter?
  true
end

#supports_check_constraints? ⇒ Boolean

Returns:

  • (Boolean) 


174
175
176
 
# File 'lib/active_record/connection_adapters/spanner_adapter.rb', line 174

def supports_check_constraints?
  true
end

#supports_common_table_expressions? ⇒ Boolean

Returns:

  • (Boolean) 


131
132
133
 
# File 'lib/active_record/connection_adapters/spanner_adapter.rb', line 131

def supports_common_table_expressions?
  true
end

#supports_explain? ⇒ Boolean

Returns:

  • (Boolean) 


135
136
137
 
# File 'lib/active_record/connection_adapters/spanner_adapter.rb', line 135

def supports_explain?
  false
end

#supports_foreign_keys? ⇒ Boolean

Returns:

  • (Boolean) 


139
140
141
 
# File 'lib/active_record/connection_adapters/spanner_adapter.rb', line 139

def supports_foreign_keys?
  true
end

#supports_index_sort_order? ⇒ Boolean

Returns:

  • (Boolean) 


143
144
145
 
# File 'lib/active_record/connection_adapters/spanner_adapter.rb', line 143

def supports_index_sort_order?
  true
end

#supports_insert_on_conflict? ⇒ Boolean Also known as: supports_insert_on_duplicate_skip?, supports_insert_on_duplicate_update?, supports_insert_conflict_target?

Returns:

  • (Boolean) 


147
148
149
 
# File 'lib/active_record/connection_adapters/spanner_adapter.rb', line 147

def supports_insert_on_conflict?
  true
end

#supports_insert_returning? ⇒ Boolean

Returns:

  • (Boolean) 


154
155
156
 
# File 'lib/active_record/connection_adapters/spanner_adapter.rb', line 154

def supports_insert_returning?
  true
end

#supports_multi_insert? ⇒ Boolean

Returns:

  • (Boolean) 


158
159
160
 
# File 'lib/active_record/connection_adapters/spanner_adapter.rb', line 158

def supports_multi_insert?
  true
end

#supports_optimizer_hints? ⇒ Boolean

Returns:

  • (Boolean) 


162
163
164
 
# File 'lib/active_record/connection_adapters/spanner_adapter.rb', line 162

def supports_optimizer_hints?
  true
end

#supports_primary_key? ⇒ Boolean

Returns:

  • (Boolean) 


166
167
168
 
# File 'lib/active_record/connection_adapters/spanner_adapter.rb', line 166

def supports_primary_key?
  true
end

#supports_virtual_columns? ⇒ Boolean

Returns:

  • (Boolean) 


178
179
180
 
# File 'lib/active_record/connection_adapters/spanner_adapter.rb', line 178

def supports_virtual_columns?
  true
end

#transform(sql) ⇒ Object 



264
265
266
267
268
269
270
 
# File 'lib/active_record/connection_adapters/spanner_adapter.rb', line 264

def transform sql
  if ActiveRecord::VERSION::MAJOR >= 7
    transform_query sql
  else
    sql
  end
end

#translate_exception(exception, message:, sql:, binds:) ⇒ Object 



281
282
283
284
285
286
287
288
289
290
291
292
293
 
# File 'lib/active_record/connection_adapters/spanner_adapter.rb', line 281

def translate_exception exception, message:, sql:, binds:
  if exception.is_a? Google::Cloud::FailedPreconditionError
    case exception.message
    when /.*does not specify a non-null value for these NOT NULL columns.*/,
         /.*must not be NULL.*/
      NotNullViolation.new message, sql: sql, binds: binds
    else
      super
    end
  else
    super
  end
end