prometheus_client_addons 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: e92df93c0d837d7a85176aaf94e7c954f4eb1de51e0f0f7f87341414514e614b
+  data.tar.gz: 80329fa945177a4acabfca5be1caacfc4f57b6dcbc6d567a8a92bd6f14391d57
+SHA512:
+  metadata.gz: fb58dc08009041432e37ff55b1e3a14cbbb36ee9545154671ad1ea1dae62bfece3544c016601c49348984c7da12878e10e9d6adfefa04179446e7b5849981eb0
+  data.tar.gz: df185a4b8b278471a3c4e49d9e96e17daca4050e5eb3a34548589b96721425507b03ee227eeea2c2fd0a9173178e66e8b62e903353b65ece62308b8e333c4f9f

data/.gitignore

@@ -0,0 +1,13 @@
+/.bundle/
+/.yardoc
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/
+.ruby-*
+Gemfile.lock
+
+# rspec failure tracking
+.rspec_status

data/.rspec

@@ -0,0 +1,3 @@
+--format documentation
+--color
+--require spec_helper

data/.travis.yml

@@ -0,0 +1,7 @@
+---
+sudo: false
+language: ruby
+cache: bundler
+rvm:
+  - 2.6.3
+before_install: gem install bundler -v 1.17.2

data/Gemfile

@@ -0,0 +1,6 @@
+source "https://rubygems.org"
+
+git_source(:github) {|repo_name| "https://github.com/#{repo_name}" }
+
+# Specify your gem's dependencies in prometheus_client_addons.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) 2019 kraaaken
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

data/README.md

