]> git.openstreetmap.org Git - chef.git/blob - cookbooks/prometheus/resources/exporter.rb
22e364f9f42cd1522c5110603b7ea7171edd2336
[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 default_action :create
21
22 property :exporter, :kind_of => String, :name_property => true
23 property :port, :kind_of => Integer, :required => [:create]
24 property :listen_switch, :kind_of => String, :default => "web.listen-address"
25 property :user, :kind_of => String, :default => "root"
26 property :options, :kind_of => [String, Array]
27 property :environment, :kind_of => Hash, :default => {}
28
29 action :create do
30   systemd_service service_name do
31     description "Prometheus #{new_resource.exporter} exporter"
32     type "simple"
33     user new_resource.user
34     environment new_resource.environment
35     exec_start "#{executable_path} #{executable_options}"
36     private_tmp true
37     protect_system "strict"
38     protect_home true
39     no_new_privileges true
40   end
41
42   service service_name do
43     action [:enable, :start]
44     subscribes :restart, "systemd_service[#{service_name}]"
45   end
46
47   firewall_rule "accept-prometheus-#{new_resource.exporter}" do
48     action :accept
49     source "osm"
50     dest "fw"
51     proto "tcp:syn"
52     dest_ports new_resource.port
53     only_if { node[:prometheus][:mode] == "external" }
54   end
55
56   node.default[:prometheus][:exporters][new_resource.exporter] = listen_address
57 end
58
59 action :delete do
60   service service_name do
61     action [:disable, :stop]
62   end
63
64   systemd_service service_name do
65     action :delete
66   end
67 end
68
69 action :restart do
70   service service_name do
71     action :restart
72   end
73 end
74
75 action_class do
76   def service_name
77     "prometheus-#{new_resource.exporter}-exporter"
78   end
79
80   def executable_path
81     "/opt/prometheus/exporters/#{new_resource.exporter}/#{new_resource.exporter}_exporter"
82   end
83
84   def executable_options
85     "--#{new_resource.listen_switch}=#{listen_address} #{Array(new_resource.options).join(' ')}"
86   end
87
88   def listen_address
89     if node[:prometheus][:mode] == "wireguard"
90       "[#{node[:prometheus][:address]}]:#{new_resource.port}"
91     else
92       "#{node[:prometheus][:address]}:#{new_resource.port}"
93     end
94   end
95 end
96
97 def after_created
98   subscribes :restart, "git[/opt/prometheus]"
99 end