# frozen_string_literal: true
require 'bundler/setup'

SEQUEL_NO_ASSOCIATIONS = true

Bundler.require(:default) # Load core modules

def connect(dbtype)
  Bundler.require(dbtype) # Load database-specific modules

  opts = {}

  if dbtype == :mysql
    adapter = 'trilogy'
    opts[:ssl] = true
    opts[:ssl_mode] = 4 # Trilogy::SSL_PREFERRED_NOVERIFY
    opts[:tls_min_version] = 3 # Trilogy::TLS_VERSION_12
  else
    adapter = 'postgresql'
  end

  # Determine threading/thread pool size and timeout
  if defined?(Puma)
    opts[:max_connections] = ENV.fetch('MAX_THREADS')
    opts[:pool_timeout] = 10
  else
    opts[:max_connections] = 512
  end

  Sequel.connect \
    '%{adapter}://%{host}/%{database}?user=%{user}&password=%{password}' % {
      adapter: adapter,
      host: 'tfb-database',
      database: 'hello_world',
      user: 'benchmarkdbuser',
      password: 'benchmarkdbpass'
    }, opts
end

DB = connect ENV.fetch('DBTYPE').to_sym

# Define ORM models
class World < Sequel::Model(:World)
  def_column_alias(:randomnumber, :randomNumber) if DB.database_type == :mysql

  def self.batch_update(worlds)
    if DB.database_type == :mysql
      worlds.map(&:save_changes)
    else
      ids = []
      sql = String.new("UPDATE world SET randomnumber = CASE id ")
      worlds.each do |world|
        sql << "when #{world[:id]} then #{world[:randomnumber]} "
        ids << world[:id]
      end
      sql << "ELSE randomnumber END WHERE id IN ( #{ids.join(',')})"
      DB.run(sql)
    end
  end
end

class Fortune < Sequel::Model(:Fortune)
  # Allow setting id to zero (0) per benchmark requirements
  unrestrict_primary_key
end

[World, Fortune].each(&:freeze)
DB.freeze