1 files changed, 28 insertions, 0 deletions
diff --git a/lib/gitlab/database/partitioning_migration_helpers/partitioned_foreign_key_validator.rb b/lib/gitlab/database/partitioning_migration_helpers/partitioned_foreign_key_validator.rb
new file mode 100644
index 00000000000..089cf2b8931
--- /dev/null
+++ b/lib/gitlab/database/partitioning_migration_helpers/partitioned_foreign_key_validator.rb
@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+
+module Gitlab
+  module Database
+    module PartitioningMigrationHelpers
+      class PartitionedForeignKeyValidator < ActiveModel::Validator
+        def validate(record)
+          validate_key_part(record, :from_table, :from_column)
+          validate_key_part(record, :to_table, :to_column)
+        end
+
+        private
+
+        def validate_key_part(record, table_field, column_field)
+          if !connection.table_exists?(record[table_field])
+            record.errors.add(table_field, 'must be a valid table')
+          elsif !connection.column_exists?(record[table_field], record[column_field])
+            record.errors.add(column_field, 'must be a valid column')
+          end
+        end
+
+        def connection
+          ActiveRecord::Base.connection
+        end
+      end
+    end
+  end
+end