# frozen_string_literal: true

# Run on Ruby 3.3 with YJIT enabled

require 'bundler/inline'

gemfile do
  source 'https://rubygems.org'
  gem 'extralite', path: '..'
  gem 'benchmark-ips'
end

require 'benchmark/ips'
require 'fileutils'

DB_PATH = "/tmp/extralite_sqlite3_perf-#{Time.now.to_i}-#{rand(10000)}.db"
puts "DB_PATH = #{DB_PATH.inspect}"

$extralite_db = Extralite::Database.new(DB_PATH, gvl_release_threshold: -1)

def prepare_database(count)
  $extralite_db.query('create table if not exists foo ( a integer primary key, b text )')
  $extralite_db.query('delete from foo')
  $extralite_db.query('begin')
  count.times { $extralite_db.query('insert into foo (b) values (?)', "hello#{rand(1000)}" )}
  $extralite_db.query('commit')
end

class Model
  def initialize(h)
    @h = h
  end

  def values
    @h
  end
end

TRANSFORM = ->(h) { Model.new(h) }

def extralite_run_map(count)
  results = $extralite_db.query('select * from foo').map(&TRANSFORM)
  raise unless results.size == count
end

def extralite_run_transform(count)
  results = $extralite_db.query(TRANSFORM, 'select * from foo')
  raise unless results.size == count
end

[10, 1000, 100000].each do |c|
  puts "Record count: #{c}"
  prepare_database(c)

  bm = Benchmark.ips do |x|
    x.config(:time => 5, :warmup => 2)

    x.report("map")       { extralite_run_map(c) }
    x.report("transform") { extralite_run_transform(c) }

    x.compare!
  end
  puts;
  bm.entries.each { |e| puts "#{e.label}: #{(e.ips * c).round.to_i} rows/s" }
  puts;
end