compsci-industry
/
TechEmpower.FrameworkBenchmarks
cermin dari https://github.com/TechEmpower/FrameworkBenchmarks.git


			
				
					
						
						
							123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158
							require "moonshine"
require "pg"
require "html/builder"

include Moonshine
include Moonshine::Utils::Shortcuts
include Moonshine::Base

# Compose Objects (like Hash) to have a to_json method
require "json/to_json"

DB = PG.connect("postgres://benchmarkdbuser:benchmarkdbpass@#{ENV["DBHOST"]? || "127.0.0.1"}/hello_world")
app = App.new

class CONTENT
  UTF8 = "; charset=UTF-8"
  JSON = "application/json" + UTF8
  PLAIN = "text/plain"
  HTML = "text/html" + UTF8
end

ID_MAXIMUM = 10_000

app.response_middleware do |req, res|
    res.headers["Server"] = "Moonshine"
    res.headers["Date"] = Time.now.to_s
    res
end

private def randomWorld
  id = rand(1..ID_MAXIMUM)
  result = DB.exec({Int32, Int32}, "SELECT id, randomNumber FROM world WHERE id = $1", [id]).rows.first
  {:id => result[0], :randomNumber => result[1]}
end

private def setWorld(world)
  DB.exec("UPDATE world set randomNumber = $1 where id = $2", [world[:randomNumber], world[:id]])
  world
end

private def fortunes
  data = [] of  Hash(Symbol, (String | Int32))

  DB.exec({Int32, String}, "select id, message from Fortune").rows.each do |row|
    data.push({:id => row[0], :message => row[1]})
  end
  data
end

private def sanitizedQueryCount(request)
  begin
    queries = request.get["queries"].to_i
  rescue
    queries = 1
  end
  queries = 1 if queries < 1
  queries = 500 if queries > 500
  queries
end

#
# Basic Tests
#

# Test 1: JSON Serialization
app.get "/json", do |request|
  res = ok({ :message => "Hello, World!" }.to_json)
  res.headers["Content-type"] = CONTENT::JSON
  res
end

# Test 6: Plaintext
app.get "/plaintext", do |request|
  res = ok("Hello, World!")
  res.headers["Content-type"] = CONTENT::PLAIN
  res
end

#
# Postgres DatabaseTests
#

# Postgres Test 2: Single database query
app.get "/db", do |request|
  res = ok(randomWorld.to_json)
  res.headers["Content-type"] = CONTENT::JSON
  res
end

# Postgres Test 3: Multiple database query
app.get "/queries", do |request|
  results = (1..sanitizedQueryCount(request)).map do
    randomWorld
  end

  res = ok(results.to_json)
  res.headers["Content-type"] = CONTENT::JSON
  res
end

# Postgres Test 4: Fortunes
app.get "/fortunes", do |request|
  data = fortunes
  
  additional_fortune = {
    :id => 0,
    :message => "Additional fortune added at request time."
  }
  data.push(additional_fortune)

  data.sort! do |a, b|
    a[:message].to_s <=> b[:message].to_s
  end

  # New builder for each request!
  html = HTML::Builder.new.build do
    html {
      head {
        title { text "Fortunes" }
      }
      body {
        table {
          tr {
            thead { text "id" }
            thead { text "message" }
          }
          data.each { |e|
            tr {
              td { text e[:id].to_s }
              td { text e[:message].to_s }
            }
          }
        }
      }
    }
  end

  # Doctype not available in builder
  # builder only supports `thead`, tests need to see `th`
  res = ok("<!doctype html>" + html.gsub("thead", "th"))
  res.headers["Content-type"] = CONTENT::HTML
  res
end

# Postgres Test 5: Database Updates
app.get "/updates", do |request|
  updated = (1..sanitizedQueryCount(request)).map do
    world = randomWorld
    world[:randomNumber] = rand(1..ID_MAXIMUM)
    setWorld(world)
  end

  res = ok(updated.to_json)
  res.headers["Content-type"] = CONTENT::JSON
  res
end

app.run(8080)