]> git.openstreetmap.org Git - chef.git/blob - cookbooks/prometheus/resources/exporter.rb
ae0596885b07220efd3d9a1d09ace9c3e34be211
[chef.git] / cookbooks / prometheus / resources / exporter.rb
1 #
2 # Cookbook:: prometheus
3 # Resource:: prometheus_exporter
4 #
5 # Copyright:: 2020, OpenStreetMap Foundation
6 #
7 # Licensed under the Apache License, Version 2.0 (the "License");
8 # you may not use this file except in compliance with the License.
9 # You may obtain a copy of the License at
10 #
11 #     https://www.apache.org/licenses/LICENSE-2.0
12 #
13 # Unless required by applicable law or agreed to in writing, software
14 # distributed under the License is distributed on an "AS IS" BASIS,
15 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16 # See the License for the specific language governing permissions and
17 # limitations under the License.
18 #
19
20 unified_mode true
21
22 default_action :create
23
24 property :exporter, :kind_of => String, :name_property => true
25 property :address, :kind_of => String
26 property :port, :kind_of => Integer, :required => [:create]
27 property :listen_switch, :kind_of => String, :default => "web.listen-address"
28 property :listen_type, :kind_of => String, :default => "address"
29 property :user, :kind_of => String, :default => "root"
30 property :command, :kind_of => String
31 property :options, :kind_of => [String, Array]
32 property :environment, :kind_of => Hash, :default => {}
33 property :service, :kind_of => String
34 property :metric_relabel, :kind_of => Array
35 property :register_target, :kind_of => [TrueClass, FalseClass], :default => true
36
37 action :create do
38   systemd_service service_name do
39     description "Prometheus #{new_resource.exporter} exporter"
40     type "simple"
41     user new_resource.user
42     environment new_resource.environment
43     exec_start "#{executable_path} #{new_resource.command} #{executable_options}"
44     private_tmp true
45     protect_system "strict"
46     protect_home true
47     no_new_privileges true
48   end
49
50   service service_name do
51     action [:enable, :start]
52     subscribes :restart, "systemd_service[#{service_name}]"
53   end
54
55   firewall_rule "accept-prometheus-#{new_resource.exporter}" do
56     action :accept
57     source "osm"
58     dest "fw"
59     proto "tcp:syn"
60     dest_ports new_resource.port
61     only_if { node[:prometheus][:mode] == "external" }
62   end
63
64   node.default[:prometheus][:addresses][new_resource.exporter] = listen_address
65
66   if new_resource.register_target
67     node.default[:prometheus][:exporters][new_resource.port] = {
68       :name => new_resource.exporter,
69       :address => listen_address,
70       :metric_relabel => new_resource.metric_relabel
71     }
72   end
73 end
74
75 action :delete do
76   service service_name do
77     action [:disable, :stop]
78   end
79
80   systemd_service service_name do
81     action :delete
82   end
83 end
84
85 action :restart do
86   service service_name do
87     action :restart
88     only_if { service_exists? }
89   end
90 end
91
92 action_class do
93   def service_name
94     if new_resource.service
95       "prometheus-#{new_resource.service}-exporter"
96     else
97       "prometheus-#{new_resource.exporter}-exporter"
98     end
99   end
100
101   def service_exists?
102     File.exist?("/etc/systemd/system/#{service_name}.service")
103   end
104
105   def executable_path
106     "/opt/prometheus/exporters/#{new_resource.exporter}/#{new_resource.exporter}_exporter"
107   end
108
109   def executable_options
110     "--#{new_resource.listen_switch}=#{listen_argument} #{Array(new_resource.options).join(' ')}"
111   end
112
113   def listen_argument
114     case new_resource.listen_type
115     when "address" then listen_address
116     when "url" then "http://#{listen_address}/metrics"
117     end
118   end
119
120   def listen_address
121     if new_resource.address
122       "#{new_resource.address}:#{new_resource.port}"
123     elsif node[:prometheus][:mode] == "wireguard"
124       "[#{node[:prometheus][:address]}]:#{new_resource.port}"
125     else
126       "#{node[:prometheus][:address]}:#{new_resource.port}"
127     end
128   end
129 end
130
131 def after_created
132   subscribes :restart, "git[/opt/prometheus]"
133 end