From 78229eef0b294a1d837e470ff6073e7af07331b7 Mon Sep 17 00:00:00 2001
From: Justin Pihony <justin.pihony@gmail.com>
Date: Mon, 4 May 2020 00:52:51 -0400
Subject: [PATCH] Use reboot version and allow for no default instance

---
 aws/build.gradle | 18 +++++++++++++-----
 1 file changed, 13 insertions(+), 5 deletions(-)

diff --git a/aws/build.gradle b/aws/build.gradle
index fa2a8d65ea3..837d369e31f 100755
--- a/aws/build.gradle
+++ b/aws/build.gradle
@@ -1,6 +1,6 @@
 import com.amazonaws.services.lambda.model.InvocationType
 import com.amazonaws.services.lambda.model.InvokeResult
-import jp.classmethod.aws.gradle.lambda.AWSLambdaInvokeTask
+import jp.classmethod.aws.reboot.gradle.lambda.AWSLambdaInvokeTask
 
 
 buildscript {
@@ -11,7 +11,7 @@ buildscript {
         maven { url "https://plugins.gradle.org/m2/" }
     }
     dependencies {
-        classpath "jp.classmethod.aws:gradle-aws-plugin:0.35"
+        classpath "jp.classmethod.aws.reboot:gradle-aws-plugin-reboot:0.42"
     }
 }
 
@@ -19,7 +19,7 @@ group = 'beam'
 version = '0.8.0'
 
 apply plugin: "base"
-apply plugin: "jp.classmethod.aws.lambda"
+apply plugin: "jp.classmethod.aws.reboot.lambda"
 
 aws {
     profileName = "gradle"
@@ -28,6 +28,11 @@ aws {
 
 lambda {
     region = "us-east-2"
+    maxErrorRetry = 0
+    requestTimeout = 900000
+    clientExecutionTimeout = 900000
+    connectionTimeout = 900000
+    socketTimeout = 900000
 }
 
 task deploy(type: AWSLambdaInvokeTask) {
@@ -73,6 +78,9 @@ task deploy(type: AWSLambdaInvokeTask) {
                 break
         }
 
+        def tempInstanceType = "${ext.getParameterValue('instanceType') ?: (project.hasProperty('defaultInstanceType') ? defaultInstanceType : '')}"
+        def finalInstanceType = tempInstanceType != "" ? tempInstanceType : null
+
         def pload = """{
   "title": "${ext.getParameterValue('runName') + '_' + getCurrentGitUserEmail()}",
   "git_user_email": "${getCurrentGitUserEmail()}",
@@ -88,7 +96,7 @@ task deploy(type: AWSLambdaInvokeTask) {
   "storage_size": ${ext.getParameterValue('storageSize')},
   "batch": ${ext.getParameterValue('beamBatch')},
   "s3_publish": ${ext.getParameterValue('s3Backup') ?: true},
-  "instance_type": "${ext.getParameterValue('instanceType') ?: defaultInstanceType}",
+  "instance_type": ${finalInstanceType},
   "region": "${ext.getParameterValue('region') ?: defaultRegion}",
   "shutdown_wait": "${ext.getParameterValue('shutdownWait')}",
   "shutdown_behaviour": "${ext.getParameterValue('shutdownBehaviour')}",
@@ -100,6 +108,7 @@ task deploy(type: AWSLambdaInvokeTask) {
   "min_memory": ${ext.getParameterValue('minMemory') ?: 0},
   "max_memory": ${ext.getParameterValue('maxMemory') ?: 0}
 }"""
+        println payload
         println "Please note that if you set isSpot to true then it could take an excessive time period. In fact it could time out at 15 minutes and still complete later on. ALSO! volumes using spot instances must be MANUALLY deleted. This is done so you do not lose data in the case of a premature shutdown."
         functionName = "simulateBeam"
         invocationType = InvocationType.RequestResponse
@@ -107,7 +116,6 @@ task deploy(type: AWSLambdaInvokeTask) {
     }
 
     doLast {
-        println payload
         println new String(invokeResult.payload.array(), "UTF-8")
     }
 }