Module: EachShardWorker

Extended by:
ActiveSupport::Concern
Includes:
Gitlab::Utils::StrongMemoize
Included in:
RepositoryCheck::DispatchWorker
Defined in:
app/workers/concerns/each_shard_worker.rb

Instance Method Summary collapse

Instance Method Details

#each_eligible_shard ⇒ Object 



7
8
9
10
11
12
13
 
# File 'app/workers/concerns/each_shard_worker.rb', line 7

def each_eligible_shard
  Gitlab::ShardHealthCache.update(eligible_shard_names)

  eligible_shard_names.each do |shard_name|
    yield shard_name
  end
end

#eligible_shard_names ⇒ Object

override when you want to filter out some shards



16
17
18
 
# File 'app/workers/concerns/each_shard_worker.rb', line 16

def eligible_shard_names
  healthy_shard_names
end

#healthy_ready_shards ⇒ Object 



26
27
28
29
30
31
32
33
34
 
# File 'app/workers/concerns/each_shard_worker.rb', line 26

def healthy_ready_shards
  success_checks, failed_checks = ready_shards.partition(&:success)

  if failed_checks.any?
    ::Gitlab::AppLogger.error(message: 'Excluding unhealthy shards', failed_checks: failed_checks.map(&:payload), class: self.class.name)
  end

  success_checks
end

#healthy_shard_names ⇒ Object 



20
21
22
23
24
 
# File 'app/workers/concerns/each_shard_worker.rb', line 20

def healthy_shard_names
  strong_memoize(:healthy_shard_names) do
    healthy_ready_shards.map { |result| result.labels[:shard] }
  end
end

#ready_shards ⇒ Object 



36
37
38
 
# File 'app/workers/concerns/each_shard_worker.rb', line 36

def ready_shards
  Gitlab::HealthChecks::GitalyCheck.readiness
end