Commit 7b8ae326 authored by skoulouzis's avatar skoulouzis Committed by GitHub

Merge pull request #46 from skoulouzis/DRIP_0.1

Created example code for component
parents 32f7b7c2 24adb25b
...@@ -16,6 +16,8 @@ ...@@ -16,6 +16,8 @@
package nl.uva.sne.drip.drip.component_example; package nl.uva.sne.drip.drip.component_example;
import java.io.File; import java.io.File;
import java.util.logging.Level;
import java.util.logging.Logger;
/** /**
* This is an example components. It is a dumy components to demonstrate a * This is an example components. It is a dumy components to demonstrate a
...@@ -25,7 +27,7 @@ import java.io.File; ...@@ -25,7 +27,7 @@ import java.io.File;
*/ */
public class Component { public class Component {
private final int input; private int input;
private final File inputTextFile; private final File inputTextFile;
private final File inputBinFile; private final File inputBinFile;
private final ExamplePOJO book; private final ExamplePOJO book;
...@@ -37,8 +39,14 @@ public class Component { ...@@ -37,8 +39,14 @@ public class Component {
this.book = book; this.book = book;
} }
String run() throws Exception { ExampleResult run() throws Exception {
throw new UnsupportedOperationException("Not supported yet."); //To change body of generated methods, choose Tools | Templates.
ExampleResult res = new ExampleResult();
res.output = input++;
res.inputTextFile = inputTextFile;
res.inputBinFile = inputBinFile;
res.book = book;
return res;
} }
} }
...@@ -30,8 +30,12 @@ import java.nio.file.Files; ...@@ -30,8 +30,12 @@ import java.nio.file.Files;
import java.nio.file.OpenOption; import java.nio.file.OpenOption;
import java.nio.file.Paths; import java.nio.file.Paths;
import java.nio.file.StandardOpenOption; import java.nio.file.StandardOpenOption;
import java.util.ArrayList;
import java.util.Base64; import java.util.Base64;
import java.util.Date; import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.logging.Level; import java.util.logging.Level;
import java.util.logging.Logger; import java.util.logging.Logger;
import org.json.JSONArray; import org.json.JSONArray;
...@@ -99,24 +103,18 @@ public class Consumer extends DefaultConsumer { ...@@ -99,24 +103,18 @@ public class Consumer extends DefaultConsumer {
JSONArray parameters = messageJson.getJSONArray("parameters"); JSONArray parameters = messageJson.getJSONArray("parameters");
int input = getInputInteger(parameters); int input = getInputInteger(parameters);
Logger.getLogger(Consumer.class.getName()).log(Level.INFO, "Input parameter is: {0}", input);
File inputTextFile = getInputFile(parameters, System.getProperty("java.io.tmpdir") + File.separator + "delete-me.txt"); File inputTextFile = getInputFile(parameters, System.getProperty("java.io.tmpdir") + File.separator + "delete-me.txt");
Logger.getLogger(Consumer.class.getName()).log(Level.INFO, "Input file is at: {0}. With size: {1}", new Object[]{inputTextFile.getAbsolutePath(), inputTextFile.length()});
File inputBinFile = getInputImageFile(parameters, System.getProperty("java.io.tmpdir")); File inputBinFile = getInputImageFile(parameters, System.getProperty("java.io.tmpdir"));
Logger.getLogger(Consumer.class.getName()).log(Level.INFO, "Input image file is at: {0}. With size: {1}", new Object[]{inputBinFile.getAbsolutePath(), inputBinFile.length()});
ExamplePOJO book = getExamplePOJO(parameters); ExamplePOJO book = getExamplePOJO(parameters);
Integer wordcount = book.getContent().trim().split("\\s+").length;
Logger.getLogger(Consumer.class.getName()).log(Level.INFO, "Created book object. Author: {0}. Langunage: {1}. Number of words: {2}", new Object[]{book.getAuthor(), book.getLanguage(), wordcount});
Component component = new Component(input, inputTextFile, inputBinFile, book); Component component = new Component(input, inputTextFile, inputBinFile, book);
String response; String response;
try { try {
response = component.run(); ExampleResult result = component.run();
response = createMessage(result);
} catch (Exception ex) { } catch (Exception ex) {
response = ex.getMessage(); response = ex.getMessage();
Logger.getLogger(Consumer.class.getName()).log(Level.SEVERE, null, ex); Logger.getLogger(Consumer.class.getName()).log(Level.SEVERE, null, ex);
...@@ -215,7 +213,7 @@ public class Consumer extends DefaultConsumer { ...@@ -215,7 +213,7 @@ public class Consumer extends DefaultConsumer {
private String getContent(URL url) throws IOException { private String getContent(URL url) throws IOException {
URLConnection conn = url.openConnection(); URLConnection conn = url.openConnection();
StringBuilder cont = null; StringBuilder cont;
try (BufferedReader br = new BufferedReader( try (BufferedReader br = new BufferedReader(
new InputStreamReader(conn.getInputStream()))) { new InputStreamReader(conn.getInputStream()))) {
String inputLine; String inputLine;
...@@ -227,4 +225,55 @@ public class Consumer extends DefaultConsumer { ...@@ -227,4 +225,55 @@ public class Consumer extends DefaultConsumer {
return cont.toString(); return cont.toString();
} }
private String createMessage(ExampleResult result) throws JSONException, IOException {
JSONObject message = new JSONObject();
message.put("creationDate", (System.currentTimeMillis()));
List parameters = new ArrayList();
//JSON objects must be added as values with "" and escape all ". For the
//message it's just a string
JSONObject bookJSON = examplePOJOtoJson(result.book);
String bookJSONString = bookJSON.toString().replaceAll("\"", "\\\"");
bookJSONString = "\"" + bookJSONString + "\"";
Map<String, String> bookParam = new HashMap<>();
bookParam.put("name", "book");
bookParam.put("value", bookJSONString);
bookParam.put("encoding", "UTF-8");
parameters.add(bookParam);
//Binarry files must be encoded as base64
byte[] bytes = Files.readAllBytes(Paths.get(result.inputBinFile.getAbsolutePath()));
byte[] encoded = Base64.getEncoder().encode(bytes);
String cont = new String(encoded, "US-ASCII");
Map<String, String> binFileParam = new HashMap<>();
binFileParam.put("name", "image");
binFileParam.put("value", cont);
binFileParam.put("encoding", "Base64");
parameters.add(binFileParam);
//Text files canh be encoded as to UTF-8
bytes = Files.readAllBytes(Paths.get(result.inputTextFile.getAbsolutePath()));
cont = new String(bytes, "UTF-8");
Map<String, String> textFileParam = new HashMap<>();
textFileParam.put("name", "text");
textFileParam.put("value", cont);
textFileParam.put("encoding", "UTF-8");
parameters.add(textFileParam);
Map<String, String> intParam = new HashMap<>();
intParam.put("name", "output");
intParam.put("value", String.valueOf(result.output));
parameters.add(intParam);
message.put("parameters", parameters);
return message.toString();
}
private JSONObject examplePOJOtoJson(ExamplePOJO book) throws JSONException {
JSONObject bookJSON = new JSONObject();
bookJSON.append("Author", book.getAuthor());
bookJSON.append("Content", book.getContent());
bookJSON.append("Language", book.getLanguage());
bookJSON.append("Translator", book.getTranslator());
return bookJSON;
}
} }
/*
* Copyright 2017 S. Koulouzis, Wang Junchao, Huan Zhou, Yang Hu
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package nl.uva.sne.drip.drip.component_example;
import java.io.File;
/**
*
* @author S. Koulouzis
*/
public class ExampleResult {
int output;
File inputTextFile;
File inputBinFile;
ExamplePOJO book;
}
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment