Sha256: f5dd92555f3bd9f588cc04bdd7cc513e4f41d184569668ae6557277fec175501
Contents?: true
Size: 1.81 KB
Versions: 64
Compression:
Stored size: 1.81 KB
Contents
# encoding: UTF-8
module Prometheus
module Client
# LabelSetValidator ensures that all used label sets comply with the
# Prometheus specification.
class LabelSetValidator
# TODO: we might allow setting :instance in the future
RESERVED_LABELS = [:job, :instance].freeze
class LabelSetError < StandardError; end
class InvalidLabelSetError < LabelSetError; end
class InvalidLabelError < LabelSetError; end
class ReservedLabelError < LabelSetError; end
def initialize(reserved_labels = [])
@reserved_labels = (reserved_labels + RESERVED_LABELS).freeze
@validated = {}
end
def valid?(labels)
unless labels.respond_to?(:all?)
raise InvalidLabelSetError, "#{labels} is not a valid label set"
end
labels.all? do |key, _|
validate_symbol(key)
validate_name(key)
validate_reserved_key(key)
end
end
def validate(labels)
return labels if @validated.key?(labels.hash)
valid?(labels)
unless @validated.empty? || match?(labels, @validated.first.last)
raise InvalidLabelSetError, 'labels must have the same signature'
end
@validated[labels.hash] = labels
end
private
def match?(a, b)
a.keys.sort == b.keys.sort
end
def validate_symbol(key)
return true if key.is_a?(Symbol)
raise InvalidLabelError, "label #{key} is not a symbol"
end
def validate_name(key)
return true unless key.to_s.start_with?('__')
raise ReservedLabelError, "label #{key} must not start with __"
end
def validate_reserved_key(key)
return true unless @reserved_labels.include?(key)
raise ReservedLabelError, "#{key} is reserved"
end
end
end
end
Version data entries
64 entries across 64 versions & 1 rubygems