1 files changed, 41 insertions, 0 deletions
diff --git a/lib/gitlab/database/query_analyzers/ci/partitioning_analyzer.rb b/lib/gitlab/database/query_analyzers/ci/partitioning_analyzer.rb
new file mode 100644
index 00000000000..c2d5dfc1a15
--- /dev/null
+++ b/lib/gitlab/database/query_analyzers/ci/partitioning_analyzer.rb
@@ -0,0 +1,41 @@
+# frozen_string_literal: true
+
+module Gitlab
+  module Database
+    module QueryAnalyzers
+      module Ci
+        # The purpose of this analyzer is to detect queries not going through a partitioning routing table
+        class PartitioningAnalyzer < Database::QueryAnalyzers::Base
+          RoutingTableNotUsedError = Class.new(QueryAnalyzerError)
+
+          ENABLED_TABLES = %w[
+            ci_builds_metadata
+          ].freeze
+
+          class << self
+            def enabled?
+              ::Feature::FlipperFeature.table_exists? &&
+                ::Feature.enabled?(:ci_partitioning_analyze_queries, type: :ops)
+            end
+
+            def analyze(parsed)
+              analyze_legacy_tables_usage(parsed)
+            end
+
+            private
+
+            def analyze_legacy_tables_usage(parsed)
+              detected = ENABLED_TABLES & (parsed.pg.dml_tables + parsed.pg.select_tables)
+
+              return if detected.none?
+
+              ::Gitlab::ErrorTracking.track_and_raise_for_dev_exception(
+                RoutingTableNotUsedError.new("Detected non-partitioned table use #{detected.inspect}: #{parsed.sql}")
+              )
+            end
+          end
+        end
+      end
+    end
+  end
+end