TechEmpower.FrameworkBenchmarks/frameworks/Ruby/grape/boot.rb

require 'erb'
require 'active_record'
require 'yaml'

MAX_PK = 10_000
ID_RANGE = (1..MAX_PK).freeze
ALL_IDS = ID_RANGE.to_a
QUERIES_MIN = 1
QUERIES_MAX = 500

Bundler.require :default

db_config = YAML.load(ERB.new(File.read('config/database.yml')).result)[ENV['RACK_ENV']]
ActiveRecord::Base.establish_connection(db_config)

class World < ActiveRecord::Base
  self.table_name = 'World'
end

module Acme
  class HelloWorld < Grape::API
    get '/json' do
      {message:'Hello, World!'}
    end
  end

  class PlainText < Grape::API
    content_type :plain, 'text/plain'
    format :plain

    get '/plaintext' do
      'Hello, World!'
    end
  end

  class DatabaseQueries < Grape::API
    logger nil
    helpers do
      def bounded_queries
        queries = params[:queries].to_i
        queries.clamp(QUERIES_MIN, QUERIES_MAX)
      end

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

    get '/db' do
      ActiveRecord::Base.with_connection do
        World.find(rand1).attributes
      end
    end

    get '/query' do
      ActiveRecord::Base.with_connection do
        ALL_IDS.sample(bounded_queries).map do |id|
          World.find(id)
        end
      end
    end

    get '/updates' do
      worlds =
        ActiveRecord::Base.with_connection do
          ALL_IDS.sample(bounded_queries).map do |id|
            world = World.find(id)
            new_value = rand1
            new_value = rand1 while new_value == world.randomNumber
            world.update_columns(randomNumber: new_value)
            world
          end
        end
    end
  end

  class API < Grape::API
    before do
      header 'Date', Time.now.httpdate if defined?(Puma)
      header 'Server', 'grape'
    end
    logger nil
    content_type :json, 'application/json'
    format :json

    mount ::Acme::HelloWorld
    mount ::Acme::PlainText
    mount ::Acme::DatabaseQueries
  end
end