initial commit

Author: kylesomers

CloudFormation/CreateZombieWorkshop.json

@@ -0,0 +1,72 @@
+var AWS = require('aws-sdk');
+var path = require('path');
+
+/* == Globals == */
+var esDomain = {
+    region: 'us-west-2',
+    endpoint: 'http://ENDPOINT_HERE',
+    index: 'messages',
+    doctype: 'message'
+};
+var endpoint = new AWS.Endpoint(esDomain.endpoint);
+var creds = new AWS.EnvironmentCredentials('AWS');
+
+
+/* Lambda "main": Execution begins here */
+exports.handler = function(event, context) {
+    
+        console.log(JSON.stringify(event, null, '  '));
+        event.Records.forEach(function(record) {
+            if (typeof record.dynamodb.NewImage != 'undefined')
+             {
+                var doc = {message: {
+                    name: record.dynamodb.NewImage.name.S,
+                    message: record.dynamodb.NewImage.message.S,
+                    channel: record.dynamodb.NewImage.channel.S,
+                    timestamp: record.dynamodb.NewImage.timestamp.N}
+                }
+                console.log('document posted to ElasticSearch: ' + JSON.stringify(doc))
+                postToES(JSON.stringify(doc), context);
+             }
+             else
+             {
+                 console.log('skipping non-inserts');
+             }
+        });
+}
+
+
+/*
+ * Post the given document to Elasticsearch
+ */
+function postToES(doc, context) {
+    var req = new AWS.HttpRequest(endpoint);
+
+    console.log('create post request');
+    req.method = 'POST';
+    req.path = path.join('/', esDomain.index, esDomain.doctype);
+    req.region = esDomain.region;
+    req.headers['presigned-expires'] = false;
+    req.headers['Host'] = endpoint.host;
+    req.body = doc;
+
+    console.log('Creating the Signer for the post request');
+    var signer = new AWS.Signers.V4(req , 'es');  // es: service code
+    signer.addAuthorization(creds, new Date());
+
+    console.log('Sending Data');
+    var send = new AWS.NodeHttpClient();
+    send.handleRequest(req, null, function(httpResp) {
+        var respBody = '';
+        httpResp.on('data', function (chunk) {
+            respBody += chunk;
+        });
+        httpResp.on('end', function (chunk) {
+            console.log('Response: ' + respBody);
+            context.succeed('Lambda added document ' + doc);
+        });
+    }, function(err) {
+        console.log('Error: ' + err);
+        context.fail('Lambda failed with error ' + err);
+    });
+}