Sha256: fcf112be6ca67c1e68bc700c6988788e16ce85cea221a2d57aacfc49fb59633b
Contents?: true
Size: 1.99 KB
Versions: 7
Compression:
Stored size: 1.99 KB
Contents
# frozen_string_literal: true
require 'concurrent/hash'
module KubernetesDeploy
class ResourceCache
def initialize(namespace, context, logger)
@namespace = namespace
@context = context
@logger = logger
@kind_fetcher_locks = Concurrent::Hash.new { |hash, key| hash[key] = Mutex.new }
@data = Concurrent::Hash.new
@kubectl = Kubectl.new(namespace: @namespace, context: @context, logger: @logger, log_failure_by_default: false)
end
def get_instance(kind, resource_name, raise_if_not_found: false)
instance = use_or_populate_cache(kind).fetch(resource_name, {})
if instance.blank? && raise_if_not_found
raise KubernetesDeploy::Kubectl::ResourceNotFoundError, "Resource does not exist (used cache for kind #{kind})"
end
instance
rescue KubectlError
{}
end
def get_all(kind, selector = nil)
instances = use_or_populate_cache(kind).values
return instances unless selector
instances.select do |r|
labels = r.dig("metadata", "labels") || {}
labels >= selector
end
rescue KubectlError
[]
end
private
def statsd_tags
{ namespace: @namespace, context: @context }
end
def use_or_populate_cache(kind)
@kind_fetcher_locks[kind].synchronize do
return @data[kind] if @data.key?(kind)
@data[kind] = fetch_by_kind(kind)
end
end
def fetch_by_kind(kind)
resource_class = KubernetesResource.class_for_kind(kind)
output_is_sensitive = resource_class.nil? ? false : resource_class::SENSITIVE_TEMPLATE_CONTENT
raw_json, _, st = @kubectl.run("get", kind, "--chunk-size=0", attempts: 5, output: "json",
output_is_sensitive: output_is_sensitive)
raise KubectlError unless st.success?
instances = {}
JSON.parse(raw_json)["items"].each do |resource|
resource_name = resource.dig("metadata", "name")
instances[resource_name] = resource
end
instances
end
end
end
Version data entries
7 entries across 7 versions & 1 rubygems