83 lines
2.1 KiB
Ruby
83 lines
2.1 KiB
Ruby
module RuboCop
|
|
# Module containing helper methods for writing migration cops.
|
|
module MigrationHelpers
|
|
WHITELISTED_TABLES = %i[
|
|
application_settings
|
|
plan_limits
|
|
].freeze
|
|
|
|
# Blacklisted tables due to:
|
|
# - size in GB (>= 10 GB on GitLab.com as of 02/2020)
|
|
# - number of records
|
|
BLACKLISTED_TABLES = %i[
|
|
audit_events
|
|
ci_build_trace_sections
|
|
ci_builds
|
|
ci_builds_metadata
|
|
ci_job_artifacts
|
|
ci_pipeline_variables
|
|
ci_pipelines
|
|
ci_stages
|
|
deployments
|
|
events
|
|
issues
|
|
merge_request_diff_commits
|
|
merge_request_diff_files
|
|
merge_request_diffs
|
|
merge_request_metrics
|
|
merge_requests
|
|
namespaces
|
|
note_diff_files
|
|
notes
|
|
project_authorizations
|
|
projects
|
|
project_ci_cd_settings
|
|
project_features
|
|
push_event_payloads
|
|
resource_label_events
|
|
routes
|
|
sent_notifications
|
|
services
|
|
system_note_metadata
|
|
taggings
|
|
todos
|
|
users
|
|
web_hook_logs
|
|
].freeze
|
|
|
|
# Blacklisted tables due to:
|
|
# - number of columns (> 50 on GitLab.com as of 03/2020)
|
|
# - number of records
|
|
WIDE_TABLES = %i[
|
|
users
|
|
projects
|
|
ci_builds
|
|
].freeze
|
|
|
|
# List of helpers that add new columns, either directly (ADD_COLUMN_METHODS)
|
|
# or through a create/alter table (TABLE_METHODS)
|
|
ADD_COLUMN_METHODS = %i(add_column add_column_with_default change_column_type_concurrently).freeze
|
|
|
|
TABLE_METHODS = %i(create_table create_table_if_not_exists change_table).freeze
|
|
|
|
# Returns true if the given node originated from the db/migrate directory.
|
|
def in_migration?(node)
|
|
dirname(node).end_with?('db/migrate', 'db/geo/migrate') || in_post_deployment_migration?(node)
|
|
end
|
|
|
|
def in_post_deployment_migration?(node)
|
|
dirname(node).end_with?('db/post_migrate', 'db/geo/post_migrate')
|
|
end
|
|
|
|
def version(node)
|
|
File.basename(node.location.expression.source_buffer.name).split('_').first.to_i
|
|
end
|
|
|
|
private
|
|
|
|
def dirname(node)
|
|
File.dirname(node.location.expression.source_buffer.name)
|
|
end
|
|
end
|
|
end
|