Class: Fluent::PrometheusMonitorInput

Inherits:
Input
  • Object
show all
Defined in:
lib/fluent/plugin/in_prometheus_monitor.rb

Defined Under Namespace

Classes: TimerWatcher

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize ⇒ PrometheusMonitorInput

Returns a new instance of PrometheusMonitorInput.



11
12
13
14
 
# File 'lib/fluent/plugin/in_prometheus_monitor.rb', line 11

def initialize
  super
  @registry = ::Prometheus::Client.registry
end

Instance Attribute Details

#registry ⇒ Object (readonly)

Returns the value of attribute registry.



9
10
11
 
# File 'lib/fluent/plugin/in_prometheus_monitor.rb', line 9

def registry
  @registry
end

Instance Method Details

#configure(conf) ⇒ Object 



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
 
# File 'lib/fluent/plugin/in_prometheus_monitor.rb', line 16

def configure(conf)
  super
  hostname = Socket.gethostname
  expander = Fluent::Prometheus.placeholder_expnader(log)
  expander.prepare_placeholders(Time.now.to_i, {}, {'hostname' => hostname})
  @base_labels = Fluent::Prometheus.parse_labels_elements(conf)
  @base_labels.each do |key, value|
    @base_labels[key] = expander.expand(value)
  end

  @monitor_agent = MonitorAgentInput.new

  buffer_queue_length = @registry.gauge(
    :fluentd_status_buffer_queue_length,
    'Current buffer queue length.')
  buffer_total_queued_size = @registry.gauge(
    :fluentd_status_buffer_total_bytes,
    'Current total size of ququed buffers.')
  retry_counts = @registry.gauge(
    :fluentd_status_retry_count,
    'Current retry counts.')

  @monitor_info = {
    'buffer_queue_length' => buffer_queue_length,
    'buffer_total_queued_size' => buffer_total_queued_size,
    'retry_count' => retry_counts,
  }
end

#labels(plugin_info) ⇒ Object 



90
91
92
93
94
95
96
 
# File 'lib/fluent/plugin/in_prometheus_monitor.rb', line 90

def labels(plugin_info)
  @base_labels.merge(
    plugin_id: plugin_info["plugin_id"],
    plugin_category: plugin_info["plugin_category"],
    type: plugin_info["type"],
  )
end

#run ⇒ Object 



73
74
75
76
77
78
 
# File 'lib/fluent/plugin/in_prometheus_monitor.rb', line 73

def run
  @loop.run
rescue
  log.error "unexpected error", :error=>$!.to_s
  log.error_backtrace
end

#shutdown ⇒ Object 



67
68
69
70
71
 
# File 'lib/fluent/plugin/in_prometheus_monitor.rb', line 67

def shutdown
  @loop.watchers.each {|w| w.detach }
  @loop.stop
  @thread.join
end

#start ⇒ Object 



60
61
62
63
64
65
 
# File 'lib/fluent/plugin/in_prometheus_monitor.rb', line 60

def start
  @loop = Coolio::Loop.new
  @timer = TimerWatcher.new(@interval, true, log, &method(:update_monitor_info))
  @loop.attach(@timer)
  @thread = Thread.new(&method(:run))
end

#update_monitor_info ⇒ Object 



80
81
82
83
84
85
86
87
88
 
# File 'lib/fluent/plugin/in_prometheus_monitor.rb', line 80

def update_monitor_info
  @monitor_agent.plugins_info_all.each do |info|
    @monitor_info.each do |name, metric|
      if info[name]
        metric.set(labels(info), info[name])
      end
    end
  end
end