puppetlabs/vmpooler
2
0
Fork 0
mirror of https://github.com/puppetlabs/vmpooler.git synced 2026-03-27 00:57:47 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

(P4DEVOPS-9434) Wire RateLimiter middleware into API Rack stack

Browse source
This commit is contained in:
Mahima Singh 2026-03-19 11:43:34 +05:30
parent 2d4eec1f85
commit dfb2a384ab
1 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

7
lib/vmpooler/api.rb
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,11 @@ 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
if config[:config]['rate_limiting_enabled']
use Vmpooler::API::RateLimiter, redis, config
end
use Vmpooler::Dashboard use Vmpooler::Dashboard
use Vmpooler::API::Dashboard use Vmpooler::API::Dashboard
use Vmpooler::API::V3 use Vmpooler::API::V3