Class: Gimei::UniqueGenerator

Inherits:
Object
  • Object
show all
Defined in:
lib/gimei/unique_generator.rb

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(klass, max_retries) ⇒ UniqueGenerator

Returns a new instance of UniqueGenerator.



11
12
13
14
 
# File 'lib/gimei/unique_generator.rb', line 11

def initialize(klass, max_retries)
  @klass = klass
  @max_retries = max_retries
end

Class Method Details

.previous_results ⇒ Object 



6
7
8
 
# File 'lib/gimei/unique_generator.rb', line 6

def previous_results
  @previous_results ||= Hash.new { |hash, key| hash[key] = Set.new }
end

Instance Method Details

#clear(key = nil) ⇒ Object 



20
21
22
23
24
25
26
 
# File 'lib/gimei/unique_generator.rb', line 20

def clear(key = nil)
  if key
    previous_results[key.to_sym].clear
  else
    previous_results.clear
  end
end

#define_unique_method(method_name, previous_result_key = method_name) ⇒ Object 



28
29
30
31
32
33
34
35
36
37
38
39
40
41
 
# File 'lib/gimei/unique_generator.rb', line 28

def define_unique_method(method_name, previous_result_key = method_name)
  define_singleton_method method_name do |*args|
    max_retries.times do
      result = klass.public_send(method_name, *args)

      next if previous_results[previous_result_key].include?(result.to_s)

      previous_results[previous_result_key] << result.to_s
      return result
    end

    raise RetryLimitExceeded, "Retry limit exceeded for #{method_name}"
  end
end

#previous_results ⇒ Object 



16
17
18
 
# File 'lib/gimei/unique_generator.rb', line 16

def previous_results
  self.class.previous_results
end