bdainton / xmpp_talk

Code samples supporting an XMPP talk I have given.

This URL has Read+Write access

Additional disclaimer about xmpp4r-simple fork not being complete yet.
bdainton (author)
Fri Sep 26 15:40:40 -0700 2008
commit  65f9e04bde85445a2dc2d9b373a9e64d158cfd93
tree    5632165c4347314048fe908bbb6a85e15679226b
parent  5e86e68613dadeb42d7d3af147b52ad5fdefdd4b
xmpp_talk / traffic / traffic.rb
100755 67 lines (57 sloc) 1.494 kb
raw blame history 
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

  
class TrafficGenerator
  
  def run(args)
    if args.size < 2
      print_usage
      return
    end
    
    case args[1]
    when 'twitter'
      print_twitter_traffic(args[0])
    when 'flickr'
      print_flickr_traffic(args[0])
    end
  end
  
  #######
  private
  #######
  
  def print_usage
    puts
    puts "USAGE: ruby traffic.rb [post | get] [twitter | flickr]"
  end
  
  def print_twitter_traffic(http_method)
    case http_method
    when 'post'
      print_traffic(10, "[TWITTER] Twitter status (tweet) created via POST")
    when 'get'
      print_traffic(200, "[TWITTER] Twitter polling: API request made via GET")
    else
      print_usage
    end
  end
  
  def print_flickr_traffic(http_method)
    case http_method
    when 'post'
      print_traffic(60, "[FLICKR] Flickr image uploaded via POST")
    when 'get'
      print_traffic(200, "[FLICKR] Flickr polling: API request made via GET")
    else
      print_usage
    end  
  end
  
  def print_traffic(num_per_second, text)
    interval = 1.0 / num_per_second
    num_printed = 0
    target = num_per_second * 5
    loop do
      num_printed += 1
      puts "#{text} #{num_printed}"
      break if num_printed >= target
      sleep interval
    end
    puts
    puts "---------------------------------------------------------------"
    puts "TOTAL #{num_printed} requests in 5 seconds"
    puts "---------------------------------------------------------------"    
  end
  
end
 
TrafficGenerator.new.run(ARGV)