Module: ActiveRecord::ConnectionAdapters::Spanner::SchemaStatements

Included in:
ActiveRecord::ConnectionAdapters::SpannerAdapter
Defined in:
lib/active_record/connection_adapters/spanner/schema_statements.rb

Overview

SchemaStatements

Collection of methods to handle database schema.

Schema Doc

Constant Summary collapse

VERSION_6_1_0 = 
Gem::Version.create "6.1.0"
VERSION_6_0_3 = 
Gem::Version.create "6.0.3"

Instance Method Summary collapse

Instance Method Details

#_add_foreign_key(from_table, to_table, **options) ⇒ Object 



370
371
372
373
374
375
376
 
# File 'lib/active_record/connection_adapters/spanner/schema_statements.rb', line 370

def _add_foreign_key from_table, to_table, **options
  options = foreign_key_options from_table, to_table, options
  at = create_alter_table from_table
  at.add_foreign_key to_table, options

  execute_schema_statements schema_creation.accept(at)
end

#_remove_columns(table_name, *column_names) ⇒ Object 



180
181
182
183
184
185
186
187
188
189
190
191
192
193
 
# File 'lib/active_record/connection_adapters/spanner/schema_statements.rb', line 180

def _remove_columns table_name, *column_names
  if column_names.empty?
    raise ArgumentError, "You must specify at least one column name. "\
      "Example: remove_columns(:people, :first_name)"
  end

  statements = []

  column_names.each do |column_name|
    statements.concat drop_column_sql(table_name, column_name)
  end

  execute_schema_statements statements
end

#add_column(table_name, column_name, type, **options) ⇒ Object 



142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
 
# File 'lib/active_record/connection_adapters/spanner/schema_statements.rb', line 142

def add_column table_name, column_name, type, **options
  # Add column with NOT NULL not supported by spanner.
  # It is currently un-implemented state in spanner service.
  nullable = options.delete(:null) == false

  at = create_alter_table table_name
  at.add_column column_name, type, **options

  statements = [schema_creation.accept(at)]

  # Alter NOT NULL
  if nullable
    cd = at.adds.first.column
    cd.null = false
    ccd = Spanner::ChangeColumnDefinition.new(
      table_name, cd, column_name
    )
    statements << schema_creation.accept(ccd)
  end

  execute_schema_statements statements
end

#add_foreign_key(from_table, to_table, **options) ⇒ Object 



361
362
363
 
# File 'lib/active_record/connection_adapters/spanner/schema_statements.rb', line 361

def add_foreign_key from_table, to_table, options = {}
  _add_foreign_key from_table, to_table, **options
end

#add_index(table_name, column_name, options = {}) ⇒ Object 



270
271
272
273
274
275
276
277
278
279
280
 
# File 'lib/active_record/connection_adapters/spanner/schema_statements.rb', line 270

def add_index table_name, column_name, options = {}
  id = create_index_definition table_name, column_name, **options

  if data_source_exists?(table_name) &&
     index_name_exists?(table_name, id.name)
    raise ArgumentError, "Index name '#{id.name}' on table" \
                         "'#{table_name}' already exists"
  end

  execute_schema_statements schema_creation.accept(id)
end

#add_reference(table_name, ref_name, **options) ⇒ Object Also known as: add_belongs_to

Reference Column



391
392
393
394
395
 
# File 'lib/active_record/connection_adapters/spanner/schema_statements.rb', line 391

def add_reference table_name, ref_name, **options
  ReferenceDefinition.new(ref_name, **options).add_to(
    update_table_definition(table_name, self)
  )
end

#assume_migrated_upto_version(version) ⇒ Object 



404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
 
# File 'lib/active_record/connection_adapters/spanner/schema_statements.rb', line 404

def assume_migrated_upto_version version
  version = version.to_i
  sm_table = quote_table_name schema_migration.table_name

  migrated = migration_context.get_all_versions
  versions = migration_context.migrations.map(&:version)

  execute "INSERT INTO #{sm_table} (version) VALUES (#{quote version.to_s})" unless migrated.include? version

  inserting = (versions - migrated).select { |v| v < version }
  return unless inserting.any?

  if (duplicate = inserting.detect { |v| inserting.count(v) > 1 })
    raise "Duplicate migration #{duplicate}. Please renumber your migrations to resolve the conflict."
  end

  execute insert_versions_sql(inserting)
end

#change_column(table_name, column_name, type, **options) ⇒ Object 



196
197
198
 
# File 'lib/active_record/connection_adapters/spanner/schema_statements.rb', line 196

