1 files changed, 19 insertions, 11 deletions
diff --git a/lib/gitlab/database/background_migration/health_status.rb b/lib/gitlab/database/background_migration/health_status.rb
index 01f9c5eb5fd..9a283074b32 100644
--- a/lib/gitlab/database/background_migration/health_status.rb
+++ b/lib/gitlab/database/background_migration/health_status.rb
@@ -4,21 +4,29 @@ module Gitlab
   module Database
     module BackgroundMigration
       module HealthStatus
+        DEFAULT_INIDICATORS = [
+          Indicators::AutovacuumActiveOnTable,
+          Indicators::WriteAheadLog
+        ].freeze
+
         # Rather than passing along the migration, we use a more explicitly defined context
-        Context = Struct.new(:tables)
+        Context = Struct.new(:connection, :tables)
 
-        def self.evaluate(migration, indicator = Indicators::AutovacuumActiveOnTable)
-          signal = begin
-            indicator.new(migration.health_context).evaluate
-          rescue StandardError => e
-            Gitlab::ErrorTracking.track_exception(e, migration_id: migration.id,
-                                                  job_class_name: migration.job_class_name)
-            Signals::Unknown.new(indicator, reason: "unexpected error: #{e.message} (#{e.class})")
-          end
+        def self.evaluate(migration, indicators = DEFAULT_INIDICATORS)
+          indicators.map do |indicator|
+            signal = begin
+              indicator.new(migration.health_context).evaluate
+            rescue StandardError => e
+              Gitlab::ErrorTracking.track_exception(e, migration_id: migration.id,
+                                                    job_class_name: migration.job_class_name)
 
-          log_signal(signal, migration) if signal.log_info?
+              Signals::Unknown.new(indicator, reason: "unexpected error: #{e.message} (#{e.class})")
+            end
 
-          signal
+            log_signal(signal, migration) if signal.log_info?
+
+            signal
+          end
         end
 
         def self.log_signal(signal, migration)