@@ -0,0 +1,273 @@
+# Prometheus Ruby Client Addons
+
+## Usage (with Rails)
+
+### Puma server
+
+config/initializers/prometheus.rb
+```ruby
+require 'prometheus/client'
+require 'prometheus_client_addons'
+
+prometheus = Prometheus::Client.registry
+puma = PrometheusClientAddons::Prometheus::Client::Puma.new(
+  prefix: 'puma', 
+  base_labels: { my_label: 'baz' }
+)
+prometheus.register(puma)
+```
+
+config/puma.rb
+```ruby
+activate_control_app('auto')
+plugin :prometheus_client_addons
+```
+
+Example response in non-clustered mode
+```
+# TYPE puma_backlog gauge
+# HELP puma_backlog How many requests are waiting
+puma_backlog{my_label="baz"} 0
+# TYPE puma_running gauge
+# HELP puma_running Number of running threads
+puma_running{my_label="baz"} 5
+# TYPE puma_pool_capacity gauge
+# HELP puma_pool_capacity Number of requests that can be processed right now
+puma_pool_capacity{my_label="baz"} 4
+# TYPE puma_max_threads gauge
+# HELP puma_max_threads Maximum number of worker threads
+puma_max_threads{my_label="baz"} 5
+```
+
+Example response in clustered mode
+```
+# TYPE puma_workers gauge
+# HELP puma_workers Number of workers
+puma_workers{my_label="baz"} 2
+# TYPE puma_phase gauge
+# HELP puma_phase Phase of worker
+puma_phase{my_label="baz"} 0
+# TYPE puma_booted_workers gauge
+# HELP puma_booted_workers Number of booted workers
+puma_booted_workers{my_label="baz"} 2
+# TYPE puma_old_workers gauge
+# HELP puma_old_workers Number of old workers
+puma_old_workers{my_label="baz"} 0
+# TYPE puma_worker_status_phase gauge
+# HELP puma_worker_status_phase Phase of worker
+puma_worker_status_phase{my_label="baz",worker_index="0"} 0
+puma_worker_status_phase{my_label="baz",worker_index="1"} 0
+# TYPE puma_worker_status_booted gauge
+# HELP puma_worker_status_booted Booted or not
+puma_worker_status_booted{my_label="baz",worker_index="0"} true
+puma_worker_status_booted{my_label="baz",worker_index="1"} true
+# TYPE puma_worker_status_last_checkin gauge
+# HELP puma_worker_status_last_checkin Last update
+puma_worker_status_last_checkin{my_label="baz",worker_index="0"} 2019-09-04T15:01:33Z
+puma_worker_status_last_checkin{my_label="baz",worker_index="1"} 2019-09-04T15:01:33Z
+# TYPE puma_worker_status_last_status_backlog gauge
+# HELP puma_worker_status_last_status_backlog How many objects have yet to be processed by the pool
+puma_worker_status_last_status_backlog{my_label="baz",worker_index="0"} 0
+puma_worker_status_last_status_backlog{my_label="baz",worker_index="1"} 0
+# TYPE puma_worker_status_last_status_running gauge
+# HELP puma_worker_status_last_status_running Number of running threads
+puma_worker_status_last_status_running{my_label="baz",worker_index="0"} 5
+puma_worker_status_last_status_running{my_label="baz",worker_index="1"} 5
+# TYPE puma_worker_status_last_status_pool_capacity gauge
+# HELP puma_worker_status_last_status_pool_capacity Number of requests that can be processed right now
+puma_worker_status_last_status_pool_capacity{my_label="baz",worker_index="0"} 5
+puma_worker_status_last_status_pool_capacity{my_label="baz",worker_index="1"} 5
+# TYPE puma_worker_status_last_status_max_threads gauge
+# HELP puma_worker_status_last_status_max_threads Maximum number of worker threads
+puma_worker_status_last_status_max_threads{my_label="baz",worker_index="0"} 5
+puma_worker_status_last_status_max_threads{my_label="baz",worker_index="1"} 5
+```
+
+### ActiveRecord::Base.connection_pool.stat
+
+*Now it's not working with pre-forked/clustered-mode web servers. Sorry.*
+
+config/initializers/prometheus.rb
+```ruby
+require 'prometheus/client'
+require 'prometheus_client_addons'
+
+prometheus = Prometheus::Client.registry
+pool = PrometheusClientAddons::Prometheus::Client::ActiveRecord.new(
+  prefix: 'activerecord', 
+  base_labels: { my_label: 'foo' }
+)
+prometheus.register(pool)
+```
+
+Example response
+```
+# TYPE activerecord_size gauge
+# HELP activerecord_size Size of pool
+activerecord_size{my_label="foo"} 5
+# TYPE activerecord_connections gauge
+# HELP activerecord_connections Connections count
+activerecord_connections{my_label="foo"} 0
+# TYPE activerecord_busy gauge
+# HELP activerecord_busy Busy count
+activerecord_busy{my_label="foo"} 0
+# TYPE activerecord_dead gauge
+# HELP activerecord_dead Dead count
+activerecord_dead{my_label="foo"} 0
+# TYPE activerecord_idle gauge
+# HELP activerecord_idle Idle count
+activerecord_idle{my_label="foo"} 0
+# TYPE activerecord_waiting gauge
+# HELP activerecord_waiting Num waiting in queue
+activerecord_waiting{my_label="foo"} 0
+# TYPE activerecord_checkout_timeout gauge
+# HELP activerecord_checkout_timeout Checkout timeout
+activerecord_checkout_timeout{my_label="foo"} 5
+```
+
+### GC.stat
+
+config/initializers/prometheus.rb
+```ruby
+require 'prometheus/client'
+require 'prometheus_client_addons'
+
+prometheus = Prometheus::Client.registry
+gc = PrometheusClientAddons::Prometheus::Client::GC.new(
+  prefix: 'gc',
+  base_labels: { my_label: 'foo' }
+)
+prometheus.register(gc)
+```
+
+Example response
+```
+# TYPE gc_count gauge
+# HELP gc_count count
+gc_count{my_label="foo"} 62
+# TYPE gc_heap_allocated_pages gauge
+# HELP gc_heap_allocated_pages heap_allocated_pages
+gc_heap_allocated_pages{my_label="foo"} 3593
+# TYPE gc_heap_sorted_length gauge
+# HELP gc_heap_sorted_length heap_sorted_length
+gc_heap_sorted_length{my_label="foo"} 3593
+# TYPE gc_heap_allocatable_pages gauge
+# HELP gc_heap_allocatable_pages heap_allocatable_pages
+gc_heap_allocatable_pages{my_label="foo"} 0
+# TYPE gc_heap_available_slots gauge
+# HELP gc_heap_available_slots heap_available_slots
+gc_heap_available_slots{my_label="foo"} 1464501
+# TYPE gc_heap_live_slots gauge
+# HELP gc_heap_live_slots heap_live_slots
+gc_heap_live_slots{my_label="foo"} 1464445
+# TYPE gc_heap_free_slots gauge
+# HELP gc_heap_free_slots heap_free_slots
+gc_heap_free_slots{my_label="foo"} 56
+# TYPE gc_heap_final_slots gauge
+# HELP gc_heap_final_slots heap_final_slots
+gc_heap_final_slots{my_label="foo"} 0
+# TYPE gc_heap_marked_slots gauge
+# HELP gc_heap_marked_slots heap_marked_slots
+gc_heap_marked_slots{my_label="foo"} 999525
+# TYPE gc_heap_eden_pages gauge
+# HELP gc_heap_eden_pages heap_eden_pages
+gc_heap_eden_pages{my_label="foo"} 3593
+# TYPE gc_heap_tomb_pages gauge
+# HELP gc_heap_tomb_pages heap_tomb_pages
+gc_heap_tomb_pages{my_label="foo"} 0
+# TYPE gc_total_allocated_pages gauge
+# HELP gc_total_allocated_pages total_allocated_pages
+gc_total_allocated_pages{my_label="foo"} 3593
+# TYPE gc_total_freed_pages gauge
+# HELP gc_total_freed_pages total_freed_pages
+gc_total_freed_pages{my_label="foo"} 0
+# TYPE gc_total_allocated_objects gauge
+# HELP gc_total_allocated_objects total_allocated_objects
+gc_total_allocated_objects{my_label="foo"} 4665284
+# TYPE gc_total_freed_objects gauge
+# HELP gc_total_freed_objects total_freed_objects
+gc_total_freed_objects{my_label="foo"} 3200839
+# TYPE gc_malloc_increase_bytes gauge
+# HELP gc_malloc_increase_bytes malloc_increase_bytes
+gc_malloc_increase_bytes{my_label="foo"} 16693664
+# TYPE gc_malloc_increase_bytes_limit gauge
+# HELP gc_malloc_increase_bytes_limit malloc_increase_bytes_limit
+gc_malloc_increase_bytes_limit{my_label="foo"} 30330547
+# TYPE gc_minor_gc_count gauge
+# HELP gc_minor_gc_count minor_gc_count
+gc_minor_gc_count{my_label="foo"} 48
+# TYPE gc_major_gc_count gauge
+# HELP gc_major_gc_count major_gc_count
+gc_major_gc_count{my_label="foo"} 14
+# TYPE gc_remembered_wb_unprotected_objects gauge
+# HELP gc_remembered_wb_unprotected_objects remembered_wb_unprotected_objects
+gc_remembered_wb_unprotected_objects{my_label="foo"} 4357
+# TYPE gc_remembered_wb_unprotected_objects_limit gauge
+# HELP gc_remembered_wb_unprotected_objects_limit remembered_wb_unprotected_objects_limit
+gc_remembered_wb_unprotected_objects_limit{my_label="foo"} 8690
+# TYPE gc_old_objects gauge
+# HELP gc_old_objects old_objects
+gc_old_objects{my_label="foo"} 992074
+# TYPE gc_old_objects_limit gauge
+# HELP gc_old_objects_limit old_objects_limit
+gc_old_objects_limit{my_label="foo"} 1743172
+# TYPE gc_oldmalloc_increase_bytes gauge
+# HELP gc_oldmalloc_increase_bytes oldmalloc_increase_bytes
+gc_oldmalloc_increase_bytes{my_label="foo"} 37700816
+# TYPE gc_oldmalloc_increase_bytes_limit gauge
+# HELP gc_oldmalloc_increase_bytes_limit oldmalloc_increase_bytes_limit
+gc_oldmalloc_increase_bytes_limit{my_label="foo"} 33438324
+```
+
+### CustomCollector
+
+config/initializers/prometheus.rb
+```ruby
+require 'prometheus/client'
+require 'prometheus_client_addons'
+
+prometheus = Prometheus::Client.registry
+custom_collector = PrometheusClientAddons::Prometheus::Client::CustomCollector.new(
+  name: 'custom_metric',
+  docstring: 'This is a custom metric',
+  base_labels: { my_label: 'foo' },
+  &Proc.new {
+    # return custom value from block
+    Time.now.to_i
+  }
+)
+prometheus.register(custom_collector)
+```
+
+Example response
+```
+# TYPE custom_metric gauge
+# HELP custom_metric This is a custom metric
+custom_metric{my_label="foo"} 1567612321
+```
+
+## Installation
+
+```ruby
+gem 'prometheus_client_addons'
+```
+
+```bash
+bundle install
+# or
+gem install prometheus_client_addons
+```
+
+## Development
+
+After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake spec` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
+
+To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and tags, and push the `.gem` file to [rubygems.org](https://rubygems.org).
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitHub at https://github.com/[USERNAME]/prometheus_client_addons.
+
+## License
+
+The gem is available as open source under the terms of the [MIT License](https://opensource.org/licenses/MIT).

data/Rakefile

@@ -0,0 +1,6 @@
+require "bundler/gem_tasks"
+require "rspec/core/rake_task"
+
+RSpec::Core::RakeTask.new(:spec)
+
+task :default => :spec

data/bin/console

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+
+require "bundler/setup"
+require "prometheus_client_addons"
+
+# You can add fixtures and/or initialization code here to make experimenting
+# with your gem easier. You can also use a different console, if you like.
+
+# (If you use this, don't forget to add pry to your Gemfile!)
+# require "pry"
+# Pry.start
+
+require "irb"
+IRB.start(__FILE__)

data/bin/setup

@@ -0,0 +1,8 @@
+#!/usr/bin/env bash
+set -euo pipefail
+IFS=$'\n\t'
+set -vx
+
+bundle install
+
+# Do any other automated setup that you need to do here

data/lib/prometheus_client_addons.rb

@@ -0,0 +1,9 @@
+require "prometheus_client_addons/version"
+
+module PrometheusClientAddons
+end
+
+require 'prometheus_client_addons/prometheus/client/puma'
+require 'prometheus_client_addons/prometheus/client/active_record'
+require 'prometheus_client_addons/prometheus/client/custom_collector'
+require 'prometheus_client_addons/prometheus/client/gc'

data/lib/prometheus_client_addons/prometheus/client/active_record.rb

@@ -0,0 +1,34 @@
+require 'prometheus_client_addons/prometheus/client/formats/text_extended'
+require 'prometheus_client_addons/prometheus/client/multi_metric'
+
+module PrometheusClientAddons
+  module Prometheus
+    module Client
+      class ActiveRecord < MultiMetric
+        HANDLES = {
+          size: 'Size of pool',
+          connections: 'Connections count',
+          busy: 'Busy count',
+          dead: 'Dead count',
+          idle: 'Idle count',
+          waiting: 'Num waiting in queue',
+          checkout_timeout: 'Checkout timeout'
+        }.freeze
+
+        def multi_name_type
+          full_handles = HANDLES.keys.map { |key| "#{prefix}#{key}" }.map(&:to_sym)
+          Hash[full_handles.zip([:gauge] * HANDLES.size)]
+        end
+
+        def multi_name_docstring
+          Hash[HANDLES.map { |key, value| ["#{prefix}#{key}".to_sym, value] }]
+        end
+
+        def multi_values
+          stat = ::ActiveRecord::Base.connection_pool.stat
+          Hash[stat.map { |key, value| ["#{prefix}#{key}".to_sym, { {} => value }] }]
+        end
+      end
+    end
+  end
+end

data/lib/prometheus_client_addons/prometheus/client/custom_collector.rb

@@ -0,0 +1,26 @@
+require 'prometheus/client/metric'
+
+module PrometheusClientAddons
+  module Prometheus
+    module Client
+      class CustomCollector < ::Prometheus::Client::Metric
+        def initialize(name:, docstring:, base_labels: {}, &block)
+          @custom_collector = block
+          super(name.to_sym, docstring, base_labels)
+        end
+
+        def get(labels = {})
+          @custom_collector.call
+        end
+
+        def values
+          { {} => @custom_collector.call }
+        end
+
+        def type
+          :gauge
+        end
+      end
+    end
+  end
+end

data/lib/prometheus_client_addons/prometheus/client/fake_gauge.rb

@@ -0,0 +1,17 @@
+module PrometheusClientAddons
+  module Prometheus
+    module Client
+      FakeGauge = Struct.new(:name, :docstring, :base_labels) do
+        def initialize(*)
+          self.base_labels ||= {}
+          super
+        end
+
+        def type
+          :gauge
+        end
+      end
+
+    end
+  end
+end

data/lib/prometheus_client_addons/prometheus/client/formats/text_extended.rb

@@ -0,0 +1,48 @@
+require 'prometheus/middleware/exporter'
+require 'prometheus/client/formats/text'
+require 'prometheus_client_addons/prometheus/client/multi_metric'
+require 'prometheus_client_addons/prometheus/client/fake_gauge'
+
+module PrometheusClientAddons
+  module Prometheus
+    module Client
+      module Formats
+        TextExtended = ::Prometheus::Client::Formats::Text.dup
+
+        module TextExtended
+          def self.marshal(registry)
+            lines = []
+
+            registry.metrics.each do |metric|
+              if metric.is_a?(MultiMetric)
+                metric.multi_values.each do |name, values|
+                  lines << format(self::TYPE_LINE, name, metric.multi_name_type[name])
+                  lines << format(self::HELP_LINE, name, escape(metric.multi_name_docstring[name]))
+
+                  values.each do |label_set, value|
+                    _metric = FakeGauge.new(name, metric.multi_name_docstring[name], metric.base_labels)
+                    representation(_metric, label_set, value) { |l| lines << l }
+                  end
+                end
+              else
+                lines << format(self::TYPE_LINE, metric.name, metric.type)
+                lines << format(self::HELP_LINE, metric.name, escape(metric.docstring))
+
+                metric.values.each do |label_set, value|
+                  representation(metric, label_set, value) { |l| lines << l }
+                end
+              end
+            end
+
+            (lines << nil).join(self::DELIMITER)
+          end
+        end
+      end
+    end
+  end
+end
+
+Prometheus::Middleware::Exporter.send(:remove_const, 'FORMATS')
+Prometheus::Middleware::Exporter.send(:remove_const, 'FALLBACK')
+Prometheus::Middleware::Exporter.send(:const_set, 'FORMATS', [PrometheusClientAddons::Prometheus::Client::Formats::TextExtended].freeze)
+Prometheus::Middleware::Exporter.send(:const_set, 'FALLBACK', PrometheusClientAddons::Prometheus::Client::Formats::TextExtended)

data/lib/prometheus_client_addons/prometheus/client/gc.rb

@@ -0,0 +1,54 @@
+require 'prometheus_client_addons/prometheus/client/formats/text_extended'
+require 'prometheus_client_addons/prometheus/client/multi_metric'
+
+module PrometheusClientAddons
+  module Prometheus
+    module Client
+      class GC < MultiMetric
+        KEYS = [
+          :count,
+          :heap_allocated_pages,
+          :heap_sorted_length,
+          :heap_allocatable_pages,
+          :heap_available_slots,
+          :heap_live_slots,
+          :heap_free_slots,
+          :heap_final_slots,
+          :heap_marked_slots,
+          :heap_eden_pages,
+          :heap_tomb_pages,
+          :total_allocated_pages,
+          :total_freed_pages,
+          :total_allocated_objects,
+          :total_freed_objects,
+          :malloc_increase_bytes,
+          :malloc_increase_bytes_limit,
+          :minor_gc_count,
+          :major_gc_count,
+          :remembered_wb_unprotected_objects,
+          :remembered_wb_unprotected_objects_limit,
+          :old_objects,
+          :old_objects_limit,
+          :oldmalloc_increase_bytes,
+          :oldmalloc_increase_bytes_limit
+        ].freeze
+
+        HANDLES = Hash[KEYS.zip(KEYS)].freeze
+
+        def multi_name_type
+          full_handles = HANDLES.keys.map { |key| "#{prefix}#{key}" }.map(&:to_sym)
+          Hash[full_handles.zip([:gauge] * HANDLES.size)]
+        end
+
+        def multi_name_docstring
+          Hash[HANDLES.map { |key, value| ["#{prefix}#{key}".to_sym, value] }]
+        end
+
+        def multi_values
+          stat = ::GC.stat
+          Hash[stat.map { |key, value| ["#{prefix}#{key}".to_sym, { {} => value }] }]
+        end
+      end
+    end
+  end
+end

data/lib/prometheus_client_addons/prometheus/client/multi_metric.rb

@@ -0,0 +1,38 @@
+require 'prometheus/client/metric'
+require 'prometheus/client/label_set_validator'
+
+module PrometheusClientAddons
+  module Prometheus
+    module Client
+      class MultiMetric < ::Prometheus::Client::Metric
+        attr_reader :name, :prefix, :base_labels
+
+        def initialize(prefix: '', base_labels: {})
+          prefix = "#{prefix}_" unless prefix == ''
+
+          @prefix = prefix
+          @name = prefix
+          @base_labels = base_labels
+
+          multi_name_type.keys.each { |name| validate_name("#{prefix}#{name}".to_sym) }
+          multi_name_docstring.keys.each { |name| validate_name("#{prefix}#{name}".to_sym) }
+          multi_name_docstring.values.each(&method(:validate_docstring))
+          @validator = ::Prometheus::Client::LabelSetValidator.new
+          @validator.valid?(base_labels)
+        end
+
+        def multi_name_type
+          fail('Should return hash {name => type}')
+        end
+
+        def multi_name_docstring
+          fail('Should return hash {name => docstrings}')
+        end
+
+        def multi_values
+          fail('Should return hash {name => {label_set => value, label_set => value, }}')
+        end
+      end
+    end
+  end
+end

data/lib/prometheus_client_addons/prometheus/client/puma.rb

@@ -0,0 +1,97 @@
+require 'json'
+require 'prometheus_client_addons/prometheus/client/formats/text_extended'
+require 'prometheus_client_addons/prometheus/client/multi_metric'
+require 'prometheus_client_addons/puma/plugin/prometheus_client_addons'
+
+module PrometheusClientAddons
+  module Prometheus
+    module Client
+      class Puma < MultiMetric
+        class << self
+          attr_accessor :control_url, :control_auth_token
+        end
+
+        HANDLES = {
+          backlog: 'How many requests are waiting',
+          running: 'Number of running threads',
+          pool_capacity: 'Number of requests that can be processed right now',
+          max_threads: 'Maximum number of worker threads',
+
+          workers: 'Number of workers',
+          phase: 'Phase of worker',
+          booted_workers: 'Number of booted workers',
+          old_workers: 'Number of old workers',
+
+          worker_status_phase: 'Phase of worker',
+          worker_status_booted: 'Booted or not',
+          worker_status_last_checkin: 'Last update',
+
+          worker_status_last_status_backlog: 'How many objects have yet to be processed by the pool',
+          worker_status_last_status_running: 'Number of running threads',
+          worker_status_last_status_pool_capacity: 'Number of requests that can be processed right now',
+          worker_status_last_status_max_threads: 'Maximum number of worker threads'
+        }.freeze
+
+        def multi_name_type
+          full_handles = HANDLES.keys.map { |key| "#{prefix}#{key}" }.map(&:to_sym)
+          Hash[full_handles.zip([:gauge] * HANDLES.size)]
+        end
+
+        def multi_name_docstring
+          Hash[HANDLES.map { |key, value| ["#{prefix}#{key}".to_sym, value] }]
+        end
+
+        def extract(input, output, nested, label_set = {})
+          key = if input.key?(nested.last.to_s)
+            nested.last.to_s
+          elsif input.key?(nested.last.to_sym)
+            nested.last.to_sym
+          end
+          return unless key
+
+          output["#{prefix}#{nested.join('_')}".to_sym] ||= {}
+          output["#{prefix}#{nested.join('_')}".to_sym][label_set] = input[key]
+        end
+
+        def multi_values
+          body = Socket.unix(self.class.control_url.gsub("unix://", '')) do |socket|
+            socket << "GET /stats?token=#{self.class.control_auth_token} HTTP/1.0\r\n\r\n"
+            socket.read
+          end
+
+          data = JSON.parse(body.split("\n").last, symbolize_names: true)
+
+          result = {}
+
+          extract(data, result, [:backlog])
+          extract(data, result, [:running])
+          extract(data, result, [:pool_capacity])
+          extract(data, result, [:max_threads])
+
+          extract(data, result, [:workers])
+          extract(data, result, [:phase])
+          extract(data, result, [:booted_workers])
+          extract(data, result, [:old_workers])
+
+          if data.key?(:worker_status) && !data[:worker_status].empty?
+            data[:worker_status].each do |worker_status|
+              worker_index = worker_status[:index]
+
+              extract(worker_status, result, [:worker_status, :phase], { worker_index: worker_index })
+              extract(worker_status, result, [:worker_status, :booted], { worker_index: worker_index })
+              extract(worker_status, result, [:worker_status, :last_checkin], { worker_index: worker_index })
+
+              last_status = worker_status[:last_status]
+              extract(last_status, result, [:worker_status, :last_status, :backlog], { worker_index: worker_index })
+              extract(last_status, result, [:worker_status, :last_status, :running], { worker_index: worker_index })
+              extract(last_status, result, [:worker_status, :last_status, :pool_capacity], { worker_index: worker_index })
+              extract(last_status, result, [:worker_status, :last_status, :max_threads], { worker_index: worker_index })
+            end
+          end
+
+          result
+        end
+      end
+    end
+  end
+end

data/lib/prometheus_client_addons/puma/plugin/prometheus_client_addons.rb

@@ -0,0 +1,19 @@
+require 'puma/plugin'
+
+module PrometheusClientAddons
+  module Puma
+    module Plugin
+      ::Puma::Plugin.create do
+        def start(launcher)
+          control_url = launcher.options[:control_url]
+          raise StandardError, "Need Puma's activate_control_app" if control_url == nil
+
+          Prometheus::Client::Puma.tap do |puma|
+            puma.control_url = control_url
+            puma.control_auth_token = launcher.options[:control_auth_token]
+          end
+        end
+      end
+    end
+  end
+end

data/lib/prometheus_client_addons/version.rb

@@ -0,0 +1,3 @@
+module PrometheusClientAddons
+  VERSION = "0.1.0"
+end

data/prometheus_client_addons.gemspec

@@ -0,0 +1,38 @@
+lib = File.expand_path("../lib", __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require "prometheus_client_addons/version"
+
+Gem::Specification.new do |spec|
+  spec.name = "prometheus_client_addons"
+  spec.version = PrometheusClientAddons::VERSION
+  spec.authors = ["Anton Osenenko"]
+  spec.email = ["anton.osenenko@gmail.com"]
+
+  spec.summary = 'The missed parts of prometheus/client_ruby'
+  spec.description = 'The missed parts of prometheus/client_ruby'
+  spec.homepage = "https://github.com/a0s/prometheus_client_addons"
+  spec.license = "MIT"
+
+  if spec.respond_to?(:metadata)
+    spec.metadata["allowed_push_host"] = "https://rubygems.org"
+    spec.metadata["homepage_uri"] = spec.homepage
+    spec.metadata["source_code_uri"] = spec.homepage
+  else
+    raise "RubyGems 2.0 or newer is required to protect against " \
+      "public gem pushes."
+  end
+
+  spec.files = Dir.chdir(File.expand_path('..', __FILE__)) do
+    `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  end
+  spec.bindir = "exe"
+  spec.executables = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency "prometheus-client", "~> 0.9.0"
+
+  spec.add_development_dependency "bundler", "~> 1.17"
+  spec.add_development_dependency "rake", "~> 10.0"
+  spec.add_development_dependency "rspec", "~> 3.0"
+  spec.add_development_dependency "puma"
+end

