Merge pull request #71 from juanjoDiaz/update_to_node_8

Update to node 8

Author: goncaloneves

src/index.js

@@ -49,16 +49,16 @@ class WarmUP {
    * */
   afterPackageInitialize () {
     // See https://github.com/serverless/serverless/issues/2631
-    this.options.stage  = this.options.stage
-      || this.serverless.service.provider.stage
-      || (this.serverless.service.defaults && this.serverless.service.defaults.stage)
-      || 'dev'
-    this.options.region = this.options.region
-      || this.serverless.service.provider.region
-      || (this.serverless.service.defaults && this.serverless.service.defaults.region)
-      || 'us-east-1'
+    this.options.stage = this.options.stage ||
+      this.serverless.service.provider.stage ||
+      (this.serverless.service.defaults && this.serverless.service.defaults.stage) ||
+      'dev'
+    this.options.region = this.options.region ||
+      this.serverless.service.provider.region ||
+      (this.serverless.service.defaults && this.serverless.service.defaults.region) ||
+      'us-east-1'
     this.custom = this.serverless.service.custom
-    
+
     this.configPlugin()
     return this.createWarmer()
   }
@@ -260,37 +260,35 @@ class WarmUP {
       return functionObject.name
     })
 
-    /** Write function invoke promises and push to array */
     const warmUpFunction = `"use strict";
 
 /** Generated by Serverless WarmUP Plugin at ${new Date().toISOString()} */
 const aws = require("aws-sdk");
 aws.config.region = "${this.options.region}";
 const lambda = new aws.Lambda();
-const functionNames = "${functionNames.join()}".split(",");
-module.exports.warmUp = (event, context, callback) => {
-  let invokes = [];
-  let errors = 0;
+const functionNames = ${JSON.stringify(functionNames)};
+module.exports.warmUp = async (event, context, callback) => {
   console.log("Warm Up Start");
-  functionNames.forEach((functionName) => {
+  const invokes = await Promise.all(functionNames.map(async (functionName) => {
     const params = {
       FunctionName: functionName,
       InvocationType: "RequestResponse",
       LogType: "None",
       Qualifier: process.env.SERVERLESS_ALIAS || "$LATEST",
-      Payload: JSON.stringify({ source: "serverless-plugin-warmup" })
+      Payload: '${JSON.stringify({ source: 'serverless-plugin-warmup' })}'
     };
-    invokes.push(lambda.invoke(params).promise().then((data) => {
-      console.log("Warm Up Invoke Success: " + functionName, data);
-    }, (error) => {
-      errors++;
-      console.log("Warm Up Invoke Error: " + functionName, error);
-    }));
-  });
-  Promise.all(invokes).then(() => {
-    console.log("Warm Up Finished with " + errors + " invoke errors");
-    callback();
-  });
+
+    try {
+      const data = await lambda.invoke(params).promise();
+      console.log(\`Warm Up Invoke Success: \${functionName}\`, data);
+      return true;
+    } catch (e) {
+      console.log(\`Warm Up Invoke Error: \${functionName}\`, e);
+      return false;
+    }
+  }));
+
+  console.log(\`Warm Up Finished with \${invokes.filter(r => !r).length} invoke errors\`);
 }`
 
     /** Write warm up file */
@@ -310,7 +308,7 @@ module.exports.warmUp = (event, context, callback) => {
       handler: this.pathHandler,
       memorySize: this.warmup.memorySize,
       name: this.warmup.name,
-      runtime: 'nodejs6.10',
+      runtime: 'nodejs8.10',
       package: {
         individually: true,
         exclude: ['**'],