Add option based configuration support for "report duration threshold…

…" feature
thundra-io · May 7, 2022 · a2835ec · a2835ec
1 parent e0f06bb
commit a2835ec
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/src/controller.js b/src/controller.js
@@ -115,9 +115,12 @@ const _beforeInvocation = async (opts, event, context) => {
     await _startProfiler(opts)
 }
 
-const _shouldReport = (invocationDuration) => {
-    if (reportDurationThreshold) {
-        return invocationDuration > reportDurationThreshold
+const _shouldReport = (opts, invocationDuration) => {
+    const _reportDurationThreshold =
+        reportDurationThreshold ||
+        (opts && opts.report && opts.report.durationThreshold)
+    if (_reportDurationThreshold) {
+        return invocationDuration > _reportDurationThreshold
     } else {
         return true
     }
@@ -151,7 +154,7 @@ const _afterInvocation = async (
             fileName ||
             (opts && opts.s3 && opts.s3.fileName) ||
             MIDDY_PROFILER_S3_FILE_NAME_DEFAULT_VALUE
-        if (_shouldReport(invocationDuration)) {
+        if (_shouldReport(opts, invocationDuration)) {
             await reportToS3(
                 _profilingData,
                 _bucketName,