def change_column table_name, column_name, type, options = {}
  _change_column table_name, column_name, type, **options
end

#change_column_default(_table_name, _column_name, _default_or_changes) ⇒ Object

Raises:



209
210
211
212
 
# File 'lib/active_record/connection_adapters/spanner/schema_statements.rb', line 209

def change_column_default _table_name, _column_name, _default_or_changes
  raise ActiveRecordSpannerAdapter::NotSupportedError, \
        "change column with default value not supported."
end

#change_column_null(table_name, column_name, null, _default = nil) ⇒ Object 



205
206
207
 
# File 'lib/active_record/connection_adapters/spanner/schema_statements.rb', line 205

def change_column_null table_name, column_name, null, _default = nil
  change_column table_name, column_name, nil, null: null
end

#check_constraints(table_name) ⇒ Object

Check Contraints



400
401
402
 
# File 'lib/active_record/connection_adapters/spanner/schema_statements.rb', line 400

def check_constraints table_name
  information_schema { |i| i.check_constraints table_name }
end

#column_definitions(table_name) ⇒ Object

Column



117
118
119
 
# File 'lib/active_record/connection_adapters/spanner/schema_statements.rb', line 117

def column_definitions table_name
  information_schema { |i| i.table_columns table_name }
end

#create_join_table(table_1, table_2, column_options: {}, **options) ⇒ Object

Creates a join table that uses all the columns in the table as the primary key by default, unless an explicit primary key has been defined for the table. ActiveRecord will by default generate join tables without a primary key. Cloud Spanner however requires all tables to have a primary key. Instead of adding an additional column to the table only for the purpose of being the primary key, the Spanner ActiveRecord adapter defines a primary key that contains all the columns in the join table, as all values in the table should be unique anyways.



97
98
99
100
101
102
103
104
105
106
107
108
 
# File 'lib/active_record/connection_adapters/spanner/schema_statements.rb', line 97

def create_join_table table_1, table_2, column_options: {}, **options
  super do |td|
    unless td.columns.any?(&:primary_key?)
      td.columns.each do |col|
        def col.primary_key?
          true
        end
      end
    end
    yield td if block_given?
  end
end

#create_schema_dumper(options) ⇒ Object 



443
444
445
 
# File 'lib/active_record/connection_adapters/spanner/schema_statements.rb', line 443

def create_schema_dumper options
  SchemaDumper.create self, options
end

#create_table(table_name, id: :primary_key, **options) {|td| ... } ⇒ Object

Yields:

  • (td) 


52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
 
# File 'lib/active_record/connection_adapters/spanner/schema_statements.rb', line 52

def create_table table_name, id: :primary_key, **options
  td = create_table_definition table_name, options

  if id
    pk = options.fetch :primary_key do
      Base.get_primary_key table_name.to_s.singularize
    end
    id = id.fetch :type, :primary_key if id.is_a? Hash

    if pk.is_a? Array
      td.primary_keys pk
    else
      td.primary_key pk, id, **{}
    end
  end

  yield td if block_given?

  statements = []

  if options[:force]
    statements.concat drop_table_with_indexes_sql(table_name, options)
  end

  statements << schema_creation.accept(td)

  td.indexes.each do |column_name, index_options|
    id = create_index_definition table_name, column_name, **index_options
    statements << schema_creation.accept(id)
  end

  execute_schema_statements statements
end

#current_database ⇒ Object 



27
28
29
 
# File 'lib/active_record/connection_adapters/spanner/schema_statements.rb', line 27

def current_database
  @connection.database_id
end

#data_sources ⇒ Object Also known as: tables

Table



33
34
35
 
# File 'lib/active_record/connection_adapters/spanner/schema_statements.rb', line 33

def data_sources
  information_schema { |i| i.tables.map(&:name) }
end

#drop_table(table_name, options = {}) ⇒ Object 



86
87
88
89
 
# File 'lib/active_record/connection_adapters/spanner/schema_statements.rb', line 86

def drop_table table_name, options = {}
  statements = drop_table_with_indexes_sql table_name, options
  execute_schema_statements statements
end

#extract_schema_qualified_name(string) ⇒ Object 



43
44
45
46
47
48
49
50
 
# File 'lib/active_record/connection_adapters/spanner/schema_statements.rb', line 43

