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 



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

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

Instance Attribute Details

#namespace ⇒ Object

Returns the value of attribute namespace.



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

def namespace
  @namespace
end

Instance Method Details

#hash_key(klass) ⇒ Object 



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

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 



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

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 



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

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

#lookup(klass, original_id) ⇒ Object 



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

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

#lookup_safe(klass, original_id, default = nil) ⇒ Object 



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

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

#mapping(klass) ⇒ Object 



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

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