Initial commit

lib/tasks/benchmark.rake

@@ -0,0 +1,72 @@
+namespace :benchmark do
+
+  desc "In memory matching engine benchmark"
+  task :matching => %w(environment) do
+    max_round = round(2)
+    puts "\n>> Setup environment (num=#{num} round=#{max_round})"
+    Dir[Rails.root.join('tmp', 'matching_result_*')].each {|f| FileUtils.rm(f) }
+
+    Benchmark::Matching.new(label, num, max_round).run
+  end
+
+  desc "Trade execution benchmark"
+  task :execution => %w(environment) do
+    max_round = round(2)
+    puts "\n>> Setup environment (executor=#{executor} num=#{num} round=#{max_round})"
+    Dir[Rails.root.join('tmp', 'matching_result_*')].each {|f| FileUtils.rm(f) }
+
+    Benchmark::Execution.new(label, num, max_round, executor).run
+  end
+
+  desc "Run integration benchmark"
+  task :integration => %w(environment) do
+    puts "Integration Benchmark (num: #{num(400)})\n"
+
+    Benchmark::Integration.new(num).run
+  end
+
+  desc "Profiling"
+  task :profiling,  [:type]=> %w(environment) do |task, args|
+
+    case args[:type]
+
+    when 'matching'
+      puts "\n>> Setup environment (num=#{num} round=#{round})"
+      Dir[Rails.root.join('tmp', 'profiling_matching_result_*')].each {|f| FileUtils.rm(f) }
+      file_path = Rails.root.join('tmp', "profiling_matching_result_#{Time.now.to_i}")
+
+      File.open(file_path, 'w') { |file| file.puts "\n>> Setup environment (num=#{num} round=#{round})" }
+
+      Benchmark::Profiling.matching(label, num, round, file_path)
+
+    when 'execution'
+      puts "\n>> Setup environment (executor=#{executor} num=#{num} round=#{round})"
+      Dir[Rails.root.join('tmp', 'profiling_execution_result_*')].each {|f| FileUtils.rm(f) }
+
+      file_path = Rails.root.join('tmp', "profiling_execution_result_#{Time.now.to_i}")
+      File.open(file_path, 'w') { |file| file.puts "\n>> Setup environment (executor=#{executor} num=#{num} round=#{round})" }
+
+      Benchmark::Profiling.execution(label, num, round, executor, file_path)
+
+    else
+      puts "\n>> Wrong parameter!"
+    end
+  end
+
+  def num
+    ENV['NUM'] ? ENV['NUM'].to_i : 100
+  end
+
+  def round r = 1
+    ENV['ROUND'] ? ENV['ROUND'].to_i : r
+  end
+
+  def label
+    ENV['LABEL'] || Time.now.to_i
+  end
+
+  def executor
+    ENV['EXECUTOR'] ? ENV['EXECUTOR'].to_i : 6
+  end
+end
+