[beta] Python Profiling support

Dockerfile

@@ -20,4 +20,5 @@ RUN rm -rf ./python/lib/$runtime/site-packages/botocore*
 
 # Remove profiling (7MB) to reduce package size.
 # Continous profiling is not yet supported anyway.
-RUN rm -rf ./python/lib/$runtime/site-packages/ddtrace/profiling
+# RUN rm -rf ./python/lib/$runtime/site-packages/ddtrace/profiling
+COPY ./ddtrace ./python/lib/$runtime/site-packages/ddtrace

datadog_lambda/wrapper.py

@@ -35,12 +35,16 @@
 )
 from datadog_lambda.trigger import extract_trigger_tags, extract_http_status_code_tag
 from datadog_lambda.tag_object import tag_object
+from ddtrace.profiling import profiler
 
 logger = logging.getLogger(__name__)
 
 dd_capture_lambda_payload_enabled = (
     os.environ.get("DD_CAPTURE_LAMBDA_PAYLOAD", "false").lower() == "true"
 )
+profiling_env_var = os.environ.get("DD_PROFILING_ENABLED", "false").lower() == "true"
+service_env_var = os.environ.get("DD_SERVICE", "DefaultServiceName")
+env_env_var = os.environ.get("DD_ENV", "dev")
 
 """
 Usage:
@@ -111,7 +115,9 @@ def __init__(self, func):
                 os.environ.get("DD_TRACE_MANAGED_SERVICES", "true").lower() == "true"
             )
             self.response = None
-
+            self.prof = profiler.Profiler(
+                env=env_env_var, service=service_env_var, version="1.0.0"
+            )
             if self.extractor_env:
                 extractor_parts = self.extractor_env.rsplit(".", 1)
                 if len(extractor_parts) == 2:
@@ -181,7 +187,9 @@ def _before(self, event, context):
                 )
             else:
                 set_correlation_ids()
-
+            if profiling_env_var == True and is_cold_start():
+                print("starting profiler in _before()")
+                self.prof.start(stop_on_exit=False, profile_children=True)
             logger.debug("datadog_lambda_wrapper _before() done")
         except Exception:
             traceback.print_exc()