Lambda Wrapper - Homogenize comments

Author: Quentin Pierre

datadog/threadstats/aws_lambda.py

@@ -16,7 +16,10 @@ def my_lambda_handle(event, context):
 
 
 class _LambdaDecorator(object):
-    _counter = 0  # Number of opened wrappers, flush when 0
+    """ Decorator to automatically init & flush metrics, created for Lambda functions"""
+
+    # Number of opened wrappers, flush when 0
+    _counter = 0
     _counter_lock = Lock()
     _flush_lock = Lock()
     _was_initialized = False
@@ -38,7 +41,8 @@ def _close(cls):
         with cls._counter_lock:
             cls._counter = cls._counter - 1
 
-            if cls._counter <= 0:  # Flush only when all wrappers are closed
+            # Flush only when all wrappers are closed
+            if cls._counter <= 0:
                 should_flush = True
 
         if should_flush:

tests/performance/test_lambda_wrapper_thread_safety.py

@@ -24,7 +24,8 @@ def flush_distributions(self, dists):
 @datadog_lambda_wrapper
 def wrapped_function(id):
     lambda_stats("dist_" + str(id), 42)
-    time.sleep(0.001)  # sleep makes the os continue another thread
+    # sleep makes the os continue another thread
+    time.sleep(0.001)
 
     lambda_stats("common_dist", 42)
 

tests/unit/threadstats/test_threadstats.py

@@ -749,7 +749,7 @@ def test_metric_type(self):
     def test_basic_lambda_decorator(self):
 
         @datadog_lambda_wrapper
-        def basic_wrapped_function():  # Test custom_metric function
+        def basic_wrapped_function():
             lambda_stats("lambda.somemetric", 100)
 
         _lambda_stats.reporter = self.reporter