1 files changed, 22 insertions, 2 deletions
diff --git a/lib/gitlab/database/partitioning.rb b/lib/gitlab/database/partitioning.rb
index 71fb995577a..c517cf94c6e 100644
--- a/lib/gitlab/database/partitioning.rb
+++ b/lib/gitlab/database/partitioning.rb
@@ -12,11 +12,31 @@ module Gitlab
       end
 
       def self.sync_partitions(models_to_sync = registered_models)
-        MultiDatabasePartitionManager.new(models_to_sync).sync_partitions
+        Gitlab::AppLogger.info(message: 'Syncing dynamic postgres partitions')
+
+        Gitlab::Database::EachDatabase.each_model_connection(models_to_sync) do |model|
+          PartitionManager.new(model).sync_partitions
+        end
+
+        Gitlab::AppLogger.info(message: 'Finished sync of dynamic postgres partitions')
+      end
+
+      def self.report_metrics(models_to_monitor = registered_models)
+        partition_monitoring = PartitionMonitoring.new
+
+        Gitlab::Database::EachDatabase.each_model_connection(models_to_monitor) do |model|
+          partition_monitoring.report_metrics_for_model(model)
+        end
       end
 
       def self.drop_detached_partitions
-        MultiDatabasePartitionDropper.new.drop_detached_partitions
+        Gitlab::AppLogger.info(message: 'Dropping detached postgres partitions')
+
+        Gitlab::Database::EachDatabase.each_database_connection do
+          DetachedPartitionDropper.new.perform
+        end
+
+        Gitlab::AppLogger.info(message: 'Finished dropping detached postgres partitions')
       end
     end
   end