TL;DR

  • 实现puma插件，获取Puma::Laucher
  • 启动Puma::Server，接收/metrics请求，返回Puma::Laucher#stats
源码

lib/puma/plugin/metrics.rb 所定义的这个 puma plugin，如下。（[[puma的plugin机制]]）

基本上是使用 Puma::Server 来运行一个 Puma::Metrics::App（一个符合 Rack 标准的 app）

# lib/puma/plugin/metrics.rb
Puma::Plugin.create do
  def start(launcher)
    # ...
    app = Puma::Metrics::App.new launcher
    # ...
    metrics = Puma::Server.new app, launcher.events
    # ...
    metrics.run
  end
end

Puma::Metrics::App 代码如下，监控puma的stats（[[用puma的stats方法检查进程与线程状态]]）

module Puma
  module Metrics
    class App
      def initialize(launcher)
        @launcher = launcher
        clustered = (@launcher.options[:workers] || 0) > 0
        @parser = Parser.new(clustered: clustered)
      end

      def call(_env)
        retrieve_and_parse_stats!
        [
          200,
          { 'Content-Type' => 'text/plain' },
          [Prometheus::Client::Formats::Text.marshal(Prometheus::Client.registry)]
        ]
      end

      def retrieve_and_parse_stats!
        puma_stats = @launcher.stats
        if puma_stats.is_a?(Hash) # Modern Puma outputs stats as a Symbol-keyed Hash
          @parser.parse(puma_stats)
        else
          @parser.parse(JSON.parse(puma_stats, symbolize_names: true))
        end
      end
    end
  end
end