Contents

# frozen_string_literal: true

require 'dynoscale_ruby/request_calculator'
require 'dynoscale_ruby/reporter'
require 'dynoscale_ruby/recorder'
require 'dynoscale_ruby/worker/sidekiq'
require 'dynoscale_ruby/worker/resque'
require 'dynoscale_ruby/logger'

module DynoscaleRuby
  class Middleware
    include Logger

    MEASUREMENT_TTL = 5 # minutes


    def initialize(app)
      @app = app
    end

    def call(env)
      is_dev = ENV['DYNOSCALE_DEV'] == 'true'
      dyno = is_dev ? "dev.1" : ENV['DYNO']

      unless ENV['DYNOSCALE_URL']
        puts "Missing DYNOSCALE_URL environment variable"
        return @app.call(env)
      end 
      return @app.call(env) if ENV['SKIP_DYNASCALE_AGENT']
      return @app.call(env) unless is_dev || ENV['DYNO']&.split(".")&.last == "1"

      request_calculator = RequestCalculator.new(env)
      workers =  DynoscaleRuby::Worker.constants.select{|c| DynoscaleRuby::Worker.const_get(c).is_a? Class }.map{|c| DynoscaleRuby::Worker.const_get(c) }
      Recorder.record!(request_calculator, workers)

      api_wrapper = DynoscaleRuby::ApiWrapper.new(dyno, ENV['DYNOSCALE_URL'], ENV['HEROKU_APP_NAME'])
      Reporter.start!(Recorder, api_wrapper) unless Reporter.running?


      @app.call(env)
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
dynoscale_ruby-1.0.2	lib/dynoscale_ruby/middleware.rb
dynoscale_ruby-1.0.1	lib/dynoscale_ruby/middleware.rb