]> git.openstreetmap.org Git - chef.git/blob - cookbooks/prometheus/resources/exporter.rb
Merge remote-tracking branch 'github/pull/350' into master
[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 :listen_type, :kind_of => String, :default => "address"
26 property :user, :kind_of => String, :default => "root"
27 property :command, :kind_of => String
28 property :options, :kind_of => [String, Array]
29 property :environment, :kind_of => Hash, :default => {}
30 property :service, :kind_of => String
31
32 action :create do
33   systemd_service service_name do
34     description "Prometheus #{new_resource.exporter} exporter"
35     type "simple"
36     user new_resource.user
37     environment new_resource.environment
38     exec_start "#{executable_path} #{new_resource.command} #{executable_options}"
39     private_tmp true
40     protect_system "strict"
41     protect_home true
42     no_new_privileges true
43   end
44
45   service service_name do
46     action [:enable, :start]
47     subscribes :restart, "systemd_service[#{service_name}]"
48   end
49
50   firewall_rule "accept-prometheus-#{new_resource.exporter}" do
51     action :accept
52     source "osm"
53     dest "fw"
54     proto "tcp:syn"
55     dest_ports new_resource.port
56     only_if { node[:prometheus][:mode] == "external" }
57   end
58
59   node.default[:prometheus][:exporters][new_resource.port] = {
60     :name => new_resource.exporter, :address => listen_address
61   }
62 end
63
64 action :delete do
65   service service_name do
66     action [:disable, :stop]
67   end
68
69   systemd_service service_name do
70     action :delete
71   end
72 end
73
74 action :restart do
75   service service_name do
76     action :restart
77   end
78 end
79
80 action_class do
81   def service_name
82     if new_resource.service
83       "prometheus-#{new_resource.service}-exporter"
84     else
85       "prometheus-#{new_resource.exporter}-exporter"
86     end
87   end
88
89   def executable_path
90     "/opt/prometheus/exporters/#{new_resource.exporter}/#{new_resource.exporter}_exporter"
91   end
92
93   def executable_options
94     "--#{new_resource.listen_switch}=#{listen_argument} #{Array(new_resource.options).join(' ')}"
95   end
96
97   def listen_argument
98     case new_resource.listen_type
99     when "address" then listen_address
100     when "url" then "http://#{listen_address}/metrics"
101     end
102   end
103
104   def listen_address
105     if node[:prometheus][:mode] == "wireguard"
106       "[#{node[:prometheus][:address]}]:#{new_resource.port}"
107     else
108       "#{node[:prometheus][:address]}:#{new_resource.port}"
109     end
110   end
111 end
112
113 def after_created
114   subscribes :restart, "git[/opt/prometheus]"
115 end