require 'avro_turf/schema_registry'

# Caches registrations and lookups to the schema registry in memory.
class AvroTurf::CachedSchemaRegistry

  def initialize(upstream)
    @upstream = upstream
    @schemas_by_id = {}
    @ids_by_schema = {}
  end

  # Delegate the following methods to the upstream
  %i(subjects subject_versions subject_version check compatible?).each do |name|
    define_method(name) do |*args|
      instance_variable_get(:@upstream).send(name, *args)
    end
  end

  def fetch(id)
    @schemas_by_id[id] ||= @upstream.fetch(id)
  end

  def register(subject, schema)
    @ids_by_schema[subject + schema.to_s] ||= @upstream.register(subject, schema)
  end
end