Class: ActiveRecord::ConnectionAdapters::CockroachDB::DatabaseTasks

Inherits:
Tasks::PostgreSQLDatabaseTasks
  • Object
show all
Defined in:
lib/active_record/connection_adapters/cockroachdb/database_tasks.rb

Instance Method Summary collapse

Instance Method Details

#structure_dump(filename, extra_flags = nil) ⇒ Object 



7
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
 
# File 'lib/active_record/connection_adapters/cockroachdb/database_tasks.rb', line 7

def structure_dump(filename, extra_flags=nil)
  if extra_flags
    raise "No flag supported yet, please raise an issue if needed. " \
      "https://github.com/cockroachdb/activerecord-cockroachdb-adapter/issues/new"
  end

  case ActiveRecord.dump_schemas
  when :all, String
    raise  "Custom schemas are not supported in CockroachDB. " \
      "See https://github.com/cockroachdb/cockroach/issues/26443."
  when :schema_search_path
    if configuration_hash[:schema_search_path]
      raise  "Custom schemas are not supported in CockroachDB. " \
        "See https://github.com/cockroachdb/cockroach/issues/26443."
    end
  end

  conn = ActiveRecord::Base.connection
  File.open(filename, "w") do |file|
    %w(SCHEMAS TYPES).each do |object_kind|
      ActiveRecord::Base.connection.execute("SHOW CREATE ALL #{object_kind}").each_row { file.puts _1 }
    end

    ignore_tables = ActiveRecord::SchemaDumper.ignore_tables.to_set

    conn.execute("SHOW CREATE ALL TABLES").each_row do |(sql)|
      if sql.start_with?("CREATE")
        table_name = sql[/CREATE TABLE (?:.*?\.)?\"?(.*?)[\" ]/, 1]
        next if ignore_tables.member?(table_name)
      elsif sql.start_with?("ALTER")
        table_name = sql[/ALTER TABLE (?:.*?\.)?\"?(.*?)[\" ]/, 1]
        ref_table_name = sql[/REFERENCES (?:.*?\.)?\"?(.*?)[\" ]/, 1]
        next if ignore_tables.member?(table_name) || ignore_tables.member?(ref_table_name)
      end

      file.puts sql
    end
  end
end

#structure_load(filename, extra_flags = nil) ⇒ Object 



47
48
49
50
51
52
53
54
 
# File 'lib/active_record/connection_adapters/cockroachdb/database_tasks.rb', line 47

def structure_load(filename, extra_flags=nil)
  if extra_flags
    raise "No flag supported yet, please raise an issue if needed. " \
      "https://github.com/cockroachdb/activerecord-cockroachdb-adapter/issues/new"
  end

  run_cmd("cockroach", ["sql", "--set", "errexit=false", "--file", filename], "loading")
end