Class: PrometheusConfigBuilder::Builder

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_src, dst_dir) ⇒ Builder

Returns a new instance of Builder.



249
250
251
252
253
254
255
256
257
258
259
 
# File 'lib/prometheus-config-builder/prometheus-config-builder.rb', line 249

def initialize(prometheus_src, dst_dir)
  @prometheus_src = prometheus_src
  @dst_dir = dst_dir

  @discoverer = ConfigDiscover.new
  @paths = []
  @last_hash = ""

  @@config_files = PrometheusExporter::Client.default.register(:gauge, "prometheusconfigbuilder_config_files", "Number of found config files")

end

Instance Method Details

#add_path(path) ⇒ Object 



261
262
263
 
# File 'lib/prometheus-config-builder/prometheus-config-builder.rb', line 261

def add_path(path)
  @paths << path
end

#write_out ⇒ Object 



265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
 
# File 'lib/prometheus-config-builder/prometheus-config-builder.rb', line 265

def write_out
  hash = ""
  cfs = ConfigFiles.new(@prometheus_src)
  files = []
  @paths.each do |path|
    found_files = @discoverer.discover(path)
    logger.info("Found #{found_files.length} files from path #{path}")
    files.push(*found_files)
  end

  files.each do |file|
    cf = ConfigFile.new
    cf.load(file)
    cfs.add(cf)
  end

  @@config_files.observe(files.length)

  rules_dir = @dst_dir + "/rules"
  scrape_files_dir = @dst_dir + "/scrape_files"
  cfs.set_rules_dir(rules_dir)
  cfs.set_scrape_files_dir(scrape_files_dir)
  hash += cfs.write_prometheus_yaml(@dst_dir + "/prometheus.yaml", scrape_files_dir)

  FileUtils.mkdir_p(rules_dir) unless File.exist?(rules_dir)
  FileUtils.mkdir_p(scrape_files_dir) unless File.exist?(scrape_files_dir)
  hash += cfs.write_rules(rules_dir)

  # Determine if something had changed since the last write_out based on the generated hashes
  if hash != @last_hash
    @last_hash = hash
    logger.debug("Wrote new #{@prometheus_src} file for Prometheus.")
    return true
  else
    logger.debug("The Prometheus #{@prometheus_src} config file did not change.")
    return false
  end
end