Skip to content

Commit 0f5ad7a

Browse files
authored
Merge pull request #696 from puppetlabs/fix-wire-rate-limiter-middleware
(P4DEVOPS-9434) Wire RateLimiter middleware into API Rack stack
2 parents 2d4eec1 + bbd299a commit 0f5ad7a

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

lib/vmpooler/api.rb

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
module Vmpooler
44
class API < Sinatra::Base
55
# Load API components
6-
%w[helpers dashboard v3 request_logger healthcheck].each do |lib|
6+
%w[helpers dashboard v3 request_logger healthcheck rate_limiter].each do |lib|
77
require "vmpooler/api/#{lib}"
88
end
99
# Load dashboard components
@@ -50,6 +50,9 @@ def self.execute(torun, config, redis, metrics, logger)
5050
# Enable API request logging only if required
5151
use Vmpooler::API::RequestLogger, logger: logger if config[:config]['request_logger']
5252

53+
# Enable rate limiting if configured
54+
use Vmpooler::API::RateLimiter, redis, config if config[:config]['rate_limiting_enabled']
55+
5356
use Vmpooler::Dashboard
5457
use Vmpooler::API::Dashboard
5558
use Vmpooler::API::V3

0 commit comments

Comments
 (0)