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