1 files changed, 23 insertions, 0 deletions
diff --git a/db/migrate/20220401113123_add_check_constraint_to_vsa_aggregation_runtime_data_columns.rb b/db/migrate/20220401113123_add_check_constraint_to_vsa_aggregation_runtime_data_columns.rb
new file mode 100644
index 00000000000..4863a1f0030
--- /dev/null
+++ b/db/migrate/20220401113123_add_check_constraint_to_vsa_aggregation_runtime_data_columns.rb
@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+class AddCheckConstraintToVsaAggregationRuntimeDataColumns < Gitlab::Database::Migration[1.0]
+  FULL_RUNTIMES_IN_SECONDS_CONSTRAINT = 'full_runtimes_in_seconds_size'
+  FULL_PROCESSED_RECORDS_CONSTRAINT = 'full_processed_records_size'
+
+  disable_ddl_transaction!
+
+  def up
+    add_check_constraint(:analytics_cycle_analytics_aggregations,
+                         'CARDINALITY(full_runtimes_in_seconds) <= 10',
+                         FULL_RUNTIMES_IN_SECONDS_CONSTRAINT)
+
+    add_check_constraint(:analytics_cycle_analytics_aggregations,
+                         'CARDINALITY(full_processed_records) <= 10',
+                         FULL_PROCESSED_RECORDS_CONSTRAINT)
+  end
+
+  def down
+    remove_check_constraint :analytics_cycle_analytics_aggregations, FULL_RUNTIMES_IN_SECONDS_CONSTRAINT
+    remove_check_constraint :analytics_cycle_analytics_aggregations, FULL_PROCESSED_RECORDS_CONSTRAINT
+  end
+end