compsci-industry
/
TechEmpower.FrameworkBenchmarks
mirror of https://github.com/TechEmpower/FrameworkBenchmarks.git


			
				
					
						
						
							123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125
							# frozen_string_literal: true

# Our Rack application to be executed by rackup
class HelloWorld
  DEFAULT_HEADERS = {}.tap do |h|
    h['Server'] = SERVER_STRING if SERVER_STRING
  end.freeze

  def bounded_queries(env)
    params = Rack::Utils.parse_query(env['QUERY_STRING'])

    queries = params['queries'].to_i
    return QUERIES_MIN if queries < QUERIES_MIN
    return QUERIES_MAX if queries > QUERIES_MAX
    queries
  end

  # Return a random number between 1 and MAX_PK
  def rand1
    Random.rand(MAX_PK).succ
  end

  # Return an array of `n' unique random numbers between 1 and MAX_PK
  def randn(n)
    (1..MAX_PK).to_a.shuffle!.take(n)
  end

  def db
    World.with_pk(rand1).values
  end

  def queries(env)
    # Benchmark requirements explicitly forbid a WHERE..IN here, so be good
    randn(bounded_queries(env))
      .map! { |id| World.with_pk(id).values }
  end

  def fortunes
    fortunes = Fortune.all
    fortunes << Fortune.new(
      :id=>0,
      :message=>'Additional fortune added at request time.'
    )
    fortunes.sort_by!(&:message)

    html = <<~'HTML'
      <!DOCTYPE html>
      <html>
      <head>
        <title>Fortunes</title>
      </head>

      <body>

      <table>
      <tr>
        <th>id</th>
        <th>message</th>
      </tr>
    HTML

    fortunes.each do |fortune|
      html += <<~"HTML"
      <tr>
        <td>#{Rack::Utils.escape_html(fortune.id)}</td>
        <td>#{Rack::Utils.escape_html(fortune.message)}</td>
      </tr>
      HTML
    end

    html += <<~'HTML'
      </table>

      </body>
      </html>
    HTML
  end

  WORLD_BY_ID_FOR_UPDATE = World.naked.for_update.where(:id=>:$id).prepare(:first, :world_by_id_for_update)
  WORLD_UPDATE = World.where(:id=>:$id).prepare(:update, :world_update, :randomnumber=>:$randomnumber)

  def updates(env)
    # Benchmark requirements explicitly forbid a WHERE..IN here, transactions
    # are optional, batch updates are allowed (but each transaction can only
    # read and write a single record?), so... be good
    randn(bounded_queries(env)).map! do |id|
      DB.transaction do
        world = WORLD_BY_ID_FOR_UPDATE.(:id=>id)
        WORLD_UPDATE.(:id=>id, :randomnumber=>(world[:randomnumber] = rand1))
        world
      end
    end
  end

  def call(env)
    content_type, *body =
      case env['PATH_INFO']
      when '/json'
        # Test type 1: JSON serialization
        ['application/json', JSON.fast_generate(:message=>'Hello, World!')]
      when '/db'
        # Test type 2: Single database query
        ['application/json', JSON.fast_generate(db)]
      when '/queries'
        # Test type 3: Multiple database queries
        ['application/json', JSON.fast_generate(queries(env))]
      when '/fortunes'
        # Test type 4: Fortunes
        ['text/html; charset=utf-8', fortunes]
      when '/updates'
        # Test type 5: Database updates
        ['application/json', JSON.fast_generate(updates(env))]
      when '/plaintext'
        # Test type 6: Plaintext
        ['text/plain', 'Hello, World!']
      end

    return 200,
      DEFAULT_HEADERS.merge(
        'Content-Type'=>content_type,
        'Date'=>Time.now.httpdate
      ),
      body
  end
end