debian-mirror-gitlab/lib/gitlab/database/partitioning.rb

84 lines
2.6 KiB
Ruby
Raw Normal View History

2021-11-11 11:23:49 +05:30
# frozen_string_literal: true
module Gitlab
module Database
module Partitioning
2022-04-04 11:22:00 +05:30
class TableWithoutModel < Gitlab::Database::SharedModel
include PartitionedTable
2021-11-11 11:23:49 +05:30
end
2021-11-18 22:05:49 +05:30
2021-12-11 22:18:48 +05:30
class << self
def register_models(models)
models.each do |model|
raise "#{model} should have partitioning strategy defined" unless model.respond_to?(:partitioning_strategy)
registered_models << model
end
end
def register_tables(tables)
registered_tables.merge(tables)
end
def sync_partitions_ignore_db_error
sync_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
2022-05-07 20:08:51 +05:30
def sync_partitions(models_to_sync = registered_for_sync, only_on: nil)
2021-12-11 22:18:48 +05:30
Gitlab::AppLogger.info(message: 'Syncing dynamic postgres partitions')
2022-05-07 20:08:51 +05:30
Gitlab::Database::EachDatabase.each_model_connection(models_to_sync, only_on: only_on) do |model|
2021-12-11 22:18:48 +05:30
PartitionManager.new(model).sync_partitions
end
Gitlab::AppLogger.info(message: 'Finished sync of dynamic postgres partitions')
end
def 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 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
def registered_models
@registered_models ||= Set.new
end
def registered_tables
@registered_tables ||= Set.new
end
private
def registered_for_sync
registered_models + registered_tables.map do |table|
2022-04-04 11:22:00 +05:30
table_without_model(**table)
end
end
def table_without_model(table_name:, partitioned_column:, strategy:, limit_connection_names: nil)
Class.new(TableWithoutModel).tap do |klass|
klass.table_name = table_name
klass.partitioned_by(partitioned_column, strategy: strategy)
klass.limit_connection_names = limit_connection_names
2021-12-11 22:18:48 +05:30
end
end
2021-11-18 22:05:49 +05:30
end
2021-11-11 11:23:49 +05:30
end
end
end