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.



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

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 



176
177
178
 
# File 'lib/prometheus-config-builder/prometheus-config-builder.rb', line 176

def add(file)
  @files << file
end

#set_rules_dir(rules_dir) ⇒ Object 



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

def set_rules_dir(rules_dir)
  @rules_dir = rules_dir
end

#set_scrape_files_dir(scrape_files_dir) ⇒ Object 



194
195
196
 
# File 'lib/prometheus-config-builder/prometheus-config-builder.rb', line 194

def set_scrape_files_dir(scrape_files_dir)
  @scrape_files_dir = scrape_files_dir
end

#write_prometheus_yaml(destination, scrape_files_dir) ⇒ Object 



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
244
245
246
247
248
249
250
251
 
# File 'lib/prometheus-config-builder/prometheus-config-builder.rb', line 198

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, ".*"))
      if configs.length > 0
        puts "Generated data for file #{scrape_files_dir + "/" + File.basename(file.basename, ".*")}"
        data["scrape_configs"].push(*configs)
        $config_file_status.observe(1, labels)
      end
    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 



180
181
182
183
184
185
186
187
188
 
# File 'lib/prometheus-config-builder/prometheus-config-builder.rb', line 180

def write_rules(path)

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

  return hash
end