Java 理解类加载过程 -- 自定义加载器

 1 package com.xzlf.test;
 2 
 3 import java.io.ByteArrayOutputStream;
 4 import java.io.FileInputStream;
 5 import java.io.IOException;
 6 import java.io.InputStream;
 7 
 8 /**
 9  * 自定义类加载器
10  * 
11  * @author xzlf
12  *
13  */
14 public class MyClassLoader extends ClassLoader {
15     // 类加载器查找的根目录
16     private String rootDir;
17 
18     public MyClassLoader(String rootDir) {
19         super();
20         this.rootDir = rootDir;
21     }
22 
23     @Override
24     protected Class<?> findClass(String name) throws ClassNotFoundException {
25         // 查看 c 是否已经被加载
26         Class<?> c = findLoadedClass(name);
27         if(c != null) {
28             // 已经加载直接返回
29             return c;
30         }else {
31             // 没有被加载，先委派给父类加载器， 最终会委派到引导类加载器
32             ClassLoader parent = this.getParent();
33             try {
34                 c = parent.loadClass(name);
35             } catch (ClassNotFoundException e) {
36                 e.printStackTrace();
37             }
38             // 如果父类加载器已加载则直接返回，如果没有加载则使用自定加载器加载类的字节码文件
39             if(c != null) {
40                 return c;
41             }else {
42                 // 获取字节码文件
43                 byte[] classDatas = getClassData(name);
44                 if(classDatas == null) {
45                     throw new ClassNotFoundException();
46                 }else {
47                     //Converts an array of bytes into an instance of class
48                     c = defineClass(name, classDatas, 0, classDatas.length);
49                 }
50             }
51         }
52         return c;
53     }
54 
55     // 获取字节码文件
56     private byte[] getClassData(String name) {
57         //com.test.A --> f:/mycode/  com/test/A.class
58         String path = rootDir + "/" + name.replace(‘.‘, ‘/‘) + ".class";
59         
60         // IO 操作 返回字节码
61         InputStream is = null;
62         ByteArrayOutputStream bos = new ByteArrayOutputStream();
63         try {
64             is = new FileInputStream(path);
65             byte[] flush = new byte[1024];
66             int len;
67             while((len = is.read(flush)) != -1) {
68                 bos.write(flush, 0, len);
69             }
70             return bos.toByteArray();
71         } catch (Exception e) {
72             e.printStackTrace();
73         }finally {
74             if(bos != null) {
75                 try {
76                     bos.close();
77                 } catch (IOException e) {
78                     e.printStackTrace();
79                 }
80             }
81             if(is != null) {
82                 try {
83                     is.close();
84                 } catch (IOException e) {
85                     e.printStackTrace();
86                 }
87             }
88         }
89         return null;
90     }
91 }

 1 package com.xzlf.test;
 2 
 3 public class TestClassLoader {
 4     public static void main(String[] args) throws Exception {
 5         ClassLoader loader = TestClassLoader.class.getClassLoader();
 6         System.out.println(loader);//sun.misc.Launcher$AppClassLoader@73d16e93
 7 
 8         ClassLoader loader2 = Thread.currentThread().getContextClassLoader();
 9         System.out.println(loader2);//sun.misc.Launcher$AppClassLoader@73d16e93
10         
11         Thread.currentThread().setContextClassLoader(new MyClassLoader("f:/mycode"));
12         //com.xzlf.test.MyClassLoader@6d06d69c
13         System.out.println(Thread.currentThread().getContextClassLoader());
14         
15         Class<?> clz = Thread.currentThread().getContextClassLoader().loadClass("com.test.Welcome");
16         System.out.println(clz);//class com.test.Welcome
17         System.out.println(clz.getClassLoader());//com.xzlf.test.MyClassLoader@6d06d69c
18                 
19     }
20 }
21 
22 
23 class A{
24     
25 }