forked from sga001/pir-test
-
Notifications
You must be signed in to change notification settings - Fork 0
/
scp_master.rb
executable file
·59 lines (44 loc) · 1.23 KB
/
scp_master.rb
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
#!/usr/bin/env ruby
require 'optparse'
require 'json'
options = {:server_addr => "127.0.0.1", :exp => "dc", :n => 2**16,
:c => 1, :i => 10}
OptionParser.new do |opts|
opts.banner = "Usage: scp_master.rb [options]"
opts.on("-h", "--host ADDR", "specify server addr") do |v|
options[:server_addr] = v
end
opts.on("-e", "--exp EXP", "specify experiment") do |v|
options[:exp] = v
end
opts.on("-n", "--num NUM", "specify exponent (2^num)") do |v|
options[:n] = 2**Integer(v)
end
opts.on("-c", "--threads THREADS", "number of client threads") do |v|
options[:c] = Integer(v)
end
opts.on("-i", "--iterations ITER", "number of iterations") do |v|
options[:i] = Integer(v)
end
end.parse!
n = options[:n]
server_addr = options[:server_addr]
exp = options[:exp]
c = options[:c]
i = options[:i]
if exp == "tput"
exp = "tpu_#{c}"
end
date = Time.now.utc
results = []
(0...i).each do |it|
start = Time.now
%x(scp #{server_addr}:/home/pir-test/scp_files/#{n} ./)
finish = Time.now
results.push((finish-start) * 1000000) # microseconds
%x(rm #{n})
end
json_obj = {"date" => date, "num" => n, "i" => i, "point" => results}
f = File.open("#{exp}_scp.log", "a")
f.puts(json_obj.to_json())
f.close()