Merge pull request #75 from richarddd/add-warmup-client-context

Add Client Context to invoke params

Author: goncaloneves

README.md

@@ -202,6 +202,20 @@ module.exports.lambdaToWarm = function(event, context, callback) {
   ... add lambda logic after
 }
 ```
+You can also check for the warmp event using the `context` variable. This could be useful if you are handling the raw input and output streams:
+
+```javascript
+...
+
+if(context.custom.source === 'serverless-plugin-warmup'){
+  console.log('WarmUP - Lambda is warm!')
+  return callback(null, 'Lambda is warm!')
+}
+
+...
+```
+
+
 
 * All done! WarmUP will run on SLS `deploy` and `package` commands
 

src/index.js

@@ -57,6 +57,7 @@ class WarmUP {
       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()
@@ -104,6 +105,7 @@ class WarmUP {
     this.pathFolder = this.getPath(this.folderName)
     this.pathFile = this.pathFolder + '/index.js'
     this.pathHandler = this.folderName + '/index.warmUp'
+    this.payload = JSON.stringify({ source: 'serverless-plugin-warmup' })
 
     /** Default options */
     this.warmup = {
@@ -271,11 +273,12 @@ module.exports.warmUp = async (event, context, callback) => {
   console.log("Warm Up Start");
   const invokes = await Promise.all(functionNames.map(async (functionName) => {
     const params = {
+      ClientContext: "${Buffer.from(`{"custom":${this.payload}}`).toString('base64')}",
       FunctionName: functionName,
       InvocationType: "RequestResponse",
       LogType: "None",
       Qualifier: process.env.SERVERLESS_ALIAS || "$LATEST",
-      Payload: '${JSON.stringify({ source: 'serverless-plugin-warmup' })}'
+      Payload: ${this.payload}
     };
 
     try {