Merge pull request #696 from puppetlabs/fix-wire-rate-limiter-middleware

(P4DEVOPS-9434) Wire RateLimiter middleware into API Rack stack
This commit is contained in:
Mahima Singh 2026-03-19 11:53:10 +05:30 committed by GitHub
commit 0f5ad7aafd
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -3,7 +3,7 @@
module Vmpooler module Vmpooler
class API < Sinatra::Base class API < Sinatra::Base
# Load API components # Load API components
%w[helpers dashboard v3 request_logger healthcheck].each do |lib| %w[helpers dashboard v3 request_logger healthcheck rate_limiter].each do |lib|
require "vmpooler/api/#{lib}" require "vmpooler/api/#{lib}"
end end
# Load dashboard components # Load dashboard components
@ -50,6 +50,9 @@ module Vmpooler
# Enable API request logging only if required # Enable API request logging only if required
use Vmpooler::API::RequestLogger, logger: logger if config[:config]['request_logger'] use Vmpooler::API::RequestLogger, logger: logger if config[:config]['request_logger']
# Enable rate limiting if configured
use Vmpooler::API::RateLimiter, redis, config if config[:config]['rate_limiting_enabled']
use Vmpooler::Dashboard use Vmpooler::Dashboard
use Vmpooler::API::Dashboard use Vmpooler::API::Dashboard
use Vmpooler::API::V3 use Vmpooler::API::V3