GroovySparkThroughGroovyShell.groovy

import groovy.lang.Binding;
import groovy.lang.GroovyShell;
import org.codehaus.groovy.control.CompilerConfiguration;
import org.codehaus.groovy.control.MultipleCompilationErrorsException;

import java.io.*;
import java.util.ArrayList;
import java.util.List;
import java.util.Properties;

//Logger logger = LoggerFactory.getLogger(GroovySparkThoughGroovyShell.class);

GroovyShell interpreter;
Binding binding;

binding = new Binding();

CompilerConfiguration compilerConfiguration = new CompilerConfiguration();
//compilerConfiguration.setClasspath();
//compilerConfiguration.setOutput(new PrintWriter());

interpreter = new GroovyShell(binding, compilerConfiguration);
//logger.info("GroovySparkInterpreter<open>: interpreter", interpreter);

String script = new File('/Users/tog/Work/GroovySpark/GroovySparkWordcount.groovy').getText('UTF-8')
//logger.info("GroovySparkInterpreter<interpret>: ", script);

PrintStream originalOut = System.out;
PrintStream originalErr = System.err;

ByteArrayOutputStream stream = new ByteArrayOutputStream();
PrintStream printStream = null;
try {
  printStream = new PrintStream(stream, true, "UTF-8");
} catch (UnsupportedEncodingException e) {
  e.printStackTrace();
}

StringWriter stacktrace = new StringWriter();
PrintWriter errWriter = new PrintWriter(stacktrace);

System.setOut(printStream);
System.setErr(printStream);

Object result = null;
try {
  result = interpreter.evaluate(script);
} catch (MultipleCompilationErrorsException e) {
   stacktrace.append(e.getMessage());
} catch (Throwable t) {
   t.printStackTrace(errWriter);
} finally {
  System.setOut(originalOut);
  System.setErr(originalErr);
}

System.out.println("" + stacktrace);

if (stacktrace.toString().length() > 0) {
  println new String(stream.toByteArray()) + "\n" + stacktrace.toString()
} else {
  println new String(stream.toByteArray())
}