Merge pull request #140 from ryandub/tracingsupport

Add support for X-Ray tracing

Author: jarosser06

README.md

@@ -53,6 +53,9 @@ Example `lambda.json` file:
     "security_groups": [
       "sg-00000000"
     ]
+  },
+  "tracing": {
+    "Mode": "Active"
   }
 }
 ```

README.rst

@@ -67,6 +67,9 @@ Example ``lambda.json`` file:
         "security_groups": [
           "sg-00000000"
         ]
+      },
+      "tracing": {
+        "Mode": "Active"
       }
     }
 

lambda_uploader/config.py

@@ -29,12 +29,13 @@
 REQUIRED_KINESIS_SUBSCRIPTION_PARAMS = {u'stream': basestring,
                                         u'batch_size': int,
                                         u'starting_position': basestring}
+REQUIRED_TRACING_MODES = ['Active', 'PassThrough']
 
 DEFAULT_PARAMS = {u'requirements': [], u'publish': False,
                   u'alias': None, u'alias_description': None,
                   u'ignore': [], u'extra_files': [], u'vpc': None,
                   u's3_bucket': None, u's3_key': None, u'runtime': 'python2.7',
-                  u'variables': {}, u'subscription': {}}
+                  u'variables': {}, u'subscription': {}, u'tracing': {}}
 
 
 class Config(object):
@@ -51,6 +52,8 @@ def __init__(self, pth, config_file=None, role=None, variables=None):
             self._validate_vpc()
         if self._config['subscription']:
             self._validate_subscription()
+        if self._config['tracing']:
+            self._validate_tracing()
 
         for param, clss in REQUIRED_PARAMS.items():
             self._validate(param, cls=clss)
@@ -126,6 +129,16 @@ def _validate_vpc(self):
                                     " strings. '%s' contains something else"
                                     % param)
 
+    '''Validate the tracing configuration'''
+    def _validate_tracing(self):
+        if len(self._config['tracing']) > 1:
+            raise TypeError("Tracing Config can only contain one item")
+        if not self._config['tracing'].get('Mode'):
+            raise TypeError("Tracing Config must contain the `Mode` key")
+        if self._config['tracing']['Mode'] not in REQUIRED_TRACING_MODES:
+            raise TypeError("Tracing Config Mode must be one of {}".format(
+                ', '.join(REQUIRED_TRACING_MODES)))
+
     '''Validate the subscription configuration.
     All kinds of subscription will be validated here'''
     def _validate_subscription(self):

lambda_uploader/uploader.py

@@ -71,6 +71,7 @@ def upload_existing(self, pkg):
             MemorySize=self._config.memory,
             VpcConfig=self._vpc_config,
             Environment=environment,
+            TracingConfig=self._config.tracing,
         )
         LOG.debug("AWS update_function_configuration response: %s"
                   % response)
@@ -117,6 +118,7 @@ def upload_new(self, pkg):
             Publish=self._config.publish,
             VpcConfig=self._vpc_config,
             Environment=environment,
+            TracingConfig=self._config.tracing,
         )
         LOG.debug("AWS create_function response: %s" % response)