Greg Karekinian
79 lines
2.2 KiB
Ruby
79 lines
2.2 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'uri'
|
|
|
|
module PrometheusCookbook
|
|
module Helpers
|
|
def archive_name(component, version, url)
|
|
basename = ::File.basename(URI.parse(url).path)
|
|
basename.sub(/(?:\.tar\.gz|\.tgz|\.tar\.bz2|\.tar\.xz|\.zip)\z/, '')
|
|
rescue URI::InvalidURIError
|
|
"#{component}-#{version}"
|
|
end
|
|
|
|
def install_dir_parent(install_dir)
|
|
::File.dirname(install_dir)
|
|
end
|
|
|
|
def install_dir_name(install_dir)
|
|
::File.basename(install_dir)
|
|
end
|
|
|
|
def prometheus_flags(resource)
|
|
flag_pairs = if Gem::Version.new(resource.version) < Gem::Version.new('2.0.0-alpha.0')
|
|
resource.flags.map { |key, value| "-#{key}=#{value}" unless value == '' }
|
|
else
|
|
resource.cli_options.map { |key, value| "--#{key}=#{value}" unless value == '' } +
|
|
resource.cli_flags.map { |flag| "--#{flag}" unless flag == '' }
|
|
end
|
|
|
|
flag_pairs.compact.join(' ')
|
|
end
|
|
|
|
def prometheus_unit_content(resource)
|
|
{
|
|
Unit: {
|
|
Description: 'Prometheus',
|
|
After: 'network.target auditd.service',
|
|
},
|
|
Service: {
|
|
Type: 'simple',
|
|
Environment: "GOMAXPROCS=#{node['cpu']['total'] || 1}",
|
|
User: resource.user,
|
|
Group: resource.group,
|
|
ExecStart: "#{resource.binary} #{prometheus_flags(resource)}",
|
|
ExecReload: '/bin/kill -HUP $MAINPID',
|
|
Restart: 'always',
|
|
},
|
|
Install: {
|
|
WantedBy: 'multi-user.target',
|
|
},
|
|
}
|
|
end
|
|
|
|
def alertmanager_unit_content(resource)
|
|
{
|
|
Unit: {
|
|
Description: 'Prometheus Alertmanager',
|
|
After: 'network.target',
|
|
},
|
|
Service: {
|
|
User: resource.user,
|
|
Group: resource.group,
|
|
ExecStart: [
|
|
resource.binary,
|
|
"--log.level=#{resource.log_level}",
|
|
"--storage.path=#{resource.storage_path}",
|
|
"--config.file=#{resource.config_file}",
|
|
"--web.external-url=#{resource.external_url}",
|
|
].join(' '),
|
|
Restart: 'always',
|
|
},
|
|
Install: {
|
|
WantedBy: 'multi-user.target',
|
|
},
|
|
}
|
|
end
|
|
end
|
|
end
|