Stoplight: ZSET Bucket Size Benchmark

analysis.ipynb

redis_bench.rb

# frozen_string_literal: true

require "benchmark/ips"
require "stoplight"
require "redis"
require "timecop"

redis = Redis.new
data_store = Stoplight::DataStore::Redis.new(redis)

# In your benchmark file
BUCKET_SIZES = [60, 300, 600, 900, 1800, 3600].freeze  # 1m to 1h
WINDOW_SIZES = (300..3600).step(300).to_a.freeze       # 5m to 1h

results = {}

BUCKET_SIZES.each do |bucket_size_value|
  puts "=== Testing bucket_size: #{bucket_size_value}s ==="

  WINDOW_SIZES.each do |window_size|
    puts "Testing window_size: #{window_size}s"

    # Configure your ZSET implementation with this bucket size
    # (You might need to make bucket size configurable)

    stoplight = Stoplight(SecureRandom.uuid, window_size: window_size, data_store: data_store)

    Stoplight::DataStore::Redis.singleton_class.class_eval(<<~RUBY)
      def bucket_size
        #{bucket_size_value}
      end
    RUBY

    start_time = Time.at(1747129162) - window_size

    result = Benchmark.ips do |b|
      b.report("#{bucket_size_value}s buckets, #{window_size}s window") do
        Timecop.freeze(start_time + rand(window_size * 2)) do
          stoplight.run {}
        end
      end
    end

    results[[bucket_size_value, window_size]] = result.data.dig(0, :ips)
  end
end

require "json"

File.write("zset_performance_matrix.json", results.to_json)