/
in_prometheus_tail_monitor.rb
94 lines (79 loc) · 2.56 KB
/
in_prometheus_tail_monitor.rb
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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
require 'fluent/plugin/input'
require 'fluent/plugin/in_monitor_agent'
require 'fluent/plugin/prometheus'
module Fluent::Plugin
class PrometheusTailMonitorInput < Fluent::Plugin::Input
Fluent::Plugin.register_input('prometheus_tail_monitor', self)
helpers :timer
config_param :interval, :time, default: 5
attr_reader :registry
MONITOR_IVARS = [
:tails,
]
def initialize
super
@registry = ::Prometheus::Client.registry
end
def multi_workers_ready?
true
end
def configure(conf)
super
hostname = Socket.gethostname
expander = Fluent::Plugin::Prometheus.placeholder_expander(log)
placeholders = expander.prepare_placeholders({'hostname' => hostname, 'worker_id' => fluentd_worker_id})
@base_labels = Fluent::Plugin::Prometheus.parse_labels_elements(conf)
@base_labels.each do |key, value|
unless value.is_a?(String)
raise Fluent::ConfigError, "record accessor syntax is not available in prometheus_tail_monitor"
end
@base_labels[key] = expander.expand(value, placeholders)
end
if defined?(Fluent::Plugin) && defined?(Fluent::Plugin::MonitorAgentInput)
# from v0.14.6
@monitor_agent = Fluent::Plugin::MonitorAgentInput.new
else
@monitor_agent = Fluent::MonitorAgentInput.new
end
end
def start
super
@metrics = {
position: @registry.gauge(
:fluentd_tail_file_position,
'Current position of file.'),
inode: @registry.gauge(
:fluentd_tail_file_inode,
'Current inode of file.'),
}
timer_execute(:in_prometheus_tail_monitor, @interval, &method(:update_monitor_info))
end
def update_monitor_info
opts = {
ivars: MONITOR_IVARS,
}
agent_info = @monitor_agent.plugins_info_all(opts).select {|info|
info['type'] == 'tail'.freeze
}
agent_info.each do |info|
tails = info['instance_variables'][:tails]
next if tails.nil?
tails.clone.each do |_, watcher|
# Access to internal variable of internal class...
# Very fragile implementation
pe = watcher.instance_variable_get(:@pe)
label = labels(info, watcher.path)
@metrics[:inode].set(label, pe.read_inode)
@metrics[:position].set(label, pe.read_pos)
end
end
end
def labels(plugin_info, path)
@base_labels.merge(
plugin_id: plugin_info["plugin_id"],
type: plugin_info["type"],
path: path,
)
end
end
end