Class: Avromatic::ModelRegistry

Inherits:
Object
  • Object
show all
Defined in:
lib/avromatic/model_registry.rb

Overview

The ModelRegistry class is used to store and fetch nested models by their fullname. An optional namespace prefix can be removed from the full name that is used to store and fetch models.

Instance Method Summary collapse

Constructor Details

#initialize(remove_namespace_prefix: nil) ⇒ ModelRegistry

Returns a new instance of ModelRegistry.



11
12
13
14
 
# File 'lib/avromatic/model_registry.rb', line 11

def initialize(remove_namespace_prefix: nil)
  @prefix = remove_namespace_prefix
  @hash = Hash.new
end

Instance Method Details

#[](fullname) ⇒ Object Also known as: fetch 



20
21
22
 
# File 'lib/avromatic/model_registry.rb', line 20

def [](fullname)
  @hash.fetch(fullname)
end

#clear ⇒ Object 



16
17
18
 
# File 'lib/avromatic/model_registry.rb', line 16

def clear
  @hash.clear
end

#ensure_registered_model(model) ⇒ Object 



42
43
44
45
46
47
48
49
50
51
 
# File 'lib/avromatic/model_registry.rb', line 42

def ensure_registered_model(model)
  name = model_fullname(model)
  if registered?(name)
    unless fetch(name).equal?(model)
      raise "attempted to replace existing model #{fetch(name)} with new model #{model} as '#{name}'"
    end
  else
    register(model)
  end
end

#model_fullname(model) ⇒ Object 



37
38
39
40
 
# File 'lib/avromatic/model_registry.rb', line 37

def model_fullname(model)
  name = model.avro_schema.fullname
  remove_prefix(name)
end

#register(model) ⇒ Object 



25
26
27
28
29
30
 
# File 'lib/avromatic/model_registry.rb', line 25

def register(model)
  raise 'models with a key schema are not supported' if model.key_avro_schema
  name = model_fullname(model)
  raise "'#{name}' has already been registered" if registered?(name)
  @hash[name] = model
end

#registered?(fullname_or_model) ⇒ Boolean

Returns:

  • (Boolean) 


32
33
34
35
 
# File 'lib/avromatic/model_registry.rb', line 32

def registered?(fullname_or_model)
  fullname = fullname_or_model.is_a?(String) ? fullname_or_model : model_fullname(fullname_or_model)
  @hash.key?(fullname)
end

#remove_prefix(name) ⇒ Object 



53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
 
# File 'lib/avromatic/model_registry.rb', line 53

def remove_prefix(name)
  return name if @prefix.nil?

  value =
    case @prefix
    when String
      name.start_with?(@prefix) ? name.from(@prefix.length) : name
    when Regexp
      name.sub(@prefix, '')
    else
      raise "unsupported `remove_namespace_prefix` value: #{@prefix}"
    end

  value.start_with?('.') ? value.from(1) : value
end