Sha256: c4c8fc7412b244ab7478260845fddbcce2582045ef188071bf2b5c5e654b7724

Contents?: true

Size: 1.4 KB

Versions: 10

Compression:

Stored size: 1.4 KB

Contents

desc "Benchmark the get_treatment method call in 4 threads"

# Usage:
# rake concurrent_benchmark api_key=YOUR_API_KEY base_uri=YOUR_API_BASE_URI [iterations=NUMBER_OF_ITERATIONS] [user_id=A_USER_ID] [feature_id=A_FEATURE_ID]
task :concurrent_benchmark do
  require 'benchmark'
  require 'splitclient-rb'

  usage_message = "Usage: rake concurrent_benchmark api_key=YOUR_API_KEY base_uri=YOUR_API_BASE_URI [iterations=NUMBER_OF_ITERATIONS] [user_id=A_USER_ID] [feature_id=A_FEATURE_ID]"
  valid_params = validate_params
  if validate_params
    execute
  else
    p usage_message
  end
end

def validate_params
  !ENV['api_key'].nil? && !ENV['base_uri'].nil?
end

def execute
  api_key = ENV['api_key'].nil? ? "fake_api_key" : ENV['api_key']
  base_uri = ENV['base_uri'].nil? ? "fake/api/" : ENV['base_uri']
  iterations = ENV['iterations'].nil? ? 1000000 : ENV['iterations'].to_i
  user_id = ENV['user_id'].nil? ? "fake_id_1" : ENV['user_id']
  feature_id = ENV['feature_id'].nil? ? "sample_feature" : ENV['feature_id']

  threads = []
  times_per_thread = iterations / 4
  split_client = SplitIoClient::SplitClient.new(api_key, {base_uri: base_uri, logger: Logger.new("/dev/null") })

  puts Benchmark.measure{
    4.times do |i|
      threads << Thread.new do
        times_per_thread.times do
          split_client.get_treatment user_id, feature_id, {attr: 123}
        end
      end
    end
    threads.map(&:join)
  }
end

Version data entries

10 entries across 10 versions & 1 rubygems

Version Path
splitclient-rb-1.0.4 tasks/benchmark_get_treatment.rake
splitclient-rb-1.0.4.wip3 tasks/benchmark_get_treatment.rake
splitclient-rb-1.0.4.wip2 tasks/benchmark_get_treatment.rake
splitclient-rb-1.0.4.wip tasks/benchmark_get_treatment.rake
splitclient-rb-1.0.3 tasks/benchmark_get_treatment.rake
splitclient-rb-1.0.3.wip2 tasks/benchmark_get_treatment.rake
splitclient-rb-1.0.3.wip1 tasks/benchmark_get_treatment.rake
splitclient-rb-1.0.2 tasks/benchmark_get_treatment.rake
splitclient-rb-1.0.2.wip2 tasks/benchmark_get_treatment.rake
splitclient-rb-1.0.2.wip tasks/benchmark_get_treatment.rake