#8: x-ray instrumentation

Appliscale · Dec 1, 2017 · 0d81d34 · 0d81d34
1 parent de64b98
commit 0d81d34
Show file tree

Hide file tree

Showing 6 changed files with 121 additions and 4 deletions.
diff --git a/cat-rekognition/package-lock.json b/cat-rekognition/package-lock.json
diff --git a/cat-rekognition/package.json b/cat-rekognition/package.json
@@ -4,12 +4,14 @@
   "description": "Detecting cat in a picture",
   "main": "src/handler.js",
   "dependencies": {
+    "aws-xray-sdk": "1.1.6",
     "serverless": "1.24.0",
     "serverless-s3-remover": "0.4.0"
   },
   "devDependencies": {
     "chai": "3.5.0",
     "mocha": "3.2.0",
+    "serverless-plugin-tracing": "2.0.0",
     "sinon": "4.1.2"
   },
   "scripts": {

diff --git a/cat-rekognition/serverless.yml b/cat-rekognition/serverless.yml
@@ -5,6 +5,7 @@ provider:
   region: eu-west-1
   profile: appliscale-dev
   runtime: nodejs6.10
+  tracing: true
   iamRoleStatements:
     - Effect: Allow
       Action:
@@ -23,6 +24,10 @@ provider:
         - dynamodb:UpdateItem
         - dynamodb:DeleteItem
       Resource: "*"
+    - Effect: Allow
+      Action:
+        - xray:*
+      Resource: "*"
 
 functions:
   uploadcatpicture:
@@ -69,7 +74,6 @@ resources:
         SourceArn: "arn:aws:s3:::serverless-cat-detector-img-repo"
     CatStatusDbTable:
       Type: 'AWS::DynamoDB::Table'
-      DeletionPolicy: Retain
       Properties:
         AttributeDefinitions:
           -
@@ -86,6 +90,7 @@ resources:
 
 plugins:
   - serverless-s3-remover
+  - serverless-plugin-tracing
 
 custom:
   remover:

diff --git a/cat-rekognition/src/persistence.js b/cat-rekognition/src/persistence.js
@@ -2,7 +2,8 @@
 
 const STATUS_TABLE_NAME = 'ServerlessCatDetectorStatus';
 
-const AWS = require('aws-sdk');
+const AWSXRay = require('aws-xray-sdk');
+const AWS = AWSXRay.captureAWS(require('aws-sdk'));
 const dynamoDb = new AWS.DynamoDB.DocumentClient();
 
 module.exports.dbItem = (fileName, scanned, scanningStatus) => {

diff --git a/cat-rekognition/src/recognition.js b/cat-rekognition/src/recognition.js
@@ -1,6 +1,7 @@
 'use strict';
 
-const AWS = require('aws-sdk');
+const AWSXRay = require('aws-xray-sdk');
+const AWS = AWSXRay.captureAWS(require('aws-sdk'));
 const rekognition = new AWS.Rekognition();
 
 module.exports.check = (fileName) => {

diff --git a/cat-rekognition/src/upload.js b/cat-rekognition/src/upload.js
@@ -1,6 +1,7 @@
 'use strict';
 
-const AWS = require('aws-sdk');
+const AWSXRay = require('aws-xray-sdk');
+const AWS = AWSXRay.captureAWS(require('aws-sdk'));
 const s3 = new AWS.S3();
 
 const CORS_HEADERS = { 'Access-Control-Allow-Origin': '*' };