merge pr trung#18&trung#19

Author: xzxiaoshan

src/main/java/org/mdkt/compiler/CompilationResult.java

@@ -0,0 +1,40 @@
+package org.mdkt.compiler;
+
+import java.util.List;
+import java.util.Map;
+
+import javax.tools.Diagnostic;
+import javax.tools.JavaFileObject;
+
+public interface CompilationResult {
+    /**
+     * Return the compiled classes
+     */
+    Map<String, Class<?>> classMap() throws ClassNotFoundException;
+
+    /**
+     * Determine if the compilation did succeed
+     */
+    boolean compilationSucceeded();
+
+    /**
+     * Determine if any warnings are present
+     */
+    boolean hasWarnings();
+
+    /**
+     * Determine if any errors are present
+     */
+    boolean hasErrors();
+
+    /**
+     * Return the diagnostics produced by the compiler
+     */
+    List<Diagnostic<? extends JavaFileObject>> getDiagnostics();
+
+    /**
+     * Throw an exception if the compilation did not succeed
+     */
+    CompilationResult checkNoErrors();
+
+}

src/main/java/org/mdkt/compiler/ExtendedStandardJavaFileManager.java

@@ -1,15 +1,14 @@
 package org.mdkt.compiler;
 
-import java.io.FileNotFoundException;
-import java.io.IOException;
-import java.util.ArrayList;
-import java.util.Arrays;
-import java.util.List;
 
 import javax.tools.FileObject;
 import javax.tools.ForwardingJavaFileManager;
 import javax.tools.JavaFileManager;
 import javax.tools.JavaFileObject;
+import java.io.IOException;
+import java.util.Arrays;
+import java.util.Map;
+import java.util.stream.Collectors;
 
 /**
  * Created by trung on 5/3/15. Edited by turpid-monkey on 9/25/15, completed
@@ -18,20 +17,20 @@
 public class ExtendedStandardJavaFileManager extends
 		ForwardingJavaFileManager<JavaFileManager> {
 
-	private List<CompiledCode> compiledCode = new ArrayList<CompiledCode>();
+	private Map<String, CompiledCode> compiledCode;
 	private DynamicClassLoader cl;
 
 	/**
 	 * Creates a new instance of ForwardingJavaFileManager.
 	 *
-	 * @param fileManager
-	 *            delegate to this file manager
+	 * @param fileManager delegate to this file manager
 	 * @param cl
 	 */
 	protected ExtendedStandardJavaFileManager(JavaFileManager fileManager,
-			DynamicClassLoader cl) {
+	                                          DynamicClassLoader cl, CompiledCode[] compiledCode) {
 		super(fileManager);
 		this.cl = cl;
+		this.compiledCode = Arrays.stream(compiledCode).collect(Collectors.toMap(CompiledCode::getClassName, c -> c));
 	}
 
 	@Override
@@ -40,8 +39,11 @@ public JavaFileObject getJavaFileForOutput(
 			JavaFileObject.Kind kind, FileObject sibling) throws IOException {
 
 		try {
-			CompiledCode innerClass = new CompiledCode(className);
-			compiledCode.add(innerClass);
+			CompiledCode innerClass = compiledCode.get(className);
+			if (innerClass == null) {
+				innerClass = new CompiledCode(className);
+				compiledCode.put(className, innerClass);
+			}
 			cl.addCode(innerClass);
 			return innerClass;
 		} catch (Exception e) {