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 :package, :kind_of => String
  26 property :package_options, :kind_of => String
  27 property :defaults, :kind_of => String
  28 property :service, :kind_of => String
  29
  30 action :create do
  31   package package_name do
  32     options new_resource.package_options
  33   end
  34
  35   template defaults_name do
  36     cookbook "prometheus"
  37     source "defaults.erb"
  38     owner "root"
  39     group "root"
  40     mode "644"
  41     variables new_resource.to_hash.merge(:listen_address => listen_address)
  42   end
  43
  44   service service_name do
  45     action [:enable, :start]
  46     subscribes :restart, "template[#{defaults_name}]"
  47   end
  48
  49   firewall_rule "accept-prometheus-#{new_resource.name}" do
  50     action :accept
  51     source "osm"
  52     dest "fw"
  53     proto "tcp:syn"
  54     dest_ports new_resource.port
  55     only_if { node[:prometheus][:mode] == "external" }
  56   end
  57
  58   node.default[:prometheus][:exporters][new_resource.exporter] = listen_address
  59 end
  60
  61 action :delete do
  62   service service_name do
  63     action [:disable, :stop]
  64   end
  65
  66   package package_name do
  67     action :purge
  68   end
  69 end
  70
  71 action_class do
  72   def package_name
  73     new_resource.package || "prometheus-#{new_resource.exporter}-exporter"
  74   end
  75
  76   def defaults_name
  77     new_resource.defaults || "/etc/default/prometheus-#{new_resource.exporter}-exporter"
  78   end
  79
  80   def listen_address
  81     if node[:prometheus][:mode] == "wireguard"
  82       "[#{node[:prometheus][:address]}]:#{new_resource.port}"
  83     else
  84       "#{node[:prometheus][:address]}:#{new_resource.port}"
  85     end
  86   end
  87
  88   def service_name
  89     new_resource.service || "prometheus-#{new_resource.exporter}-exporter"
  90   end
  91 end