在上一章中我們加載Servlet用的是URLClassLoader恩够，在這一章中我們使用自定義的類加載器來替換URLClassLoader。關(guān)于類加載器的文章請(qǐng)參考這里：深入探討 Java 類加載器蜂桶。這篇文章寫得很棒。
具體的代碼如下：

package com.zkn.imitate.tomcat.secondchapter.first;

import com.zkn.imitate.tomcat.secondchapter.Request;
import com.zkn.imitate.tomcat.secondchapter.Response;
import com.zkn.imitate.tomcat.secondchapter.StaticResourceProcessor;
import com.zkn.imitate.tomcat.utils.Constants;
import com.zkn.imitate.tomcat.utils.StringUtil;

import java.io.IOException;
import java.net.InetAddress;
import java.net.ServerSocket;
import java.net.Socket;

/**
 * Created by wb-zhangkenan on 2016/12/29.
 */
public class HttpServer02 {

    public static void main(String[] args){

        await();
    }

    private static void await() {

        ServerSocket serverSocket = null;
        try {
            boolean shutDown = false;
            serverSocket = new ServerSocket(8004,1, InetAddress.getByName("127.0.0.1"));
            while (!shutDown){
                Socket socket = serverSocket.accept();
                Request request = new Request(socket.getInputStream());
                request.parseRequest();
                Response response = new Response(socket.getOutputStream());
                String uri = request.getUri();
                if(uri !=null && uri.startsWith("/favicon.ico")){

                }else if(!StringUtil.isEmpty(uri) && uri.startsWith("/static/")){
                    StaticResourceProcessor resouce = new StaticResourceProcessor();
                    resouce.process(request,response);
                }else{
                    ServletProcessor01 servletProcessor = new ServletProcessor01();
                    servletProcessor.process(request,response);
                }
                socket.close();
                shutDown = Constants.SHUT_DOWN.equals(request.getUri());
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

public class ServletProcessor01 {

    public void process(Request request, Response response) {

        String str = request.getUri();
        if(!StringUtil.isEmpty(str) && str.lastIndexOf("/") >= 0){
            str = str.substring(str.lastIndexOf("/")+1);
        }else {
            return;
        }
        FileSystemClassLoader fileClassLoader =
                new FileSystemClassLoader(Constants.WEB_ROOT);
        try {
            Class clazz = fileClassLoader.findClass(str);
            Servlet servlet = (Servlet) clazz.newInstance();
            servlet.service(request,response);
        } catch (ClassNotFoundException e) {
            e.printStackTrace();
        } catch (InstantiationException e) {
            e.printStackTrace();
        } catch (IllegalAccessException e) {
            e.printStackTrace();
        } catch (ServletException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

自定義的類加載器：

package com.zkn.imitate.tomcat.utils;

import java.io.*;
import java.util.Arrays;

/**
 * Created by wb-zhangkenan on 2017/1/3.
 */
public class FileSystemClassLoader extends ClassLoader{
    /**
     * 應(yīng)用根路徑
     */
    private String rootDir;

    public FileSystemClassLoader(String rootDir) {
        this.rootDir = rootDir;
    }

    public FileSystemClassLoader() {
    }

    @Override
    public Class<?> findClass(String name) throws ClassNotFoundException {
        byte[] bytes = getByteData(name);
        System.out.println(Arrays.toString(bytes));
        return defineClass(null,bytes,0,bytes.length);
    }

    private byte[] getByteData(String path){

        String className = getClassName(path);
        byte[] bytes = new byte[1024];
        try {
            FileInputStream fis = new FileInputStream(className);
            //ByteArrayOutputStream內(nèi)部有一個(gè)數(shù)組用來保存讀取的字節(jié)流
            ByteArrayOutputStream baiStream = new ByteArrayOutputStream();
            int flag = 0;
            while ((flag = fis.read(bytes)) != -1){
                /**
                 * 這里需要注意：寫入的時(shí)候，寫入的范圍一定是0褂萧，flag葵萎。
                 * 原因是：有可能讀取的bytes不夠1024個(gè)字節(jié)，這個(gè)時(shí)候如果不寫入讀取范圍的話谎痢，
                 *       則會(huì)把bytes中存留的上次讀取的數(shù)據(jù)也寫入到ByteArrayOutputStream中节猿。
                 *       這樣在defineClass的時(shí)候會(huì)出現(xiàn)異常漫雕。
                 *       異常信息如下：
                 *Exception in thread "main" java.lang.ClassFormatError:
                 *      Extra bytes at the end of class file FirstServlet
                 */
                baiStream.write(bytes,0,flag);
            }
            return baiStream.toByteArray();
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
        return bytes;
    }

    private String getClassName(String className){

        if(StringUtil.isEmpty(className)){
             throw new RuntimeException("Class名稱不能為空");
        }
        return rootDir+ File.separatorChar+className.replace('.',File.separatorChar)+".class";
    }

}

TomCat中的應(yīng)用加載器是WebappLoader浸间。WebappLoader創(chuàng)建一個(gè)org.apache.catalina.loader.WebappClassLoader類的實(shí)例作為它的類加載器魁蒜。有時(shí)間的話我會(huì)寫一個(gè)關(guān)于WebAppLoader的一個(gè)分析。