File: google_cloud_profiler.rb

package info (click to toggle)
gitlab 17.6.5-19
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 629,368 kB
  • sloc: ruby: 1,915,304; javascript: 557,307; sql: 60,639; xml: 6,509; sh: 4,567; makefile: 1,239; python: 406
file content (25 lines) | stat: -rw-r--r-- 856 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
# frozen_string_literal: true

return unless Gitlab::Utils.to_boolean(ENV['GITLAB_GOOGLE_CLOUD_PROFILER_ENABLED'])
return unless ENV['GITLAB_GOOGLE_CLOUD_PROFILER_PROJECT_ID']

# For the initial iteration, we enable it only for `web`.
# This is because we have global service accounts configured this way, details:
# https://gitlab.com/gitlab-com/gl-infra/reliability/-/issues/17492#note_1303914983
return unless Gitlab::Runtime.puma?

Gitlab::Cluster::LifecycleEvents.on_worker_start do
  require 'cloud_profiler_agent'

  agent = CloudProfilerAgent::Agent.new(
    service: 'gitlab-web',
    project_id: ENV['GITLAB_GOOGLE_CLOUD_PROFILER_PROJECT_ID'],
    logger: ::Gitlab::AppJsonLogger.build,
    log_labels: {
      message: 'Google Cloud Profiler Ruby',
      pid: $$,
      worker_id: ::Prometheus::PidProvider.worker_id
    }
  )
  agent.start
end