Class: CloneKit::SharedIdMap

Inherits:
Object
  • Object
show all
Defined in:
lib/clone_kit/shared_id_map.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(namespace, redis: Redis.new) ⇒ SharedIdMap

Returns a new instance of SharedIdMap.



10
11
12
13
 
# File 'lib/clone_kit/shared_id_map.rb', line 10

def initialize(namespace, redis: Redis.new)
  self.namespace = namespace
  self.redis = redis
end

Instance Attribute Details

#namespace ⇒ Object

Returns the value of attribute namespace.



8
9
10
 
# File 'lib/clone_kit/shared_id_map.rb', line 8

def namespace
  @namespace
end

Instance Method Details

#hash_key(klass) ⇒ Object 



46
47
48
49
 
# File 'lib/clone_kit/shared_id_map.rb', line 46

def hash_key(klass)
  klass = klass.name if klass.is_a?(Class)
  "clone_kit_id_map/#{namespace}/#{klass}"
end

#insert(klass, original_id, new_id) ⇒ Object 



30
31
32
 
# File 'lib/clone_kit/shared_id_map.rb', line 30

def insert(klass, original_id, new_id)
  redis.hset(hash_key(klass), original_id.to_s, new_id.to_s)
end

#insert_many(klass, hash) ⇒ Object 



34
35
36
37
38
39
40
 
# File 'lib/clone_kit/shared_id_map.rb', line 34

def insert_many(klass, hash)
  redis.pipelined do
    hash.each do |k, v|
      insert(klass, k, v)
    end
  end
end

#lookup(klass, original_id, id_generator: IdGenerators::Bson) ⇒ Object 



15
16
17
18
19
 
# File 'lib/clone_kit/shared_id_map.rb', line 15

def lookup(klass, original_id, id_generator: IdGenerators::Bson)
  id_generator.from_string(redis.hget(hash_key(klass), original_id.to_s))
rescue IdGenerators::InvalidId
  raise ArgumentError, "No mapping found for #{klass}. This usually indicates a dependency has not be specified"
end

#lookup_safe(klass, original_id, default = nil, id_generator: IdGenerators::Bson) ⇒ Object 



21
22
23
24
25
26
27
28
 
# File 'lib/clone_kit/shared_id_map.rb', line 21

def lookup_safe(klass, original_id, default = nil, id_generator: IdGenerators::Bson)
  val = redis.hget(hash_key(klass), original_id.to_s)
  if val.blank?
    default
  else
    id_generator.from_string(val)
  end
end

#mapping(klass) ⇒ Object 



42
43
44
 
# File 'lib/clone_kit/shared_id_map.rb', line 42

def mapping(klass)
  Hash[redis.hgetall(hash_key(klass)).map { |k, v| [k, v] }]
end