summaryrefslogtreecommitdiff
path: root/config/initializers/postgres_partitioning.rb
diff options
context:
space:
mode:
Diffstat (limited to 'config/initializers/postgres_partitioning.rb')
-rw-r--r--config/initializers/postgres_partitioning.rb4
1 files changed, 3 insertions, 1 deletions
diff --git a/config/initializers/postgres_partitioning.rb b/config/initializers/postgres_partitioning.rb
index 6c8a72d9bd5..b3f12c2ceb1 100644
--- a/config/initializers/postgres_partitioning.rb
+++ b/config/initializers/postgres_partitioning.rb
@@ -3,8 +3,10 @@
# Make sure we have loaded partitioned models here
# (even with eager loading disabled).
+Gitlab::Database::Partitioning::PartitionCreator.register(AuditEventPartitioned)
+
begin
- Gitlab::Database::Partitioning::PartitionCreator.new.create_partitions
+ Gitlab::Database::Partitioning::PartitionCreator.new.create_partitions unless ENV['DISABLE_POSTGRES_PARTITION_CREATION_ON_STARTUP']
rescue ActiveRecord::ActiveRecordError, PG::Error
# ignore - happens when Rake tasks yet have to create a database, e.g. for testing
end