java 自制类加载器的简单实现

package com.xiaomo.reflex;


import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.lang.reflect.Method;


public class CompileClassLoader extends ClassLoader{


//读取一个文件的内容
private byte[] getBytes(String filename)throws IOException{
File file = new File(filename);
long len = file.length();
byte[] raw = new byte[(int)len];
try(
FileInputStream fin = new FileInputStream(file);
){
//一次读取Class文件的全部二进制数据
int r = fin.read(raw);
if(r != len){
throw new IOException("无法读取全部文件："+r+" != "+len);
}


return raw;
}
}

private boolean compile(String javaFile) throws IOException{
System.out.println("CompileClassLoader:正在编译 "+javaFile+"...");
//调用系统的javac命令
Process p = Runtime.getRuntime().exec("javac "+javaFile);
try{
p.waitFor();
}catch (InterruptedException e) {
System.out.println(e);
}
//获取javac线程的退出值
int ret = p.exitValue();
return ret == 0;
}

protected Class<?> findClass(String name) throws ClassNotFoundException{
Class clazz = null;
//将包路径中的点（.）替换成斜线(/)
String fileStub = name.replace(".", "/");
String javaFilename = fileStub + ".java";
String classFilename = fileStub + ".class";
File javaFile = new File(javaFilename);
File classFile = new File(classFilename);
//当指定的java源文件存在，且class文件不存在，或者Java源文件的修改时间比class文件的修改时间更晚时，重新编译
if(javaFile.exists()&&(!classFile.exists()||javaFile.lastModified()>classFile.lastModified())){
try{
//如果编译失败，或者改class文件不存在
if(!compile(javaFilename)||!classFile.exists()){
throw new ClassNotFoundException("ClassNotFoundException:"+javaFilename);
}
}catch (IOException e) {
e.printStackTrace();
}
}
//如果class文件存在，系统负责将该文件转换成class对象
if(classFile.exists()){
try{
//将class文件的二进制数据读入数组
byte[] raw = getBytes(classFilename);
//调用classloader的defineclass方法将二进制数据转换成class对象
clazz = defineClass(name, raw, 0, raw.length);
}catch(IOException e){
e.printStackTrace();
}
}
if(clazz==null){
throw new ClassNotFoundException(name);
}
return clazz;
}

public static void main(String[] args) throws Exception {
//如果运行该程序时没有参数，既没有目标类
if(args.length<1){
System.out.println("缺少目标类，请按如下格式运行java源文件：");
System.out.println("java CompileClassLoader ClassName");
}
//第一个参数是需要运行的类
String progClass = args[0];
//剩下的参数将作为运行目标类的参数
//将这些参数复制到一个新数组中
String[] progArgs = new String[args.length-1];
System.arraycopy(args, 1, progArgs, 0, progArgs.length);
CompileClassLoader cc1 = new CompileClassLoader();
//加载需要运行的类
Class<?> clazz = cc1.loadClass(progClass);
//获取需要运行的类的主要方法
Method main = clazz.getMethod("main",(new String[0]).getClass());
Object argsArray[] = {progArgs};
main.invoke(null, argsArray);
}


}