diff options
author | GitLab Bot <gitlab-bot@gitlab.com> | 2020-03-30 18:07:51 +0300 |
---|---|---|
committer | GitLab Bot <gitlab-bot@gitlab.com> | 2020-03-30 18:07:51 +0300 |
commit | 4e9acbfba3682c552b3de707c535e6257ef41054 (patch) | |
tree | 8b1fd5f89ad3f1be68d8944815b13bb7d498e4a6 /rubocop/cop/migration | |
parent | 506d6dcd7c787ba71a8a53102f3d4fdb6adcfa5e (diff) |
Add latest changes from gitlab-org/gitlab@master
Diffstat (limited to 'rubocop/cop/migration')
-rw-r--r-- | rubocop/cop/migration/add_columns_to_wide_tables.rb | 47 |
1 files changed, 47 insertions, 0 deletions
diff --git a/rubocop/cop/migration/add_columns_to_wide_tables.rb b/rubocop/cop/migration/add_columns_to_wide_tables.rb new file mode 100644 index 00000000000..4618e4ae890 --- /dev/null +++ b/rubocop/cop/migration/add_columns_to_wide_tables.rb @@ -0,0 +1,47 @@ +# frozen_string_literal: true + +require_relative '../../migration_helpers' + +module RuboCop + module Cop + module Migration + # Cop that prevents adding columns to wide tables. + class AddColumnsToWideTables < RuboCop::Cop::Cop + include MigrationHelpers + + MSG = '`%s` is a wide table with several columns, addig more should be avoided unless absolutely necessary.' \ + ' Consider storing the column in a different table or creating a new one.'.freeze + + BLACKLISTED_METHODS = %i[ + add_column + add_column_with_default + add_reference + add_timestamps_with_timezone + ].freeze + + def on_send(node) + return unless in_migration?(node) + + method_name = node.children[1] + table_name = node.children[2] + + return unless offense?(method_name, table_name) + + add_offense(node, location: :selector, message: format(MSG, table_name.value)) + end + + private + + def offense?(method_name, table_name) + wide_table?(table_name) && + BLACKLISTED_METHODS.include?(method_name) + end + + def wide_table?(table_name) + table_name && table_name.type == :sym && + WIDE_TABLES.include?(table_name.value) + end + end + end + end +end |