From 58942ad3dcc6141e43ce5e97320ac74da4cebf06 Mon Sep 17 00:00:00 2001 From: Eric Glasser Date: Tue, 14 Mar 2017 17:27:24 -0600 Subject: [PATCH] added the ability to set totalCountHeader on a specific request --- src/operations.js | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/src/operations.js b/src/operations.js index bfdf56f4..fc74bd5a 100644 --- a/src/operations.js +++ b/src/operations.js @@ -28,10 +28,19 @@ module.exports = function (model, options, excludedMap) { options.contextFilter(model, req, (filteredContext) => { buildQuery(filteredContext.find(), req._ermQueryOptions).then((items) => { + var totalCountHeader; req.erm.result = items req.erm.statusCode = 200 - if (options.totalCountHeader && !req._ermQueryOptions['distinct']) { + if (req.query.totalCountHeader === 'true') { + totalCountHeader = true; + } else if (req.query.totalCountHeader === 'false') { + totalCountHeader = false; + } else { + totalCountHeader = options.totalCountHeader; + } + + if (totalCountHeader && !req._ermQueryOptions['distinct']) { options.contextFilter(model, req, (countFilteredContext) => { buildQuery(countFilteredContext.count(), _.assign(req._ermQueryOptions, { skip: 0,