File: weak_instance_cache.rb

package info (click to toggle)
ruby-roo 3.0.0-1
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 8,216 kB
  • sloc: ruby: 6,529; xml: 88; makefile: 6
file content (41 lines) | stat: -rw-r--r-- 1,079 bytes parent folder | download | duplicates (5)
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
# frozen_string_literal: true

require "weakref"

module Roo
  module Helpers
    module WeakInstanceCache
      private

      def instance_cache(key)
        object = nil

        if instance_variable_defined?(key) && (ref = instance_variable_get(key)) && ref.weakref_alive?
          begin
            object = ref.__getobj__
          rescue => e
            unless (defined?(::WeakRef::RefError) && e.is_a?(::WeakRef::RefError)) || (defined?(RefError) && e.is_a?(RefError))
              raise e
            end
          end
        end

        unless object
          object = yield
          ObjectSpace.define_finalizer(object, instance_cache_finalizer(key))
          instance_variable_set(key, WeakRef.new(object))
        end

        object
      end

      def instance_cache_finalizer(key)
        proc do |object_id|
          if instance_variable_defined?(key) && (ref = instance_variable_get(key)) && (!ref.weakref_alive? || ref.__getobj__.object_id == object_id)
            remove_instance_variable(key)
          end
        end
      end
    end
  end
end