Added response message and readme.

Author: burkat

README.md

@@ -0,0 +1,3 @@
+This handler is created for hyperflow java tasks executed on aws lambda.
+To use it, build jar with "mvn clean package" command and upload it into lambda function.
+Function runtime and handler should be set to "Java 8" and "Handler::handleRequest" respectively.

src/main/java/Handler.java

@@ -3,6 +3,8 @@
 import com.amazonaws.services.s3.model.S3Object;
 import com.amazonaws.util.IOUtils;
 
+import java.io.IOException;
+import java.net.HttpURLConnection;
 import java.time.Duration;
 import java.time.Instant;
 import java.util.Map;
@@ -11,33 +13,38 @@ public class Handler {
 
     private static final String FOLDER_PATH = "/tmp/";
     private static final String BUCKET_KEY = "bucket";
+    private static final String PREFIX_KEY = "prefix";
 
-    public long handleRequest(Request request, Context context) throws Exception {
+    public Response handleRequest(Request request, Context context) {
 
         Instant instant1 = Instant.now();
-
         LambdaLogger logger = context.getLogger();
 
-        logger.log("executable: " + request.getExecutable());
-        logger.log("args:       " + request.getArgs());
-        logger.log("inputs:     " + request.getInputs());
-        logger.log("outputs:    " + request.getOutputs());
-        logger.log("bucket:     " + request.getOptions().get(BUCKET_KEY));
-        logger.log("prefix:     " + request.getOptions().get("prefix"));
-
-        downloadData(request, logger);
-        execute(request.getExecutable(), String.join(" ", request.getArgs()), logger);
-        uploadData(request, logger);
-
-        Instant instant2 = Instant.now();
-
-        return Duration.between(instant1, instant2).getSeconds();
+        try {
+            logger.log("executable: " + request.getExecutable());
+            logger.log("args:       " + request.getArgs());
+            logger.log("inputs:     " + request.getInputs());
+            logger.log("outputs:    " + request.getOutputs());
+            logger.log("bucket:     " + request.getOptions().get(BUCKET_KEY));
+            logger.log("prefix:     " + request.getOptions().get(PREFIX_KEY));
+
+            downloadData(request, logger);
+            execute(request.getExecutable(), String.join(" ", request.getArgs()), logger);
+            uploadData(request, logger);
+        } catch (Exception e) {
+            return handleException(e, request);
+        }
+        Response response = new Response();
+        response.setStatusCode(HttpURLConnection.HTTP_OK);
+        response.setMessage("Execution of " + request.getExecutable() + " successful, duration: " + Duration.between(instant1, Instant.now()).getSeconds() + " seconds");
+        response.setArgs(request.getArgs().toString());
+        return response;
     }
 
-    private void downloadData(Request request, LambdaLogger logger) throws Exception {
+    private void downloadData(Request request, LambdaLogger logger) throws IOException {
         for (Map<String, Object> input : request.getInputs()) {
             String fileName = input.get("name").toString();
-            String key = request.getOptions().get("prefix") + "/" + fileName;
+            String key = request.getOptions().get(PREFIX_KEY) + "/" + fileName;
             logger.log("Downloading " + request.getOptions().get(BUCKET_KEY) + "/" + key);
             S3Object s3Object = S3Utils.getObject(request.getOptions().get(BUCKET_KEY), key);
             S3Utils.saveToFile(s3Object, FOLDER_PATH + fileName);
@@ -63,11 +70,27 @@ private void uploadData(Request request, LambdaLogger logger) {
         for (Map<String, String> input : request.getOutputs()) {
             String fileName = input.get("name");
             String filePath = FOLDER_PATH + fileName;
-            String key = request.getOptions().get("prefix") + "/" + fileName;
-            //String key = "test" + "/" + fileName;
+            String key = request.getOptions().get(PREFIX_KEY) + "/" + fileName;
             logger.log("Uploading " + request.getOptions().get(BUCKET_KEY) + "/" + key);
             S3Utils.putObject(request.getOptions().get(BUCKET_KEY), key, filePath);
         }
     }
 
+    private Response handleException(Exception e, Request request) {
+        Throwable cause;
+        if (e.getCause() != null) {
+            cause = e.getCause();
+        } else {
+            cause = e;
+        }
+        while (cause.getCause() != null) {
+            cause = cause.getCause();
+        }
+        Response response = new Response();
+        response.setStatusCode(HttpURLConnection.HTTP_INTERNAL_ERROR);
+        response.setMessage("Execution of " + request.getExecutable() + " failed, cause: " + cause.getMessage());
+        response.setArgs(request.getArgs().toString());
+        return response;
+    }
+
 }

src/main/java/Response.java

@@ -0,0 +1,30 @@
+public class Response {
+
+    private int statusCode;
+    private String message;
+    private String args;
+
+    public int getStatusCode() {
+        return statusCode;
+    }
+
+    public void setStatusCode(int statusCode) {
+        this.statusCode = statusCode;
+    }
+
+    public String getMessage() {
+        return message;
+    }
+
+    public void setMessage(String message) {
+        this.message = message;
+    }
+
+    public String getArgs() {
+        return args;
+    }
+
+    public void setArgs(String args) {
+        this.args = args;
+    }
+}