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
|
# frozen_string_literal: true
$LOAD_PATH.unshift File.expand_path('../../../lib', __dir__)
task driver_bench: %i[ driver_bench:data driver_bench:run ]
SPECS_REPO_URI = 'https://github.com/mongodb/specifications'
SPECS_PATH = File.expand_path('../../../specifications', __dir__)
DRIVER_BENCH_DATA = File.expand_path('../../data/driver_bench', __dir__)
# rubocop:disable Metrics/BlockLength
namespace :driver_bench do
desc 'Downloads the DriverBench data files, if necessary'
task :data do
if File.directory?('./profile/data/driver_bench')
puts 'DriverBench data files are already downloaded'
next
end
if File.directory?(SPECS_PATH)
puts 'specifications repo is already checked out'
else
sh 'git', 'clone', SPECS_REPO_URI
end
mkdir_p DRIVER_BENCH_DATA
Dir.glob(File.join(SPECS_PATH, 'source/benchmarking/data/*.tgz')) do |archive|
Dir.chdir(DRIVER_BENCH_DATA) do
sh 'tar', 'xzf', archive
end
end
end
desc 'Runs the DriverBench benchmark suite'
task :run do
require_relative '../suite'
Mongo::DriverBench::Suite.run!
end
desc 'Runs the crypto benchmark'
task :crypto do
require_relative '../crypto/decrypt'
Mongo::DriverBench::Crypto::Decrypt.new.run
end
end
# rubocop:enable Metrics/BlockLength
|
