SimpleJavaFileObject如何导入自定义(由我创建)包?

时间:2015-12-13 14:34:00

标签: java import classloader javacompiler

我是第一次来这里......

任务是编译动态类(从字符串创建)和后续创建实例化。

问题在于导入我的自定义包/类(放置在不同的文件夹中)到动态类。例如:import controller.*;发生错误后:

/CompiledClass.java:4: error: package controller does not exist

但标准包(库)已经很好地导入,没有任何问题,编译以相同的方式结束。

我尝试通过发送带路径的optionList但通过JavaCompiler Api的getTask方法解决但没有成功。

那么如何纠正导入自定义包?

这是我的代码:

package view;

import controller.*;

import java.io.*;

import java.lang.reflect.Constructor;

import java.lang.reflect.Method;

import java.net.URI;

import java.security.SecureClassLoader;

import java.util.*;

import javax.tools.*;
import javax.tools.JavaCompiler.*;
import javax.tools.JavaFileObject.Kind;

public class CompileJavaManager {

public static void compileJavaFileObject(StringBuilder inputString, Writer jspOut) throws Exception {
    String className = "CompiledClass";
    String classMethod = "methodOf" + className;
    Class compiledClass = null;
    //Class<?> c = Class.forName(className);

    StringBuilder javaFileContents =
        new StringBuilder("" +
            //"package compiled;" +
            '\n' +
            "import java.io.*;\n" +
            "import java.util.*;\n" +
            "import controller.*;\n" +
            '\n' +
            "public class " + className +"{\n" +
            "   private Writer out = null;\n" +
            '\n' +
            "   public " + className +"(Writer out){\n" +
            "       this.out = out;\n" +
            //"       this.container = container;" +
            "   }" +
            "   public void " + classMethod + "() throws java.io.IOException{\n" +
                    //inputString +
            "   }\n" +
            "}\n");

    JavaCompiler compiler = ToolProvider.getSystemJavaCompiler();
    DiagnosticCollector diagnosticsCollector = new DiagnosticCollector();
    StandardJavaFileManager standardJavaFileManager = compiler.getStandardFileManager(diagnosticsCollector, null, null);
    SpecialClassLoader classLoader = new SpecialClassLoader();
    SpecialJavaFileManager fileManager = new SpecialJavaFileManager(standardJavaFileManager, classLoader);           

    List<String> optionList = new ArrayList<String>();
    optionList.addAll(Arrays.asList("-classpath", System.getProperty("java.class.path")));

    JavaObjectFromString javaObjectFromString = new JavaObjectFromString(className, javaFileContents.toString());

    Iterable fileObjects = Arrays.asList(javaObjectFromString);
    Iterable classes = null;           
    Writer out = new PrintWriter(jspOut);

    CompilationTask task = compiler.getTask(out, fileManager, diagnosticsCollector, optionList, classes, fileObjects);
    Boolean result = task.call();
    List<Diagnostic> diagnostics = diagnosticsCollector.getDiagnostics();

    if (result) {
        compiledClass = classLoader.findClass(className);
        Constructor<?> constructor = compiledClass.getConstructor(Writer.class);
        Object instance = constructor.newInstance(jspOut);
        //call the method, pass a null params 
        Method instanceMethod = compiledClass.getDeclaredMethod(classMethod, null);
        instanceMethod.invoke(instance, null);
        //System.out.println(instance);
    } else {
        // Compilation fails
        for (Diagnostic d : diagnostics) {
            System.out.println(d);
        }
    }
}

private static class JavaObjectFromString extends SimpleJavaFileObject {
    private String sourceCode = null;

    public JavaObjectFromString(String className, String sourceCode) throws Exception {
        super(URI.create("file:///" + className + Kind.SOURCE.extension), Kind.SOURCE);
        this.sourceCode = sourceCode;
    }

    @Override
    public CharSequence getCharContent(boolean ignoreEncodingErrors) throws IOException {
        return sourceCode;
    }

    @Override
    public OutputStream openOutputStream() {       
        throw new IllegalStateException();   
    }

    @Override
    public InputStream openInputStream() {       
        return new ByteArrayInputStream(sourceCode.getBytes());   
    }
}

private static class JavaObjectFromByteCode extends SimpleJavaFileObject {   
    private ByteArrayOutputStream baos;  

    public JavaObjectFromByteCode(String name) {       
        super(URI.create("byte:///" + name + Kind.CLASS.extension), Kind.CLASS);   
    }

    @Override
    public CharSequence getCharContent(boolean ignoreEncodingErrors) {       
        throw new IllegalStateException();   
    }

    @Override
    public OutputStream openOutputStream() {       
        baos = new ByteArrayOutputStream();       
        return baos;   
    }

    @Override
    public InputStream openInputStream() {       
        throw new IllegalStateException();   
    }

    public byte[] getBytes() {       
        return baos.toByteArray();   
    }
}

private static class SpecialClassLoader extends ClassLoader {   
    private Map<String,JavaObjectFromByteCode> m = new HashMap<String, JavaObjectFromByteCode>();

    protected Class<?> findClass(String name) throws ClassNotFoundException {       
        JavaObjectFromByteCode jobc = m.get(name);       
        if (jobc==null){           
            jobc = m.get(name.replace(".","/"));           
            if (jobc==null){               
                return super.findClass(name);           
            }       
        }       
        return defineClass(name, jobc.getBytes(), 0, jobc.getBytes().length);   
    }

    public void addClass(String name, JavaObjectFromByteCode jobc) {       
        m.put(name, jobc);   
    }
}

private static class SpecialJavaFileManager extends ForwardingJavaFileManager {   
    private SpecialClassLoader scl;   

    public SpecialJavaFileManager(StandardJavaFileManager sjfm, SpecialClassLoader scl) {       
        super(sjfm);       
        this.scl = scl;   
    }   

    public JavaFileObject getJavaFileForOutput(Location location, String name, JavaFileObject.Kind kind, FileObject sibling) throws IOException {       
        JavaObjectFromByteCode jobc = new JavaObjectFromByteCode(name);       
        scl.addClass(name, jobc);       
        return jobc;   
    }

    public ClassLoader getClassLoader(Location location) {       
        return scl;   
    }
}

}

提前致谢!

1 个答案:

答案 0 :(得分:0)

问题已经解决:

除了System.getProperty("java.class.path")之外,我还添加了路径到jar controller.jar。

以下是代码:

optionList.addAll(Arrays.asList("-classpath", System.getProperty("java.class.path") + ";path/to/your/controller.jar"));