# frozen_string_literal: true

require_relative "setup"

driver = ENV.fetch("DRIVER", "ruby").to_sym
redis_client = RedisClient.new(host: "localhost", port: Servers::REDIS.real_port, driver: driver)
redis = Redis.new(host: "localhost", port: Servers::REDIS.real_port, driver: driver)

redis_client.call("SET", "key", "value")
redis_client.call("SET", "large", "value" * 10_000)
redis_client.call("LPUSH", "list", *5.times.to_a)
redis_client.call("LPUSH", "large-list", *1000.times.to_a)
redis_client.call("HMSET", "hash", *8.times.to_a)
redis_client.call("HMSET", "large-hash", *1000.times.to_a)

benchmark("small string") do |x|
  x.report("redis-rb") { redis.get("key") }
  x.report("redis-client") { redis_client.call("GET", "key") }
end

benchmark("large string") do |x|
  x.report("redis-rb") { redis.get("large").valid_encoding? }
  x.report("redis-client") { redis_client.call("GET", "large").valid_encoding? }
end

benchmark("small list") do |x|
  x.report("redis-rb") { redis.lrange("list", 0, -1) }
  x.report("redis-client") { redis_client.call("LRANGE", "list", 0, -1) }
end

benchmark("large list") do |x|
  x.report("redis-rb") { redis.lrange("large-list", 0, -1) }
  x.report("redis-client") { redis_client.call("LRANGE", "large-list", 0, -1) }
end

benchmark("small hash") do |x|
  x.report("redis-rb") { redis.hgetall("hash") }
  x.report("redis-client") { redis_client.call("HGETALL", "hash") }
end

benchmark("large hash") do |x|
  x.report("redis-rb") { redis.hgetall("large-hash") }
  x.report("redis-client") { redis_client.call("HGETALL", "large-hash") }
end