60 lines
1.1 KiB
Ruby
60 lines
1.1 KiB
Ruby
# frozen_string_literal: false
|
|
require_relative 'drb'
|
|
require 'monitor'
|
|
|
|
module DRb
|
|
|
|
# To use WeakIdConv:
|
|
#
|
|
# DRb.start_service(nil, nil, {:idconv => DRb::WeakIdConv.new})
|
|
|
|
class WeakIdConv < DRbIdConv
|
|
class WeakSet
|
|
include MonitorMixin
|
|
def initialize
|
|
super()
|
|
@immutable = {}
|
|
@map = ObjectSpace::WeakMap.new
|
|
end
|
|
|
|
def add(obj)
|
|
synchronize do
|
|
begin
|
|
@map[obj] = self
|
|
rescue ArgumentError
|
|
@immutable[obj.__id__] = obj
|
|
end
|
|
return obj.__id__
|
|
end
|
|
end
|
|
|
|
def fetch(ref)
|
|
synchronize do
|
|
@immutable.fetch(ref) {
|
|
@map.each { |key, _|
|
|
return key if key.__id__ == ref
|
|
}
|
|
raise RangeError.new("invalid reference")
|
|
}
|
|
end
|
|
end
|
|
end
|
|
|
|
def initialize()
|
|
super()
|
|
@weak_set = WeakSet.new
|
|
end
|
|
|
|
def to_obj(ref) # :nodoc:
|
|
return super if ref.nil?
|
|
@weak_set.fetch(ref)
|
|
end
|
|
|
|
def to_id(obj) # :nodoc:
|
|
return @weak_set.add(obj)
|
|
end
|
|
end
|
|
end
|
|
|
|
# DRb.install_id_conv(WeakIdConv.new)
|