metadata

@@ -0,0 +1,136 @@
+--- !ruby/object:Gem::Specification
+name: prometheus_client_addons
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Anton Osenenko
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2019-09-04 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: prometheus-client
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.9.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.9.0
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.17'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.17'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+- !ruby/object:Gem::Dependency
+  name: puma
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: The missed parts of prometheus/client_ruby
+email:
+- anton.osenenko@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".rspec"
+- ".travis.yml"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/console
+- bin/setup
+- lib/prometheus_client_addons.rb
+- lib/prometheus_client_addons/prometheus/client/active_record.rb
+- lib/prometheus_client_addons/prometheus/client/custom_collector.rb
+- lib/prometheus_client_addons/prometheus/client/fake_gauge.rb
+- lib/prometheus_client_addons/prometheus/client/formats/text_extended.rb
+- lib/prometheus_client_addons/prometheus/client/gc.rb
+- lib/prometheus_client_addons/prometheus/client/multi_metric.rb
+- lib/prometheus_client_addons/prometheus/client/puma.rb
+- lib/prometheus_client_addons/puma/plugin/prometheus_client_addons.rb
+- lib/prometheus_client_addons/version.rb
+- prometheus_client_addons.gemspec
+homepage: https://github.com/a0s/prometheus_client_addons
+licenses:
+- MIT
+metadata:
+  allowed_push_host: https://rubygems.org
+  homepage_uri: https://github.com/a0s/prometheus_client_addons
+  source_code_uri: https://github.com/a0s/prometheus_client_addons
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.0.3
+signing_key: 
+specification_version: 4
+summary: The missed parts of prometheus/client_ruby
+test_files: []