1 files changed, 19 insertions, 0 deletions
diff --git a/lib/gitlab/database/partitioning.rb b/lib/gitlab/database/partitioning.rb
new file mode 100644
index 00000000000..bbde2063c41
--- /dev/null
+++ b/lib/gitlab/database/partitioning.rb
@@ -0,0 +1,19 @@
+# frozen_string_literal: true
+
+module Gitlab
+  module Database
+    module Partitioning
+      def self.register_models(models)
+        registered_models.merge(models)
+      end
+
+      def self.registered_models
+        @registered_models ||= Set.new
+      end
+
+      def self.sync_partitions(models_to_sync = registered_models)
+        MultiDatabasePartitionManager.new(models_to_sync).sync_partitions
+      end
+    end
+  end
+end