Decorating a retry-after header for clients

avoidwork · Feb 17, 2019 · 26c39a5 · 26c39a5
1 parent a15f004
commit 26c39a5
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/lib/middleware.js b/lib/middleware.js
@@ -182,18 +182,21 @@ function keymaster (req, res) {
 }
 
 function rate (req, res, next) {
-	if (req.server.config.rate.enabled === false || req.unprotect) {
+	const config = req.server.config.rate;
+
+	if (config.enabled === false || req.unprotect) {
 		next();
 	} else {
-		const results = req.server.rate(req, req.server.config.rate.override),
+		const results = req.server.rate(req, config.override),
 			good = results.shift();
 
 		each(rateHeaders, (i, idx) => res.header(i, results[idx]));
+		res.header('retry-after', config.reset);
 
 		if (good) {
 			next();
 		} else {
-			res.error(req.server.config.rate.status || 429);
+			res.error(config.status || 429);
 		}
 	}
 }