1 files changed, 27 insertions, 0 deletions
diff --git a/app/helpers/admin/background_migrations_helper.rb b/app/helpers/admin/background_migrations_helper.rb
new file mode 100644
index 00000000000..698d81cc8a2
--- /dev/null
+++ b/app/helpers/admin/background_migrations_helper.rb
@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+module Admin
+  module BackgroundMigrationsHelper
+    def batched_migration_status_badge_class_name(migration)
+      class_names = {
+        'active' => 'badge-info',
+        'paused' => 'badge-warning',
+        'failed' => 'badge-danger',
+        'finished' => 'badge-success'
+      }
+
+      class_names[migration.status]
+    end
+
+    # The extra logic here is needed because total_tuple_count is just
+    # an estimate and completed_rows also does not account for last jobs
+    # whose batch size is likely larger than the actual number of rows processed
+    def batched_migration_progress(migration, completed_rows)
+      return 100 if migration.finished?
+      return 0 unless completed_rows.to_i > 0
+      return unless migration.total_tuple_count.to_i > 0
+
+      [100 * completed_rows / migration.total_tuple_count, 99].min
+    end
+  end
+end