1 files changed, 7 insertions, 1 deletions
diff --git a/lib/gitlab/database/background_migration/batched_migration.rb b/lib/gitlab/database/background_migration/batched_migration.rb
index 9c8db2243f9..6aed1eed994 100644
--- a/lib/gitlab/database/background_migration/batched_migration.rb
+++ b/lib/gitlab/database/background_migration/batched_migration.rb
@@ -206,7 +206,7 @@ module Gitlab
         end
 
         def health_context
-          HealthStatus::Context.new([table_name])
+          HealthStatus::Context.new(connection, [table_name])
         end
 
         def hold!(until_time: 10.minutes.from_now)
@@ -231,6 +231,12 @@ module Gitlab
           "BatchedMigration[id: #{id}]"
         end
 
+        def progress
+          return unless total_tuple_count.to_i > 0
+
+          100 * migrated_tuple_count / total_tuple_count
+        end
+
         private
 
         def validate_batched_jobs_status