module Gitlab module Metrics class UnicornSampler < BaseSampler def initialize(interval) super(interval) end def unicorn_active_connections @unicorn_active_connections ||= Gitlab::Metrics.gauge(:unicorn_active_connections, 'Unicorn active connections', {}, :max) end def unicorn_queued_connections @unicorn_queued_connections ||= Gitlab::Metrics.gauge(:unicorn_queued_connections, 'Unicorn queued connections', {}, :max) end def enabled? # Raindrops::Linux.tcp_listener_stats is only present on Linux unicorn_with_listeners? && Raindrops::Linux.respond_to?(:tcp_listener_stats) end def sample Raindrops::Linux.tcp_listener_stats(tcp_listeners).each do |addr, stats| unicorn_active_connections.set({ type: 'tcp', address: addr }, stats.active) unicorn_queued_connections.set({ type: 'tcp', address: addr }, stats.queued) end Raindrops::Linux.unix_listener_stats(unix_listeners).each do |addr, stats| unicorn_active_connections.set({ type: 'unix', address: addr }, stats.active) unicorn_queued_connections.set({ type: 'unix', address: addr }, stats.queued) end end private def tcp_listeners @tcp_listeners ||= Unicorn.listener_names.grep(%r{\A[^/]+:\d+\z}) end def unix_listeners @unix_listeners ||= Unicorn.listener_names - tcp_listeners end def unicorn_with_listeners? defined?(Unicorn) && Unicorn.listener_names.any? end end end end