File: puma.rb

package info (click to toggle)
ruby-prometheus-client-mmap 1.2.9-1
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 700 kB
  • sloc: ruby: 3,149; sh: 54; makefile: 21
file content (44 lines) | stat: -rw-r--r-- 1,020 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
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
module Prometheus
  module Client
    module Support
      module Puma
        extend self

        def worker_pid_provider
          wid = worker_id
          if wid = worker_id
            wid
          else
            "process_id_#{Process.pid}"
          end
        end

        private

        def object_based_worker_id
          return unless defined?(::Puma::Cluster::Worker)

          workers = ObjectSpace.each_object(::Puma::Cluster::Worker)
          return if workers.nil?

          workers_first = workers.first
          workers_first.index unless workers_first.nil?
        end

        def program_name
          $PROGRAM_NAME
        end

        def worker_id
          if matchdata = program_name.match(/puma.*cluster worker ([0-9]+):/)
            "puma_#{matchdata[1]}"
          elsif object_worker_id = object_based_worker_id
            "puma_#{object_worker_id}"
          elsif program_name.include?('puma')
            'puma_master'
          end
        end
      end
    end
  end
end