Java自定义类加载器

野性酷女 2022-09-12 01:59 176阅读 0赞

如何自定义类加载器，加载自定义路径的`.class`文件，然后执行定义的方法？

### 源码 ###

package com.hua.jvm.classloader;
    
    import java.io.ByteArrayOutputStream;
    import java.io.File;
    import java.io.FileInputStream;
    import java.io.IOException;
    import java.lang.reflect.Method;
    
    /** * <pre> * 程序目的：测试自定义类加载器。 * 1、继承ClassLoader，不破坏双亲委派机制，重写findClass方法 * 2、在自定义目录寻找.class字节码文件 * 3、使用反射机制，调用m()方法 * 4、查看该类的类加载器，是否为自定义的类加载器 * </pre> * created at 2021/9/17 * @author lerry */
    public class T006_MSBClassloader extends ClassLoader { 
    
    	public static void main(String[] args) throws Exception { 
    		// 自定义类加载器
    		ClassLoader classLoader = new T006_MSBClassloader();
    		// com.hua.jvm.classloader.Hello
    		// 加载.class文件
    		Class<?> helloCls = classLoader.loadClass("Hello");
    		// 使用反射获取实例
    		Object hello1 = helloCls.newInstance();
    		// 调用m方法
    		Method m = helloCls.getMethod("m");
    		m.invoke(hello1);
    		// 查看该类的类加载器
    		System.out.println(hello1.getClass().getClassLoader());
    		// 类加载器的父加载器
    		System.out.println(hello1.getClass().getClassLoader().getParent());
    	}
    
    	/** * 在自定义的路径中找类 * @param name * @return * @throws ClassNotFoundException */
    	@Override
    	public Class<?> findClass(String name) throws ClassNotFoundException { 
    		System.out.println("进入自定义的findClass()方法");
    		String path = String.format("%s%s", "/Users/lerry/Temp/", name.replaceAll("\\.", "/").concat(".class"));
    		File file = new File(path);
    
    		FileInputStream fis = null;
    		ByteArrayOutputStream baos = null;
    		try { 
    			fis = new FileInputStream(file);
    			System.out.println("类加载文件全路径为：" + file.getAbsolutePath());
    			baos = new ByteArrayOutputStream();
    
    			int length = 0;
    			byte[] byteBuffer = new byte[1024];
    			while (-1 != (length = fis.read(byteBuffer))) { 
    				baos.write(byteBuffer, 0, length);
    			}
    			// 会报错： Exception in thread "main" java.lang.ClassFormatError: Truncated class file
    // int b = 0;
    // while ((b = fis.read()) != 0) { 
    // baos.write(b);
    // }
    			byte[] bytes = baos.toByteArray();
    			return defineClass(name, bytes, 0, bytes.length);
    		}
    		catch (Exception e) { 
    			e.printStackTrace();
    		}
    		finally { 
    			try { 
    				fis.close();
    				baos.close();
    			}
    			catch (IOException e) { 
    				e.printStackTrace();
    			}
    		}
    		return super.findClass(name);
    	}
    }

执行结果：

进入自定义的findClass()方法
    类加载文件全路径为：/Users/lerry/Temp/Hello.class
    Hello,JVM
    com.hua.jvm.classloader.T006_MSBClassloader@1218025c
    sun.misc.Launcher$AppClassLoader@18b4aac2

可以看到，`Hello.class`是由类加载器`T006_MSBClassloader`加载的。它的父加载器是：`AppClassLoader`。

根据双亲委派机制，如果父加载器可以加载类，则不会再交给子加载器加载。  
为了证明，改源码的1个地方：

// 加载.class文件
    Class<?> helloCls = classLoader.loadClass("com.hua.jvm.classloader.Hello");

修改后执行结果如下：

Hello,JVM
    sun.misc.Launcher$AppClassLoader@18b4aac2
    sun.misc.Launcher$ExtClassLoader@2f2c9b19

并没有输出：

进入自定义的findClass()方法

说明，当父加载器可以加载该类时，就不会交由子加载器加载。

`/Users/lerry/Temp/`路径下的文件内容如下：

/** * created at 2021-09-17 08:11 * @author lerry */
    public class Hello { 
    	public static void main(String[] args) { 
    		System.out.println("Hello,JVM");
    	}
    
    	public void m() { 
    		System.out.println("Hello,JVM");
    	}
    }

需要使用`javac Hello.java`编译生成`Hello.class`字节码文件。