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
|
# frozen_string_literal: true
# rubocop:todo all
require 'singleton'
class LocalResourceRegistry
include Singleton
def initialize
@resources = []
end
def register(resource, finalizer)
@resources << [resource, finalizer]
# Return resource for chaining
resource
end
def unregister(resource)
@resources.delete_if do |_resource, finalizer|
_resource == resource
end
end
def close_all
@resources.each do |resource, finalizer|
if finalizer.is_a?(Symbol)
resource.send(finalizer)
elsif finalizer.is_a?(Proc)
finalizer.call(resource)
else
raise "Unknown finalizer: #{finalizer}"
end
end
@resources = []
end
end
|
