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.



257
258
259
260
261
262
263
264
265
266
267
 
# File 'lib/prometheus-config-builder/prometheus-config-builder.rb', line 257

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 



269
270
271
 
# File 'lib/prometheus-config-builder/prometheus-config-builder.rb', line 269

def add_path(path)
  @paths << path
end

#write_out ⇒ Object 



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
303
304
305
306
307
308
309
310
 
# File 'lib/prometheus-config-builder/prometheus-config-builder.rb', line 273

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