summaryrefslogtreecommitdiff
path: root/lib/drb/timeridconv.rb
blob: 6d8935b1efbc15b749afebe2ed15750376074861 (plain)
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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
require 'drb/drb'
require 'monitor'

module DRb
  class TimerIdConv < DRbIdConv
    class TimerHolder2
      include MonitorMixin

      class InvalidIndexError < RuntimeError; end

      def initialize(timeout=600)
	super()
	@sentinel = Object.new
	@gc = {}
	@curr = {}
	@renew = {}
	@timeout = timeout
	@keeper = keeper
      end

      def add(obj)
	synchronize do
	  key = obj.__id__
	  @curr[key] = obj
	  return key
	end
      end

      def fetch(key, dv=@sentinel)
	synchronize do
	  obj = peek(key)
	  if obj == @sentinel
	    return dv unless dv == @sentinel
	    raise InvalidIndexError
	  end
	  @renew[key] = obj # KeepIt
	  return obj
	end
      end

      def include?(key)
	synchronize do
	  obj = peek(key)
	  return false if obj == @sentinel
	  true
	end
      end

      def peek(key)
	synchronize do
	  return @curr.fetch(key, @renew.fetch(key, @gc.fetch(key, @sentinel)))
	end
      end

      private
      def alternate
	synchronize do
	  @gc = @curr       # GCed
	  @curr = @renew
	  @renew = {}
	end
      end

      def keeper
	Thread.new do
	  loop do
	    size = alternate
	    sleep(@timeout)
	  end
	end
      end
    end

    def initialize(timeout=600)
      @holder = TimerHolder2.new(timeout)
    end

    def to_obj(ref)
      return super if ref.nil?
      @holder.fetch(ref)
    rescue TimerHolder2::InvalidIndexError
      raise "invalid reference"
    end

    def to_id(obj)
      return @holder.add(obj)
    end
  end
end

# DRb.install_id_conv(TimerIdConv.new)