Merge pull request #359 from jmehnle/patch-1

Lambda wrapper - Always flush, even on exceptions
DataDog · Mar 29, 2019 · bc33215 · bc33215
2 parents d5d5332 + c4d1d85
commit bc33215
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/datadog/threadstats/aws_lambda.py b/datadog/threadstats/aws_lambda.py
@@ -62,9 +62,10 @@ def _close(cls):
 
     def __call__(self, *args, **kw):
         _LambdaDecorator._enter()
-        result = self.func(*args, **kw)
-        _LambdaDecorator._close()
-        return result
+        try:
+            return self.func(*args, **kw)
+        finally:
+            _LambdaDecorator._close()
 
 
 _lambda_stats = ThreadStats()