Module: ActiveRecord::ConnectionAdapters::SQLite3::SchemaStatements

Included in:
ActiveRecord::ConnectionAdapters::SQLite3Adapter
Defined in:
activerecord/lib/active_record/connection_adapters/sqlite3/schema_statements.rb

Overview

:nodoc:

Instance Method Summary collapse

Instance Method Details

#add_check_constraint(table_name, expression, **options) ⇒ Object


99
100
101
102
103
# File 'activerecord/lib/active_record/connection_adapters/sqlite3/schema_statements.rb', line 99

def add_check_constraint(table_name, expression, **options)
  alter_table(table_name) do |definition|
    definition.check_constraint(expression, **options)
  end
end

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


55
56
57
58
59
60
# File 'activerecord/lib/active_record/connection_adapters/sqlite3/schema_statements.rb', line 55

def add_foreign_key(from_table, to_table, **options)
  alter_table(from_table) do |definition|
    to_table = strip_table_name_prefix_and_suffix(to_table)
    definition.foreign_key(to_table, **options)
  end
end

#check_constraints(table_name) ⇒ Object


83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
# File 'activerecord/lib/active_record/connection_adapters/sqlite3/schema_statements.rb', line 83

def check_constraints(table_name)
  table_sql = query_value(<<-SQL, "SCHEMA")
    SELECT sql
    FROM sqlite_master
    WHERE name = #{quote_table_name(table_name)} AND type = 'table'
    UNION ALL
    SELECT sql
    FROM sqlite_temp_master
    WHERE name = #{quote_table_name(table_name)} AND type = 'table'
  SQL

  table_sql.to_s.scan(/CONSTRAINT\s+(?<name>\w+)\s+CHECK\s+\((?<expression>(:?[^()]|\(\g<expression>\))+)\)/i).map do |name, expression|
    CheckConstraintDefinition.new(table_name, expression, name: name)
  end
end

#create_schema_dumper(options) ⇒ Object


112
113
114
# File 'activerecord/lib/active_record/connection_adapters/sqlite3/schema_statements.rb', line 112

def create_schema_dumper(options)
  SQLite3::SchemaDumper.create(self, options)
end

#indexes(table_name) ⇒ Object

Returns an array of indexes for the given table.


8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
# File 'activerecord/lib/active_record/connection_adapters/sqlite3/schema_statements.rb', line 8

def indexes(table_name)
  exec_query("PRAGMA index_list(#{quote_table_name(table_name)})", "SCHEMA").filter_map do |row|
    # Indexes SQLite creates implicitly for internal use start with "sqlite_".
    # See https://www.sqlite.org/fileformat2.html#intschema
    next if row["name"].start_with?("sqlite_")

    index_sql = query_value(<<~SQL, "SCHEMA")
      SELECT sql
      FROM sqlite_master
      WHERE name = #{quote(row['name'])} AND type = 'index'
      UNION ALL
      SELECT sql
      FROM sqlite_temp_master
      WHERE name = #{quote(row['name'])} AND type = 'index'
    SQL

    /\bON\b\s*"?(\w+?)"?\s*\((?<expressions>.+?)\)(?:\s*WHERE\b\s*(?<where>.+))?\z/i =~ index_sql

    columns = exec_query("PRAGMA index_info(#{quote(row['name'])})", "SCHEMA").map do |col|
      col["name"]
    end

    orders = {}

    if columns.any?(&:nil?) # index created with an expression
      columns = expressions
    else
      # Add info on sort order for columns (only desc order is explicitly specified,
      # asc is the default)
      if index_sql # index_sql can be null in case of primary key indexes
        index_sql.scan(/"(\w+)" DESC/).flatten.each { |order_column|
          orders[order_column] = :desc
        }
      end
    end

    IndexDefinition.new(
      table_name,
      row["name"],
      row["unique"] != 0,
      columns,
      where: where,
      orders: orders
    )
  end
end

#remove_check_constraint(table_name, expression = nil, **options) ⇒ Object


105
106
107
108
109
110
# File 'activerecord/lib/active_record/connection_adapters/sqlite3/schema_statements.rb', line 105

def remove_check_constraint(table_name, expression = nil, **options)
  check_constraints = check_constraints(table_name)
  chk_name_to_delete = check_constraint_for!(table_name, expression: expression, **options).name
  check_constraints.delete_if { |chk| chk.name == chk_name_to_delete }
  alter_table(table_name, foreign_keys(table_name), check_constraints)
end

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


62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
# File 'activerecord/lib/active_record/connection_adapters/sqlite3/schema_statements.rb', line 62

def remove_foreign_key(from_table, to_table = nil, **options)
  return if options.delete(:if_exists) == true && !foreign_key_exists?(from_table, to_table)

  to_table ||= options[:to_table]
  options = options.except(:name, :to_table, :validate)
  foreign_keys = foreign_keys(from_table)

  fkey = foreign_keys.detect do |fk|
    table = to_table || begin
      table = options[:column].to_s.delete_suffix("_id")
      Base.pluralize_table_names ? table.pluralize : table
    end
    table = strip_table_name_prefix_and_suffix(table)
    fk_to_table = strip_table_name_prefix_and_suffix(fk.to_table)
    fk_to_table == table && options.all? { |k, v| fk.options[k].to_s == v.to_s }
  end || raise(ArgumentError, "Table '#{from_table}' has no foreign key for #{to_table || options}")

  foreign_keys.delete(fkey)
  alter_table(from_table, foreign_keys)
end