Class: Dbsketch::Automation::DatabaseImporter

Inherits:
Object
  • Object
show all
Defined in:
lib/dbsketch/automation/database_importer.rb

Instance Method Summary collapse

Constructor Details

#initialize(db) ⇒ DatabaseImporter

Returns a new instance of DatabaseImporter.

Raises:

  • (ArgumentError) 


15
16
17
18
19
20
21
 
# File 'lib/dbsketch/automation/database_importer.rb', line 15

def initialize db
  ### Preconditions

  raise ArgumentError, "database is not a Dbsketch::Automation::DatabaseProxy" unless db.is_a? DatabaseProxy
  ###

  @db = db
  @table_importer = TableImporter.new @db
end

Instance Method Details

#import(options: {}) ⇒ Object

Raises:

  • (ArgumentError) 


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
54
55
56
57
58
59
60
61
62
63
64
65
 
# File 'lib/dbsketch/automation/database_importer.rb', line 23

def import options: {}
  options = {
    :index_white_list => [], :index_black_list => [],
    :operation_white_list => [], :operation_black_list => [],
    :table_white_list => [], :table_black_list => [],
    :view_white_list => [], :view_black_list => []
  }.merge options
  options[:index_white_list] = (options[:index_white_list].is_a? Array) ? options[:index_white_list] : [options[:index_white_list]]
  options[:index_black_list] = (options[:index_black_list].is_a? Array) ? options[:index_black_list] : [options[:index_black_list]]
  options[:operation_white_list] = (options[:operation_white_list].is_a? Array) ? options[:operation_white_list] : [options[:operation_white_list]]
  options[:operation_black_list] = (options[:operation_black_list].is_a? Array) ? options[:operation_black_list] : [options[:operation_black_list]]
  options[:table_white_list] = (options[:table_white_list].is_a? Array) ? options[:table_white_list] : [options[:table_white_list]]
  options[:table_black_list] = (options[:table_black_list].is_a? Array) ? options[:table_black_list] : [options[:table_black_list]]
  options[:view_white_list] = (options[:view_white_list].is_a? Array) ? options[:view_white_list] : [options[:view_white_list]]
  options[:view_black_list] = (options[:view_black_list].is_a? Array) ? options[:view_black_list] : [options[:view_black_list]]
  ### Preconditions

  raise ArgumentError, "options[:operation_white_list] and options[:operation_black_list] are exclusive." if not options[:operation_white_list].empty? and not options[:operation_black_list].empty?
  raise ArgumentError, "options[:table_white_list] and options[:table_black_list] are exclusive." if not options[:table_white_list].empty? and not options[:table_black_list].empty?
  raise ArgumentError, "options[:view_white_list] and options[:view_black_list] are exclusive." if not options[:view_white_list].empty? and not options[:view_black_list].empty?
  ###


  imported_database = Dbsketch::Model::Database.new

  @db.fetch("select * from sys.tables").all.each do |db_table|
    if (options[:table_white_list].empty? or name_included_in_list? db_table[:name], options[:table_white_list]) and not name_included_in_list? db_table[:name], options[:table_black_list]
      table = import_table(db_table[:name])
      imported_database.add table
      import_indexes imported_database, table, db_table, options
      import_triggers imported_database, table, db_table
    end
  end

  imported_database.tables.each do |table|
    import_foreign_keys imported_database, table
  end

  import_views imported_database, options

  import_functions imported_database, options
  import_procedures imported_database, options

  imported_database
end

#import_table(table_name) ⇒ Object 



67
68
69
 
# File 'lib/dbsketch/automation/database_importer.rb', line 67

def import_table table_name
  @table_importer.import table_name
end