【转】JDK 6动态编译类的一个例子

JDK6开始提供了动态编译的API，在许多应用场景都可以用得着，如动态加载（修改）服务、高性动态业务逻辑实现（用脚本或模板引擎实现效率满足不了需求）等都非常好用。

API对应的接口都在javax.tools包下面，常用编译方式有基于文本文件、内存字符串等，实际上基于URI的字节流都可以，也就是远程Java源代码也可以。

package demo;

import javax.tools.JavaCompiler;
import javax.tools.JavaFileObject;
import javax.tools.SimpleJavaFileObject;
import javax.tools.StandardJavaFileManager;
import javax.tools.ToolProvider;
import java.io.IOException;
import java.lang.reflect.Method;
import java.net.URI;
import java.util.Arrays;

public class CompileString {
public static void main(String[] args) throws Exception {
//通过系统工具提供者获得动态编译器
JavaCompiler compiler = ToolProvider.getSystemJavaCompiler();
//获得一个文件管理器，它的功能主要是提供所有文件操作的规则，
//如源代码路径、编译的classpath，class文件目标目录等，其相关属性都提供默认值
StandardJavaFileManager fileManager = compiler.getStandardFileManager(null, null, null);

StringObject so = new StringObject("demo.CalculatorTest", "package demo;" +
"  class CalculatorTest {" +
"	public int multiply(int multiplicand, int multiplier) {" +
"		System.out.println(multiplicand);" +
"		System.out.println(multiplier);" +
"		return multiplicand * multiplier;" +
"	}" +
"}");

JavaFileObject file = so;

// 指定 javac 的命令行参数
// 注意：最好将编译后的.class文件放到当前JVM实例的类路径上，否则加载不了
Iterable<String> options = Arrays.asList("-d", "E:\\workspace\\demo\\bin");
// 指定有哪些源文件需要被编译
Iterable<? extends JavaFileObject> files = Arrays.asList(file);

JavaCompiler.CompilationTask task = compiler.getTask(null, fileManager, null, options, null, files);

Boolean result = task.call();

if (result) {
System.out.println("编译成功");
Class<?> clazz = Class.forName("demo.CalculatorTest");

Object instance = clazz.newInstance();

Method m = clazz.getMethod("multiply", new Class[]{int.class, int.class});

Object[] o = new Object[]{1, 2};

Object returnObj =  m.invoke(instance, o);
System.out.println("方法返回的结果：" + returnObj.toString());
}
}
}

class StringObject extends SimpleJavaFileObject {
private String contents = null;

public StringObject(String className, String contents) throws Exception {
super(new URI(className), Kind.SOURCE);
this.contents = contents;
}

public CharSequence getCharContent(boolean ignoreEncodingErrors) throws IOException {
return contents;
}
}

PS：这里作者给的例子是返回一个int型的结果，所以，可以直接用

Object returnObj =  m.invoke(instance, o);
System.out.println("方法返回的结果：" + returnObj.toString());

来得到返回的结果。

如果，返回的结果集是一个对象，那么，就需要对返回的结果进行类型转换。

以List为例：

package demo;

import javax.tools.JavaCompiler;
import javax.tools.JavaFileObject;
import javax.tools.SimpleJavaFileObject;
import javax.tools.StandardJavaFileManager;
import javax.tools.ToolProvider;
import java.io.IOException;
import java.lang.reflect.Method;
import java.net.URI;
import java.util.Arrays;
import java.util.List;

public class CompileString {
public static void main(String[] args) throws Exception {
//通过系统工具提供者获得动态编译器
JavaCompiler compiler = ToolProvider.getSystemJavaCompiler();
//获得一个文件管理器，它的功能主要是提供所有文件操作的规则，
//如源代码路径、编译的classpath，class文件目标目录等，其相关属性都提供默认值
StandardJavaFileManager fileManager = compiler.getStandardFileManager(null, null, null);

StringObject so = new StringObject("demo.CalculatorTest",
"package demo;"
+"import java.util.List;"
+"import java.util.ArrayList;"
+"class CalculatorTest {"
+"	public List<String> multiply(String str1, String str2) {"
+"		System.out.println(str1);"
+"		System.out.println(str2);"
+"		List<String> result = new ArrayList<String>();"
+"		result.add(str1);"
+"		result.add(str2);"
+"		return result;"
+"	}"
+"}");

JavaFileObject file = so;

// 指定 javac 的命令行参数
// 注意：最好将编译后的.class文件放到当前JVM实例的类路径上，否则加载不了
Iterable<String> options = Arrays.asList("-d", "E:\\workspace\\demo\\bin");
// 指定有哪些源文件需要被编译
Iterable<? extends JavaFileObject> files = Arrays.asList(file);

JavaCompiler.CompilationTask task = compiler.getTask(null, fileManager, null, options, null, files);

Boolean result = task.call();

if (result) {
System.out.println("编译成功");
Class<?> clazz = Class.forName("demo.CalculatorTest");

Object instance = clazz.newInstance();

Method m = clazz.getMethod("multiply", new Class[]{String.class, String.class});

Object[] o = new Object[]{"1", "2"};

List returnObj =  (List) m.invoke(instance, o);

System.out.println("方法返回的结果集个数：" + returnObj.size());
}
}
}

class StringObject extends SimpleJavaFileObject {
private String contents = null;

public StringObject(String className, String contents) throws Exception {
super(new URI(className), Kind.SOURCE);
this.contents = contents;
}

public CharSequence getCharContent(boolean ignoreEncodingErrors) throws IOException {
return contents;
}
}