Class: Shard

Inherits:
ApplicationRecord show all
Defined in:
app/models/shard.rb

Class Method Summary collapse

Methods inherited from ApplicationRecord

at_most, id_in, id_not_in, iid_in, pluck_primary_key, primary_key_in, safe_ensure_unique, safe_find_or_create_by, safe_find_or_create_by!, underscore, without_order

Class Method Details

.by_name(name) ⇒ Object


20
21
22
23
24
25
26
# File 'app/models/shard.rb', line 20

def self.by_name(name)
  transaction(requires_new: true) do
    find_or_create_by(name: name)
  end
rescue ActiveRecord::RecordNotUnique
  retry
end

.populate!Object

Store shard names from the configuration file in the database. This is not a list of active shards - we just want to assign an immutable, unique ID to every shard name for easy indexing / referencing.


7
8
9
10
11
12
13
14
15
16
17
18
# File 'app/models/shard.rb', line 7

def self.populate!
  return unless table_exists?

  # The GitLab config does not change for the lifecycle of the process
  in_config = Gitlab.config.repositories.storages.keys.map(&:to_s)
  in_db = all.pluck(:name)

  # This may race with other processes creating shards at the same time, but
  # `by_name` will handle that correctly
  missing = in_config - in_db
  missing.map { |name| by_name(name) }
end