1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
|
# frozen_string_literal: true
# rubocop:todo all
module Unified
class EntityMap
extend Forwardable
def initialize
@map = {}
end
def set(type, id, value)
@map[type] ||= {}
if @map[type][id]
raise Error::EntityMapOverwriteAttempt,
"Cannot set #{type} #{id} because it is already defined"
end
@map[type][id] = value
end
def get(type, id)
unless @map[type]
raise Error::EntityMissing, "There are no #{type} entities known"
end
unless v = @map[type][id]
raise Error::EntityMissing, "There is no #{type} #{id} known"
end
v
end
def get_any(id)
@map.each do |type, sub|
if sub[id]
return sub[id]
end
end
raise Error::EntityMissing, "There is no #{id} known"
end
def_delegators :@map, :[], :fetch
end
end
|
