动态生成类并加载

转载自：Java运行时动态生成class的方法

Java是一门静态语言，一般，咱们须要的class在编译的时候就已经生成了，为何有时候咱们还想在运行时动态生成class呢？

由于在有些时候，咱们还真得在运行时为一个类动态建立子类。好比，编写一个ORM框架，如何得知一个简单的JavaBean是否被用户修改过呢？

以User为例：

public class User {
    private String id;
    private String name;

    public String getId() {
        return id;
    }

    public void setId(String id) {
        this.id = id;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }
}

其实UserProxy实现起来很简单，就是建立一个User的子类，覆写全部setXxx()方法，作个标记就能够了：

public class UserProxy extends User {
    private boolean dirty;

    public boolean isDirty() {
        return this.dirty;
    }

    public void setDirty(boolean dirty) {
        this.dirty = dirty;
    }

    @Override
    public void setId(String id) {
        super.setId(id);
        setDirty(true);
    }

    @Override
    public void setName(String name) {
        super.setName(name);
        setDirty(true);
    }
}

可是这个UserProxy就必须在运行时动态建立出来了，由于编译时ORM框架根本不知道User类。

如今问题来了，动态生成字节码，难度有多大？

若是咱们要本身直接输出二进制格式的字节码，在完成这个任务前，必须先认真阅读JVM规范第4章，详细了解class文件结构。估计读完规范后，两个月过去了。

因此，第一种方法，本身动手，从零开始建立字节码，理论上可行，实际上很难。

第二种方法，使用已有的一些能操做字节码的库，帮助咱们建立class。

目前，可以操做字节码的开源库主要有CGLib和Javassist两种，它们都提供了比较高级的API来操做字节码，最后输出为class文件。

好比CGLib，典型的用法以下：

Enhancer e = new Enhancer();
e.setSuperclass(...);
e.setStrategy(new DefaultGeneratorStrategy() {
    protected ClassGenerator transform(ClassGenerator cg) {
        return new TransformingGenerator(cg,
            new AddPropertyTransformer(new String[]{ "foo" },
                    new Class[] { Integer.TYPE }));
    }});
Object obj = e.create();

比本身生成class要简单，可是，要学会它的API仍是得花大量的时间，而且，上面的代码很难看懂对不对？

有木有更简单的方法？

有！

换一个思路，若是咱们能建立UserProxy.java这个源文件，再调用Java编译器，直接把源码编译成class，再加载进虚拟机，任务完成！

毕竟，建立一个字符串格式的源码是很简单的事情，就是拼字符串嘛，高级点的作法能够用一个模版引擎。

如何编译？

Java的编译器是javac，可是，在很早很早的时候，Java的编译器就已经用纯Java重写了，本身能编译本身，行业黑话叫“自举”。从Java 1.6开始，编译器接口正式放到JDK的公开API中，因而，咱们不须要建立新的进程来调用javac，而是直接使用编译器API来编译源码。

使用起来也很简单：

JavaCompiler compiler = ToolProvider.getSystemJavaCompiler();
int compilationResult = compiler.run(null, null, null, '/path/to/Test.java');

这么写编译是没啥问题，问题是咱们在内存中建立了Java代码后，必须先写到文件，再编译，最后还要手动读取class文件内容并用一个ClassLoader加载。

有木有更简单的方法？

有！

其实Java编译器根本不关心源码的内容是从哪来的，你给它一个String看成源码，它就能够输出byte[]做为class的内容。

因此，咱们须要参考Java Compiler API的文档，让Compiler直接在内存中完成编译，输出的class内容就是byte[]。

代码改造以下：

Map<String, byte[]> results;
JavaCompiler compiler = ToolProvider.getSystemJavaCompiler();
StandardJavaFileManager stdManager = compiler.getStandardFileManager(null, null, null);
try (MemoryJavaFileManager manager = new MemoryJavaFileManager(stdManager)) {
    JavaFileObject javaFileObject = manager.makeStringSource(fileName, source);
    CompilationTask task = compiler.getTask(null, manager, null, null, null, Arrays.asList(javaFileObject));
    if (task.call()) {
        results = manager.getClassBytes();
    }
}

上述代码的几个关键在于：

1. 用MemoryJavaFileManager替换JDK默认的StandardJavaFileManager，以便在编译器请求源码内容时，不是从文件读取，而是直接返回String；
2. 用MemoryOutputJavaFileObject替换JDK默认的SimpleJavaFileObject，以便在接收到编译器生成的byte[]内容时，不写入class文件，而是直接保存在内存中。

最后，编译的结果放在Map<String, byte[]>中，Key是类名，对应的byte[]是class的二进制内容。

为何编译后不是一个byte[]呢？

由于一个.java的源文件编译后可能有多个.class文件！只要包含了静态类、匿名类等，编译出的class确定多于一个。

如何加载编译后的class呢？

加载class相对而言就容易多了，咱们只须要建立一个ClassLoader，覆写findClass()方法：

class MemoryClassLoader extends URLClassLoader {

    Map<String, byte[]> classBytes = new HashMap<String, byte[]>();

    public MemoryClassLoader(Map<String, byte[]> classBytes) {
        super(new URL[0], MemoryClassLoader.class.getClassLoader());
        this.classBytes.putAll(classBytes);
    }

    @Override
    protected Class<?> findClass(String name) throws ClassNotFoundException {
        byte[] buf = classBytes.get(name);
        if (buf == null) {
            return super.findClass(name);
        }
        classBytes.remove(name);
        return defineClass(name, buf, 0, buf.length);
    }
}

除了写ORM用以外，还能干什么？

能够用它来作一个Java脚本引擎。实际上本文的代码主要就是参考了Scripting项目的源码。

完整的源码呢？

在这里：https://github.com/michaelliao/compiler，连Maven的包都给你准备好了！

也就200行代码吧！动态建立class不是梦