require 'benchmark'
require 'helper/all'
require 'proto/test_service_impl'

# Including a way to turn on debug logger for spec runs
if ENV["DEBUG"]
  puts 'debugging'
  debug_log = File.expand_path('../debug_bench.log', File.dirname(__FILE__) )
  Protobuf::Logger.configure(:file => debug_log, :level => ::Logger::DEBUG)
end

namespace :benchmark do
  include SilentConstants

  def benchmark_wrapper(global_bench = nil)
    if global_bench
      yield global_bench
    else
      Benchmark.bm(10) do |bench|
        yield bench
      end
    end
  end

  def em_client_em_server(number_tests, test_length, global_bench = nil)
    EM.stop if EM.reactor_running?

    EventMachine.fiber_run do 
      StubServer.new do |server|
        client = Spec::Proto::TestService.client(:async => false)

        benchmark_wrapper(global_bench) do |bench|
          bench.report("ES / EC") do 
            (1..number_tests.to_i).each { client.find(:name => "Test Name" * test_length.to_i, :active => true) }
          end
        end
      end

      EM.stop
    end
  end

  def em_client_sock_server(number_tests, test_length, global_bench = nil)
    EM.stop if EM.reactor_running?

    EventMachine.fiber_run do
      StubServer.new(:server => Protobuf::Rpc::Socket::Server, :port => 9399) do |server| 
        client = Spec::Proto::TestService.client(:async => false, :port => 9399)

        benchmark_wrapper(global_bench) do |bench|
          bench.report("SS / EC") do 
            (1..number_tests.to_i).each { client.find(:name => "Test Name" * test_length.to_i, :active => true) }
          end
        end
      end

      EM.stop
    end
  end

  def sock_client_sock_server(number_tests, test_length, global_bench = nil)
    EM.stop if EM.reactor_running?

    StubServer.new(:server => Protobuf::Rpc::Socket::Server, :port => 9399) do |server| 
      with_constants "Protobuf::ClientType" => "Socket" do
        client = Spec::Proto::TestService.client(:async => false, :port => 9399)

        benchmark_wrapper(global_bench) do |bench|
          bench.report("SS / SC") do 
            (1..number_tests.to_i).each { client.find(:name => "Test Name" * test_length.to_i, :active => true) }
          end
        end
      end
    end
  end

  def sock_client_em_server(number_tests, test_length, global_bench = nil)
    EM.stop if EM.reactor_running?
    em_thread = Thread.new { EM.run }
    Thread.pass until EM.reactor_running?

    StubServer.new(:port => 9399) do |server| 
      with_constants "Protobuf::ClientType" => "Socket" do
        client = Spec::Proto::TestService.client(:async => false, :port => 9399)

        benchmark_wrapper(global_bench) do |bench|
          bench.report("ES / SC") do 
            (1..number_tests.to_i).each { client.find(:name => "Test Name" * test_length.to_i, :active => true) }
          end
        end
      end
    end

    EM.stop
    Thread.kill(em_thread) if em_thread
  end

  def zmq_client_zmq_server(number_tests, test_length, global_bench = nil)
    require 'ffi-rzmq'
    require 'protobuf/rpc/connectors/zmq'
    require 'protobuf/rpc/servers/zmq/server'
    StubServer.new(:port => 9399, :server => Protobuf::Rpc::Zmq::Server) do |server| 
      with_constants "Protobuf::ClientType" => "Zmq" do
        client = Spec::Proto::TestService.client(:async => false, :port => 9399)

        benchmark_wrapper(global_bench) do |bench|
          bench.report("ZS / ZC") do 
            (1..number_tests.to_i).each { client.find(:name => "Test Name" * test_length.to_i, :active => true) }
          end
        end
      end
      server.stop
    end
  end

  desc "benchmark ZMQ client with ZMQ server"
  task :zmq_client_zmq_server, [:number, :length] do |t, args|
    args.with_defaults(:number => 1000, :length => 100)
    zmq_client_zmq_server(args[:number], args[:length])
  end

  desc "benchmark EventMachine client with EventMachine server"
  task :em_client_em_server, [:number, :length] do |t, args|
    args.with_defaults(:number => 1000, :length => 100)
    em_client_em_server(args[:number], args[:length])
  end

  desc "benchmark EventMachine client with Socket server"
  task :em_client_sock_server, [:number, :length] do |t, args|
    args.with_defaults(:number => 1000, :length => 100)
    em_client_sock_server(args[:number], args[:length])
  end

  desc "benchmark Socket client with Socket server"
  task :sock_client_sock_server, [:number, :length] do |t, args|
    args.with_defaults(:number => 1000, :length => 100)
    sock_client_sock_server(args[:number], args[:length])
  end

  desc "benchmark Socket client with EventMachine server"
  task :sock_client_em_server, [:number, :length] do |t, args|
    args.with_defaults(:number => 1000, :length => 100)
    sock_client_em_server(args[:number], args[:length])
  end

  desc "benchmark server performance"
  task :servers, [:number, :length] do |t, args|
    args.with_defaults(:number => 1000, :length => 100)

    Benchmark.bm(10) do |bench|
      # zmq_client_zmq_server(args[:number], args[:length], bench)
      #      em_client_em_server(args[:number], args[:length], bench)
      #      em_client_sock_server(args[:number], args[:length], bench)
      sock_client_sock_server(args[:number], args[:length], bench)
      #      sock_client_em_server(args[:number], args[:length], bench)
    end
  end
end