手寫mvc之ClassLoadlistener

package com.xuchen.demo.listener;

import java.io.File;
import java.io.FileFilter;
import java.io.InputStream;
import java.net.JarURLConnection;
import java.net.URL;
import java.util.ArrayList;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.jar.JarEntry;
import java.util.jar.JarFile;

import javax.servlet.ServletContext;
import javax.servlet.ServletContextEvent;
import javax.servlet.ServletContextListener;

import org.dom4j.Document;
import org.dom4j.Element;
import org.dom4j.io.SAXReader;

import com.alibaba.fastjson.JSON;
import com.xuchen.demo.annotation.Service;
import com.xuchen.demo.util.StringUtils;

public class ClassLoadlistener implements ServletContextListener{
    
    private static final String INITPARAMETER = "contextConfigLocation";
    private List<String> classNames = new ArrayList<String>();
    private Map<String, Object> beanContries = new HashMap<String, Object>();
    @Override
    public void contextInitialized(ServletContextEvent event) {
        System.out.println("********** init ContextListener start *************");
        try {
            ServletContext servletContext = event.getServletContext();
            String webContextXml = servletContext.getInitParameter(INITPARAMETER).trim();
            String[] parameters = webContextXml.split(",");
            if (null != parameters) {
                for (String path : parameters) {
                    System.out.println(this.getClass().getName() + "{}" + StringUtils.trim(path));
                    String servletXmlPath = path.substring(path.lastIndexOf(":") + 1);
                    Document document = null;
                    SAXReader render = new SAXReader();
                    InputStream inputStream = this.getClassLoader().getResourceAsStream(servletXmlPath);
                    if (null == inputStream) {
                        System.out.println(this.getClass().getName() + " not find " + servletXmlPath);
                        return;
                    }
                    document = render.read(inputStream);
                    String xmlPacakScanPath = "//context-package-scan";
                    @SuppressWarnings("unchecked")
                    List<Element> nodelist = document.selectNodes(xmlPacakScanPath);
                    for (Element element : nodelist) {
                        Object packageName = element.getData();
                        contextScanPackage(String.valueOf(packageName));
                    }
                }
            }
            if (null != classNames && !classNames.isEmpty()) {
                servletContext.setAttribute("classNames", classNames);
                servletContext.setAttribute("beanContries", beanContries);
            }
            System.out.println("beanContries {} " + beanContries);
            System.out.println(this.getClass().getName() + " classNames " + JSON.toJSONString(classNames));
        } catch (Exception e) {
            System.out.println(this.getClass().getName() + " init ContextListener excetion " + e);
        }
        System.out.println("********** init ContextListener end   *************");
    }

    @Override
    public void contextDestroyed(ServletContextEvent sce) {
        
        
    }
    
    public ClassLoader getClassLoader() {
        return Thread.currentThread().getContextClassLoader();
    }
    
    private void contextScanPackage (String packageName) {
        try {
            Enumeration<URL> urlEnum = getClassLoader().getResources(packageName.replace(".", "/"));
            while(urlEnum.hasMoreElements()) {
                URL url = urlEnum.nextElement();
                String protocol = url.getProtocol();
                if ("file".equals(protocol)) {
                    String clazz = url.getPath().replaceAll("%20", "");
                    File[] files = new File(clazz).listFiles(new FileFilter() {
                        @Override
                        public boolean accept(File file) {
                            return (file.isFile() && file.getName().endsWith(".class") || file.isDirectory());
                        }
                    });
                    for (File file : files) {
                        String fileName = file.getName();
                        if (file.isDirectory()) {
                            contextScanPackage(packageName + "." + fileName);
                        } else {
                            String className = packageName + "." + fileName.substring(0, fileName.lastIndexOf("."));
                            if (!"".equals(className)) {
                                Class<?> claxx = Class.forName(className, false, getClassLoader());
                                if (claxx.isAnnotationPresent(Service.class)) {
                                    String aliasValue = claxx.getAnnotation(Service.class).value();
                                    System.out.println("aliasValue " + aliasValue);
                                    classNames.add(className);
                                }
                            }
                        }
                    }
                } else {
                    System.out.println("---------- now package is in a jar file.");
                    Map<String, Object> paraMap = new HashMap<String, Object>();
                    JarURLConnection jarConnect = (JarURLConnection) url.openConnection();
                    if (null != jarConnect) {
                        JarFile jarFile = jarConnect.getJarFile();
                        if (null != jarFile) {
                            Enumeration<JarEntry> entries = jarFile.entries();
                            if (null != entries) {
                                while (entries.hasMoreElements()) {
                                    JarEntry jarEntry = (JarEntry) entries.nextElement();
                                    String fileName = jarEntry.getName();
                                    if (fileName.endsWith(".class")) {
                                        fileName = jarEntry.getName().replace("/", ".");
                                        String claxxName = fileName.substring(0, fileName.lastIndexOf(".class"));
                                        Class<?> claxx = Class.forName(claxxName);
                                        if (claxx.isAnnotationPresent(Service.class)) {
                                            System.out.println("class 【" + claxxName + "】 has service @interface");
                                            String aliasValue = claxx.getAnnotation(Service.class).value();
                                            System.out.println("aliasValue " + aliasValue);
                                            classNames.add(claxxName);
                                            beanContries.put(claxxName, claxx.newInstance());
                                        }
                                    } else if (fileName.endsWith(".xml")) {
                                        System.out.println(getClass().getName() + " 我是xml文件" + fileName);
                                    }
                                }
                            }
                            
                        }
                    }
                }
            }
        } catch (Exception e) {
            System.out.println(this.getClass().getName() + " analasis object instance package exception "  + e);
            throw new RuntimeException("analasis object instance package exception", e);
        }
    }
    
    /**
     * 首字母小寫
     * @param name
     * @return
     */
    private String firstLowerName(String name) {
        name = name.substring(0, 1).toLowerCase() + name.substring(1);
       return  name;
    }
}