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
|
module EventMachine
# EventMachine based Multi request client, based on a streaming HTTPRequest class,
# which allows you to open multiple parallel connections and return only when all
# of them finish. (i.e. ideal for parallelizing workloads)
#
# == Example
#
# EventMachine.run {
#
# multi = EventMachine::MultiRequest.new
#
# # add multiple requests to the multi-handler
# multi.add(:a, EventMachine::HttpRequest.new('http://www.google.com/').get)
# multi.add(:b, EventMachine::HttpRequest.new('http://www.yahoo.com/').get)
#
# multi.callback {
# p multi.responses[:callback]
# p multi.responses[:errback]
#
# EventMachine.stop
# }
# }
#
class MultiRequest
include EventMachine::Deferrable
attr_reader :requests, :responses
def initialize
@requests = {}
@responses = {:callback => {}, :errback => {}}
end
def add(name, conn)
raise 'Duplicate Multi key' if @requests.key? name
@requests[name] = conn
conn.callback { @responses[:callback][name] = conn; check_progress }
conn.errback { @responses[:errback][name] = conn; check_progress }
end
def finished?
(@responses[:callback].size + @responses[:errback].size) == @requests.size
end
protected
# invoke callback if all requests have completed
def check_progress
succeed(self) if finished?
end
end
end
|
