mirror of
https://gh.wpcy.net/https://github.com/discourse/discourse.git
synced 2026-05-06 13:18:14 +08:00
c31035caf5
By default, rails makes timestamp columns (`created_at` and `updated_at`) non-nullable, we also have some required core and plugins columns we wouldn't necessarily want to enforce in the intermediate DB schema. It'll be better to set the default values for these during import instead of enforcing these at the converter level. This change adds support for globally modifying a column’s `nullable` state, defaulting all `created_at` columns to be `nullable` while allowing for table level overrides. --------- Co-authored-by: Gerhard Schlager <gerhard.schlager@discourse.org>
67 lines
2 KiB
Ruby
67 lines
2 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Migrations::Database::Schema
|
|
class GlobalConfig
|
|
attr_reader :excluded_column_names, :modified_columns
|
|
|
|
def initialize(schema_config)
|
|
@schema_config = schema_config
|
|
@excluded_table_names = load_globally_excluded_table_names.freeze
|
|
@excluded_column_names = load_globally_excluded_column_names.freeze
|
|
@modified_columns = load_globally_modified_columns.freeze
|
|
end
|
|
|
|
def excluded_table_name?(table_name)
|
|
@excluded_table_names.include?(table_name)
|
|
end
|
|
|
|
def modified_name(column_name)
|
|
if (modified_column = find_modified_column(column_name))
|
|
modified_column[:rename_to]
|
|
end
|
|
end
|
|
|
|
def modified_datatype(column_name)
|
|
if (modified_column = find_modified_column(column_name))
|
|
modified_column[:datatype]
|
|
end
|
|
end
|
|
|
|
def modified_nullable(column_name)
|
|
if (modified_column = find_modified_column(column_name))
|
|
modified_column[:nullable]
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
def find_modified_column(column_name)
|
|
@modified_columns.find { |column| column[:name] == column_name } ||
|
|
@modified_columns.find { |column| column[:name_regex]&.match?(column_name) }
|
|
end
|
|
|
|
def load_globally_excluded_table_names
|
|
table_names = @schema_config.dig(:global, :tables, :exclude)
|
|
table_names.presence&.to_set || Set.new
|
|
end
|
|
|
|
def load_globally_excluded_column_names
|
|
column_names = @schema_config.dig(:global, :columns, :exclude)
|
|
column_names.presence || []
|
|
end
|
|
|
|
def load_globally_modified_columns
|
|
modified_columns = @schema_config.dig(:global, :columns, :modify)
|
|
return {} if modified_columns.blank?
|
|
|
|
modified_columns.map do |column|
|
|
if column[:name_regex]
|
|
column[:name_regex_original] = column[:name_regex]
|
|
column[:name_regex] = Regexp.new(column[:name_regex])
|
|
end
|
|
column[:datatype] = column[:datatype]&.to_sym
|
|
column
|
|
end
|
|
end
|
|
end
|
|
end
|
