Class: Dbsketch::Rendering::Meta::TableRenderer

Inherits:
Object
  • Object
show all
Defined in:
lib/dbsketch/rendering/meta/table_renderer.rb

Instance Method Summary collapse

Constructor Details

#initialize(column_renderer: nil, fk_renderer: nil, options: {}, keywords: {}) ⇒ TableRenderer

Returns a new instance of TableRenderer.

Raises:

  • (ArgumentError) 


16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
 
# File 'lib/dbsketch/rendering/meta/table_renderer.rb', line 16

def initialize column_renderer: nil, fk_renderer: nil, options: {}, keywords: {}
  ### Preconditions

  raise ArgumentError, "options is not a Hash" unless options.is_a? Hash
  raise ArgumentError, "keywords is not a Hash" unless keywords.is_a? Hash
  ###

  @options = {
    :pretty => false,
  }.merge options
  @keywords = {
    :check_constraint => "Dbsketch::Model::CheckConstraint.new",
    :primary_key    => "Dbsketch::Model::PrimaryKey.new",
    :foreign_key    => "Dbsketch::Model::ForeignKey.new",
    :table        => "Dbsketch::Model::Table.new",
    :unique_constraint  => "Dbsketch::Model::UniqueConstraint.new"
  }.merge keywords
  @column_renderer = (nil == column_renderer ? ColumnRenderer.new(:keywords => @keywords) : column_renderer)
  @fk_renderer = (nil == fk_renderer ? ForeignKeyRenderer.new(:foreign_key_keyword => @keywords[:foreign_key] ) : fk_renderer)
end

Instance Method Details

#alter(table_diff) ⇒ Object

Raises:

  • (ArgumentError) 


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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
 
# File 'lib/dbsketch/rendering/meta/table_renderer.rb', line 57

def alter table_diff
  ### Preconditions

  raise ArgumentError, "table_diff is not a Dbsketch::Comparison::TableDiff" unless table_diff.is_a? Dbsketch::Comparison::TableDiff
  ###

  queries = []

  # Columns

  deleted_columns = table_diff.columns.select { |d| d.deletion? }
  if not deleted_columns.empty?
    queries << "Table.alter #{table_diff.new_table.name} drop column\n\t#{deleted_columns.map { |d| d.old_column.name }.join("\n\t")}"
  end
  added_columns = table_diff.columns.select { |d| d.addition? }
  if not added_columns.empty?
    queries << "Table.alter #{table_diff.new_table.name} add column\n\t#{added_columns.map { |d| @column_renderer.create d.new_column }.join("\n\t")}"
  end
  changed_columns = table_diff.columns.select { |d| d.change? }
  if not changed_columns.empty?
    queries << "Table.alter #{table_diff.new_table.name} alter column\n\t#{changed_columns.map { |d| @column_renderer.create d.new_column }.join("\n\t")}"
  end

  # Primary key

  if nil != table_diff.primary_key
    queries << "Table.alter #{table_diff.new_table.name} drop primary key #{table_diff.primary_key.old_key.name}" if table_diff.primary_key.deletion?
    queries << "Table.alter #{table_diff.new_table.name} alter #{create_primary_key table_diff.new_table, table_diff.primary_key.new_key}" if table_diff.primary_key.change?
    queries << "Table.alter #{table_diff.new_table.name} add #{create_primary_key table_diff.new_table, table_diff.primary_key.new_key}" if table_diff.primary_key.addition?
  end

  # Check constraints

  deleted = table_diff.check_constraints.select { |d| d.deletion? }
  if not deleted.empty?
    queries << "Table.alter #{table_diff.new_table.name} drop check constraint\n\t#{deleted.map { |d| d.old_constraint.name }.join("\n\t")}"
  end
  added = table_diff.check_constraints.select { |d| d.addition? }
  if not added.empty?
    queries << "Table.alter #{table_diff.new_table.name} add\n\t#{added.map { |d| create_check_constraint d.new_constraint }.join("\n\t")}"
  end
  changed = table_diff.check_constraints.select { |d| d.change? }
  if not changed.empty?
    queries << "Table.alter #{table_diff.new_table.name} alter\n\t#{changed.map { |d| create_check_constraint d.new_constraint }.join("\n\t")}"
  end
  queries.flatten.join("\n")

  # Foreign keys

  deleted = table_diff.foreign_keys.select { |d| d.deletion? }
  if not deleted.empty?
    queries << "Table.alter #{table_diff.new_table.name} drop foreign key\n\t#{deleted.map { |d| d.old_key.name }.join("\n\t")}"
  end
  added = table_diff.foreign_keys.select { |d| d.addition? }
  if not added.empty?
    queries << "Table.alter #{table_diff.new_table.name} add\n\t#{added.map { |d| @fk_renderer.create d.new_key }.join("\n\t")}"
  end
  changed = table_diff.foreign_keys.select { |d| d.change? }
  if not changed.empty?
    queries << "Table.alter #{table_diff.new_table.name} alter\n\t#{changed.map { |d| @fk_renderer.create d.new_key }.join("\n\t")}"
  end
  queries.flatten.join("\n")

  # Unique constraints

  deleted = table_diff.unique_constraints.select { |d| d.deletion? }
  if not deleted.empty?
    queries << "Table.alter #{table_diff.new_table.name} drop unique constraint\n\t#{deleted.map { |d| d.old_constraint.name }.join("\n\t")}"
  end
  added = table_diff.unique_constraints.select { |d| d.addition? }
  if not added.empty?
    queries << "Table.alter #{table_diff.new_table.name} add\n\t#{added.map { |d| create_unique_constraint d.new_constraint, table_diff.new_table }.join("\n\t")}"
  end
  changed = table_diff.unique_constraints.select { |d| d.change? }
  if not changed.empty?
    queries << "Table.alter #{table_diff.new_table.name} alter\n\t#{changed.map { |d| create_unique_constraint d.new_constraint, table_diff.new_table }.join("\n\t")}"
  end
  queries.flatten.join("\n")
end

#create(table) ⇒ Object

Raises:

  • (ArgumentError) 


35
36
37
38
39
40
41
42
43
44
45
46
47
48
 
# File 'lib/dbsketch/rendering/meta/table_renderer.rb', line 35

def create table
  ### Preconditions

  raise ArgumentError, "table is not a Dbsketch::Model::Table" unless table.is_a? Dbsketch::Model::Table
  ###

  columns = collect_columns table
  str = "#{@keywords[:table]}(\"#{table.name}\""
  str << ", :meaning => \"#{table.meaning}\"" if nil != table.meaning
  str << ", :comment => \"#{table.comment}\"" if nil != table.comment
  str << ", #{columns}" if not columns.empty?
  str << ")"
  constraints = collect_constraints table
  str << "\n#{constraints.join("\n")}" if not constraints.empty?
  str
end

#drop(table) ⇒ Object

Raises:

  • (ArgumentError) 


50
51
52
53
54
55
 
# File 'lib/dbsketch/rendering/meta/table_renderer.rb', line 50

def drop table
  ### Preconditions

  raise ArgumentError, "table is not a Dbsketch::Model::Table" unless table.is_a? Dbsketch::Model::Table
  ###

  "Table.drop #{table.name}"
end