Class: PrometheusConfigBuilder::ConfigFiles

Inherits:
Object
  • Object
show all
Includes:
PrometheusConfigBuilderLogger
Defined in:
lib/prometheus-config-builder/prometheus-config-builder.rb

Instance Method Summary collapse

Methods included from PrometheusConfigBuilderLogger

#logger, logger, logger=

Constructor Details

#initialize(prometheus_yaml_filename) ⇒ ConfigFiles

Returns a new instance of ConfigFiles.



164
165
166
167
168
169
 
# File 'lib/prometheus-config-builder/prometheus-config-builder.rb', line 164

def initialize(prometheus_yaml_filename)
  @files = []
  @prometheus_yaml_filename = prometheus_yaml_filename
  @rules_dir = nil
  @scrape_files_dir = nil
end

Instance Method Details

#add(file) ⇒ Object 



171
172
173
 
# File 'lib/prometheus-config-builder/prometheus-config-builder.rb', line 171

def add(file)
  @files << file
end

#set_rules_dir(rules_dir) ⇒ Object 



185
186
187
 
# File 'lib/prometheus-config-builder/prometheus-config-builder.rb', line 185

def set_rules_dir(rules_dir)
  @rules_dir = rules_dir
end

#set_scrape_files_dir(scrape_files_dir) ⇒ Object 



189
190
191
 
# File 'lib/prometheus-config-builder/prometheus-config-builder.rb', line 189

def set_scrape_files_dir(scrape_files_dir)
  @scrape_files_dir = scrape_files_dir
end

#write_prometheus_yaml(destination, scrape_files_dir) ⇒ Object 



193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
 
# File 'lib/prometheus-config-builder/prometheus-config-builder.rb', line 193

def write_prometheus_yaml(destination, scrape_files_dir)
  data = YAML.load_file(@prometheus_yaml_filename)

  if !data["scrape_configs"]
    data["scrape_configs"] = []
  end

  FileUtils.mkdir_p(scrape_files_dir) if !File.exist?(scrape_files_dir)

  errors = 0
  @files.each do |file|

    # Build labels for the $config_file_status metric.
    labels = {
      file: file.source
    }
    labels[:owner] = file.owner if file.owner
    labels[:runbook_url] = file.runbook_url if file.runbook_url
    begin
      configs = file.get_scrape_configs(scrape_files_dir + "/" + File.basename(file.basename, ".*"))
      data["scrape_configs"].push(*configs)
      $config_file_status.observe(1, labels)
    rescue Exception => e
      logger.error("Unable to get scrape configs correctly from file #{file.source}. Error: #{e}")
      $config_file_status.observe(0, labels)
      errors += 1
    end
  end

  $config_has_errors.observe(errors)


  if !data["rule_files"] or data["rule_files"].class != Array
    data["rule_files"] = []
  end

  if @rules_dir != nil && !data["rule_files"].include?(@rules_dir)
    data["rule_files"] << File.expand_path(@rules_dir) + "/*.yaml"
  end

  File.open(destination, "w") do |file|
    file.write("# AUTOGENERATED FILE, DO NOT MODIFY\n")
    file.write("#\n")
    file.write("# This file was automatically generated by prometheus-config-from-s3 at #{Time.now.utc.iso8601}\n")
    file.write("# based on #{@prometheus_yaml_filename}\n")
    file.write("#\n\n")
    file.write(data.to_yaml)
  end

  return Digest::SHA256.digest data.to_yaml
end

#write_rules(path) ⇒ Object 



175
176
177
178
179
180
181
182
183
 
# File 'lib/prometheus-config-builder/prometheus-config-builder.rb', line 175

def write_rules(path)

  hash = ""
  @files.each do |file|
    hash += file.write_rules(path)
  end

  return hash
end