cn.hutool.core.compiler.JavaClassFileManager Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of hutool-all Show documentation

Hutool是一个小而全的Java工具类库，通过静态方法封装，降低相关API的学习成本，提高工作效率，使Java拥有函数式语言般的优雅，让Java语言也可以“甜甜的”。

There is a newer version: 5.8.34

package cn.hutool.core.compiler;

import cn.hutool.core.io.resource.FileObjectResource;
import cn.hutool.core.lang.ResourceClassLoader;
import cn.hutool.core.util.ClassLoaderUtil;
import cn.hutool.core.util.ObjectUtil;

import javax.tools.FileObject;
import javax.tools.ForwardingJavaFileManager;
import javax.tools.JavaFileManager;
import javax.tools.JavaFileObject;
import javax.tools.JavaFileObject.Kind;
import java.util.HashMap;
import java.util.Map;

/**
 * Java 字节码文件对象管理器
 *
 * 
 * 正常我们使用javac命令编译源码时会将class文件写入到磁盘中，但在运行时动态编译类不适合保存在磁盘中
 * 我们采取此对象来管理运行时动态编译类生成的字节码。
 * 
 *
 * @author lzpeng
 * @since 5.5.2
 */
class JavaClassFileManager extends ForwardingJavaFileManager {

	/**
	 * 存储java字节码文件对象映射
	 */
	private final Map classFileObjectMap = new HashMap<>();

	/**
	 * 加载动态编译生成类的父类加载器
	 */
	private final ClassLoader parent;

	/**
	 * 构造
	 *
	 * @param parent      父类加载器
	 * @param fileManager 字节码文件管理器
	 */
	protected JavaClassFileManager(ClassLoader parent, JavaFileManager fileManager) {
		super(fileManager);
		this.parent = ObjectUtil.defaultIfNull(parent, ClassLoaderUtil::getClassLoader);
	}

	/**
	 * 获得动态编译生成的类的类加载器
	 *
	 * @param location 源码位置
	 * @return 动态编译生成的类的类加载器
	 */
	@Override
	public ClassLoader getClassLoader(final Location location) {
		return new ResourceClassLoader<>(this.parent, this.classFileObjectMap);
	}

	/**
	 * 获得Java字节码文件对象
	 * 编译器编译源码时会将Java源码对象编译转为Java字节码对象
	 *
	 * @param location  源码位置
	 * @param className 类名
	 * @param kind      文件类型
	 * @param sibling   Java源码对象
	 * @return Java字节码文件对象
	 */
	@Override
	public JavaFileObject getJavaFileForOutput(final Location location, final String className, final Kind kind, final FileObject sibling) {
		final JavaFileObject javaFileObject = new JavaClassFileObject(className);
		this.classFileObjectMap.put(className, new FileObjectResource(javaFileObject));
		return javaFileObject;
	}

}