summaryrefslogtreecommitdiffstats
path: root/sample/drb/http0.rb
blob: 7649925282a7d34da32f96e398c22c9e43ce1047 (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
require 'drb/drb'
require 'net/http'
require 'uri'

module DRb
  module HTTP0
    class StrStream
      def initialize(str='')
	@buf = str
      end
      attr_reader :buf

      def read(n)
	begin
	  return @buf[0,n]
	ensure
	  @buf[0,n] = ''
	end
      end

      def write(s)
	@buf.concat s
      end
    end

    def self.uri_option(uri, config)
      return uri, nil
    end

    def self.open(uri, config)
      unless /^http:/ =~ uri
	raise(DRbBadScheme, uri) unless uri =~ /^http:/
	raise(DRbBadURI, 'can\'t parse uri:' + uri)
      end
      ClientSide.new(uri, config)
    end

    class ClientSide
      def initialize(uri, config)
	@uri = uri
	@res = nil
	@config = config
	@msg = DRbMessage.new(config)
	@proxy = ENV['HTTP_PROXY']
      end

      def close; end
      def alive?; false; end
      
      def send_request(ref, msg_id, *arg, &b)
	stream = StrStream.new
	@msg.send_request(stream, ref, msg_id, *arg, &b)
	@reply_stream = StrStream.new
	post(@uri, stream.buf)
      end
      
      def recv_reply
	@msg.recv_reply(@reply_stream)
      end

      def post(url, data)
	it = URI.parse(url)
	path = [(it.path=='' ? '/' : it.path), it.query].compact.join('?')
	http = Net::HTTP.new(it.host, it.port)
	sio = StrStream.new
	http.post(path, data, {'Content-Type'=>'application/octetstream;'}) do |str|
	  sio.write(str)
	  if @config[:load_limit] < sio.buf.size
	    raise TypeError, 'too large packet' 
	  end
	end
	@reply_stream = sio
      end
    end
  end
  DRbProtocol.add_protocol(HTTP0)
end