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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
|
# frozen_string_literal: true
require 'set'
module FFaker
class UniqueUtils
RetryLimitExceeded = Class.new(StandardError)
class << self
def add_instance(generator, max_retries)
instances[generator] ||= FFaker::UniqueUtils.new(generator, max_retries)
end
def instances
Thread.current[:ffaker_unique_utils] ||= {}
end
def clear
instances.each_value(&:clear)
instances.clear
end
end
def initialize(generator, max_retries)
@generator = generator
@max_retries = max_retries
end
def clear
previous_results.clear
end
private
def method_missing(name, *args, **kwargs)
@max_retries.times do
result = @generator.public_send(name, *args, **kwargs)
next if previous_results[[name, args, kwargs]].include?(result)
previous_results[[name, args, kwargs]] << result
return result
end
raise RetryLimitExceeded, "Retry limit exceeded for #{name}"
end
def respond_to_missing?(name, *args)
@generator.respond_to?(name, *args) || super
end
def previous_results
@previous_results ||= Hash.new { |hash, key| hash[key] = Set.new }
end
end
end
|
