Drop temporary tracking table when finished

1 files changed, 6 insertions, 0 deletions

diff --git a/lib/gitlab/background_migration/populate_untracked_uploads.rb b/lib/gitlab/background_migration/populate_untracked_uploads.rb
index 8529e8d1d0b..f28892174bb 100644
--- a/lib/gitlab/background_migration/populate_untracked_uploads.rb
+++ b/lib/gitlab/background_migration/populate_untracked_uploads.rb
@@ -208,6 +208,8 @@ module Gitlab
             # to do.
           end
         end
+
+        drop_temp_table_if_finished
       end
 
       private
@@ -215,6 +217,10 @@ module Gitlab
       def migrate?
         UntrackedFile.table_exists? && Upload.table_exists?
       end
+
+      def drop_temp_table_if_finished
+        UntrackedFile.connection.drop_table(:untracked_files_for_uploads) if UntrackedFile.untracked.empty?
+      end
     end
   end
 end