def extract_schema_qualified_name string
  schema, name = string.to_s.scan(/[^`.\s]+|`[^`]*`/)
  unless name
    name = schema
    schema = nil
  end
  [schema, name]
end

#fetch_type_metadata(sql_type, ordinal_position = nil, allow_commit_timestamp = nil, generated = nil) ⇒ Object 



134
135
136
137
138
139
140
 
# File 'lib/active_record/connection_adapters/spanner/schema_statements.rb', line 134

def fetch_type_metadata sql_type, ordinal_position = nil, allow_commit_timestamp = nil, generated = nil
  Spanner::TypeMetadata.new \
    super(sql_type),
    ordinal_position: ordinal_position,
    allow_commit_timestamp: allow_commit_timestamp,
    generated: generated
end

#foreign_keys(table_name, column: nil) ⇒ Object

Foreign Keys

Raises:

  • (ArgumentError) 


338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
 
# File 'lib/active_record/connection_adapters/spanner/schema_statements.rb', line 338

def foreign_keys table_name, column: nil
  raise ArgumentError if table_name.blank?

  result = information_schema { |i| i.foreign_keys table_name }

  if column
    result = result.select { |fk| fk.columns.include? column.to_s }
  end

  result.map do |fk|
    options = {
      column: fk.columns.first,
      name: fk.name,
      primary_key: fk.ref_columns.first,
      on_delete: fk.on_update,
      on_update: fk.on_update
    }

    ForeignKeyDefinition.new table_name, fk.ref_table, options
  end
end

#index_name_exists?(table_name, index_name) ⇒ Boolean

Returns:

  • (Boolean) 


266
267
268
 
# File 'lib/active_record/connection_adapters/spanner/schema_statements.rb', line 266

def index_name_exists? table_name, index_name
  information_schema { |i| i.index table_name, index_name }.present?
end

#indexes(table_name) ⇒ Object

Index



247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
 
# File 'lib/active_record/connection_adapters/spanner/schema_statements.rb', line 247

def indexes table_name
  result = information_schema do |i|
    i.indexes table_name, index_type: "INDEX"
  end

  result.map do |index|
    IndexDefinition.new(
      index.table,
      index.name,
      index.columns.map(&:name),
      unique: index.unique,
      null_filtered: index.null_filtered,
      interleave_in: index.interleave_in,
      storing: index.storing,
      orders: index.orders
    )
  end
end

#insert_versions_sql(versions) ⇒ Object 



423
424
425
426
427
428
429
430
431
432
433
434
 
# File 'lib/active_record/connection_adapters/spanner/schema_statements.rb', line 423

def insert_versions_sql versions
  sm_table = quote_table_name schema_migration.table_name

  if versions.is_a? Array
    sql = +"INSERT INTO #{sm_table} (version) VALUES\n"
    sql << versions.reverse.map { |v| "(#{quote v.to_s})" }.join(",\n")
    sql << ";"
    sql
  else
    "INSERT INTO #{sm_table} (version) VALUES (#{quote versions.to_s});"
  end
end

#new_column_from_field(_table_name, field, _definitions = nil) ⇒ Object 



121
122
123
124
125
126
127
128
129
130
131
132
 
# File 'lib/active_record/connection_adapters/spanner/schema_statements.rb', line 121

def new_column_from_field _table_name, field, _definitions = nil
  Spanner::Column.new \
    field.name,
    field.default,
    fetch_type_metadata(field.spanner_type,
                        field.ordinal_position,
                        field.allow_commit_timestamp,
                        field.generated),
    field.nullable,
    field.default_function,
    primary_key: field.primary_key
end

#primary_and_parent_keys(table_name) ⇒ Object 



328
329
330
331
332
333
334
 
# File 'lib/active_record/connection_adapters/spanner/schema_statements.rb', line 328

def primary_and_parent_keys table_name
  columns = information_schema do |i|
    i.table_primary_keys table_name, true
  end

  columns.map(&:name)
end

#primary_keys(table_name) ⇒ Object

Primary Keys



320
321
322
323
324
325
326
 
# File 'lib/active_record/connection_adapters/spanner/schema_statements.rb', line 320

def primary_keys table_name
  columns = information_schema do |i|
    i.table_primary_keys table_name
  end

  columns.map(&:name)
end

#quoted_scope(name = nil, type: nil) ⇒ Object 



436
437
438
439
440
441
 
# File 'lib/active_record/connection_adapters/spanner/schema_statements.rb', line 436

def quoted_scope name = nil, type: nil
  scope = { schema: quote("") }
  scope[:name] = quote name if name
  scope[:type] = quote type if type
  scope
end

#remove_column(table_name, column_name, _type = nil, _options = {}) ⇒ Object 



165
166
167
168
 
# File 'lib/active_record/connection_adapters/spanner/schema_statements.rb', line 165

def remove_column table_name, column_name, _type = nil, _options = {}
  statements = drop_column_sql table_name, column_name
  execute_schema_statements statements
end

#remove_columns(table_name, *column_names, _type: nil, **_options) ⇒ Object 



171
172
173
 
# File 'lib/active_record/connection_adapters/spanner/schema_statements.rb', line 171

def remove_columns table_name, *column_names
  _remove_columns table_name, *column_names
end

#remove_foreign_key(from_table, to_table = nil, **options) ⇒ Object 



378
379
380
381
382
383
384
385
386
387
 
# File 'lib/active_record/connection_adapters/spanner/schema_statements.rb', line 378

def remove_foreign_key from_table, to_table = nil, **options
  fk_name_to_delete = foreign_key_for!(
    from_table, to_table: to_table, **options
  ).name

  at = create_alter_table from_table
  at.drop_foreign_key fk_name_to_delete

  execute_schema_statements schema_creation.accept(at)
end

#remove_index(table_name, column_name = nil, **options) ⇒ Object 



283
284
285
286
 
# File 'lib/active_record/connection_adapters/spanner/schema_statements.rb', line 283

def remove_index table_name, options = {}
  index_name = index_name_for_remove table_name, options
  execute "DROP INDEX #{quote_table_name index_name}"
end

#rename_column(table_name, column_name, new_column_name) ⇒ Object 



214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
 
# File 'lib/active_record/connection_adapters/spanner/schema_statements.rb', line 214

def rename_column table_name, column_name, new_column_name
  if ActiveRecord::Base.connection.ddl_batch?
    raise ActiveRecordSpannerAdapter::NotSupportedError, \
          "rename_column in a DDL Batch is not supported."
  end
  column = information_schema do |i|
    i.table_column table_name, column_name
  end

  unless column
    raise ArgumentError,
          "Column '#{column_name}' not exist for table '#{table_name}'"
  end

  # Add Column
  cast_type = lookup_cast_type column.spanner_type
  add_column table_name, new_column_name, cast_type.type, **column.options

  # Copy data
  copy_data table_name, column_name, new_column_name

  # Recreate Indexes
  recreate_indexes table_name, column_name, new_column_name

  # Recreate Foreign keys
  recreate_foreign_keys table_name, column_name, new_column_name

  # Drop Indexes, Drop Foreign keys and columns
  remove_column table_name, column_name
end

#rename_index(table_name, old_name, new_name) ⇒ Object 



294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
 
# File 'lib/active_record/connection_adapters/spanner/schema_statements.rb', line 294

def rename_index table_name, old_name, new_name
  validate_index_length! table_name, new_name

  old_index = information_schema { |i| i.index table_name, old_name }
  return unless old_index

  statements = [
    schema_creation.accept(DropIndexDefinition.new(old_name))
  ]

  id = IndexDefinition.new \
    old_index.table,
    new_name,
    old_index.columns.map(&:name),
    unique: old_index.unique,
    null_filtered: old_index.null_filtered,
    interleave_in: old_index.interleave_in,
    storing: old_index.storing,
    orders: old_index.orders

  statements << schema_creation.accept(id)
  execute_schema_statements statements
end

#rename_table(_table_name, _new_name) ⇒ Object

Raises:



110
111
112
113
 
# File 'lib/active_record/connection_adapters/spanner/schema_statements.rb', line 110

def rename_table _table_name, _new_name
  raise ActiveRecordSpannerAdapter::NotSupportedError, \
        "rename_table is not implemented"
end

#table_exists?(table_name) ⇒ Boolean Also known as: data_source_exists?

Returns:

  • (Boolean) 


38
39
40
 
# File 'lib/active_record/connection_adapters/spanner/schema_statements.rb', line 38

def table_exists? table_name
  information_schema { |i| i.table table_name }.present?
end

#type_to_sql(type, limit: nil, precision: nil, scale: nil, **opts) ⇒ Object

rubocop:disable Lint/UnusedMethodArgument



448
449
450
451
452
453
454
455
456
457
458
459
460
 
# File 'lib/active_record/connection_adapters/spanner/schema_statements.rb', line 448

def type_to_sql type, limit: nil, precision: nil, scale: nil, **opts
  type = opts[:passed_type] || type&.to_sym
  native = native_database_types[type]

  return type.to_s unless native

  sql_type = (native.is_a?(Hash) ? native[:name] : native).dup

  sql_type = "#{sql_type}(#{limit || native[:limit]})" if [:string, :text, :binary].include? type
  sql_type = "ARRAY<#{sql_type}>" if opts[:array]

  sql_type
end