xray testing

lib/server.js

@@ -58,10 +58,21 @@ module.exports = function (properties) {
   var error = properties.modules && properties.modules.error
   var logger = properties.modules && properties.modules.logger
   var requestid = properties.modules && properties.modules.requestid
+  var xray_sample = properties.xray && properties.xray.sample
+  var xray_service = properties.xray && properties.xray.service
+  var xray_app_name = properties.xray && properties.xray.app_name
   // /////////////////////////Building the express server///////////////////////////
 
   //  Express Server
   var app = express()
+  var AWSXRay
+  if (xray_sample) {
+    AWSXRay = require('aws-xray-sdk')
+    if (xray_service) {
+      AWSXRay.config([AWSXRay.plugins[xray_service]])
+    }
+    app.use(AWSXRay.express.openSegment(xray_app_name || 'default_app_name'))
+  }
 
   // If proxy procotol is enabled we need to fix standard headers
   if (proxy === 'true') {
@@ -151,6 +162,10 @@ module.exports = function (properties) {
   })
   error && app.use(error)
 
+  if (xray_sample) {
+    app.use(AWSXRay.express.closeSegment())
+  }
+
   // ////////////////////Building The returned Server Object//////////////////////
   var server = {
     port: http_port,