@@ -31,10 +31,10 @@ exports.register = function(plugin, options, next) {
31
31
if ( token ) {
32
32
if ( token >= limit . limit ) {
33
33
error = plugin . hapi . error . badRequest ( 'Rate limit exceeded' ) ;
34
- error . response . code = 429 ; // Assign a Too Many Requests response
35
- error . response . headers [ 'X-Rate-Limit-Limit' ] = request . plugins [ 'hapi-ratelimit' ] . limit ;
36
- error . response . headers [ 'X-Rate-Limit-Remaining' ] = request . plugins [ 'hapi-ratelimit' ] . remaining ;
37
- error . response . headers [ 'X-Rate-Limit-Reset' ] = request . plugins [ 'hapi-ratelimit' ] . reset ;
34
+ error . output . statusCode = 429 ; // Assign a Too Many Requests response
35
+ error . output . headers [ 'X-Rate-Limit-Limit' ] = request . plugins [ 'hapi-ratelimit' ] . limit ;
36
+ error . output . headers [ 'X-Rate-Limit-Remaining' ] = request . plugins [ 'hapi-ratelimit' ] . remaining ;
37
+ error . output . headers [ 'X-Rate-Limit-Reset' ] = request . plugins [ 'hapi-ratelimit' ] . reset ;
38
38
error . reformat ( ) ;
39
39
}
40
40
redisClient . incr ( ipts , function ( err ) {
@@ -54,11 +54,11 @@ exports.register = function(plugin, options, next) {
54
54
plugin . ext ( 'onPostHandler' , function ( request , callback ) {
55
55
var response ;
56
56
if ( 'hapi-ratelimit' in request . plugins ) {
57
- response = request . response ( ) ;
57
+ response = request . response ;
58
58
if ( ! response . isBoom ) {
59
- response . _headers [ 'X-Rate-Limit-Limit' ] = request . plugins [ 'hapi-ratelimit' ] . limit ;
60
- response . _headers [ 'X-Rate-Limit-Remaining' ] = request . plugins [ 'hapi-ratelimit' ] . remaining ;
61
- response . _headers [ 'X-Rate-Limit-Reset' ] = request . plugins [ 'hapi-ratelimit' ] . reset ;
59
+ response . headers [ 'X-Rate-Limit-Limit' ] = request . plugins [ 'hapi-ratelimit' ] . limit ;
60
+ response . headers [ 'X-Rate-Limit-Remaining' ] = request . plugins [ 'hapi-ratelimit' ] . remaining ;
61
+ response . headers [ 'X-Rate-Limit-Reset' ] = request . plugins [ 'hapi-ratelimit' ] . reset ;
62
62
}
63
63
}
64
64
callback ( ) ;
0 commit comments