File: schema_dumping_helper.rb

package info (click to toggle)
rails 2%3A7.2.2.1%2Bdfsg-7
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 43,352 kB
  • sloc: ruby: 349,799; javascript: 30,703; yacc: 46; sql: 43; sh: 29; makefile: 27
file content (28 lines) | stat: -rw-r--r-- 893 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
# frozen_string_literal: true

module SchemaDumpingHelper
  def dump_table_schema(*tables)
    pool = ActiveRecord::Base.connection_pool
    old_ignore_tables = ActiveRecord::SchemaDumper.ignore_tables
    pool.with_connection do |connection|
      ActiveRecord::SchemaDumper.ignore_tables = connection.data_sources - tables
    end

    output, = capture_io do
      ActiveRecord::SchemaDumper.dump(pool)
    end
    output
  ensure
    ActiveRecord::SchemaDumper.ignore_tables = old_ignore_tables
  end

  def dump_all_table_schema(ignore_tables = [], pool: ActiveRecord::Base.connection_pool)
    old_ignore_tables, ActiveRecord::SchemaDumper.ignore_tables = ActiveRecord::SchemaDumper.ignore_tables, ignore_tables
    output, = capture_io do
      ActiveRecord::SchemaDumper.dump(pool)
    end
    output
  ensure
    ActiveRecord::SchemaDumper.ignore_tables = old_ignore_tables
  end
end