（八）——WebHandler

package my.tomcat2;

import org.xml.sax.Attributes;
import org.xml.sax.SAXException;
import org.xml.sax.helpers.DefaultHandler;

import javax.xml.parsers.ParserConfigurationException;
import javax.xml.parsers.SAXParser;
import javax.xml.parsers.SAXParserFactory;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

/**
 * 在解析一个node节点结束标签(例如</name>)时，会触发endElement和characters方法，要在endElement方法内，将currentTag = null .
 * 这里是 解析xml 文档所需的一个 处理器，这里不懂的同学可以学习一下 Java 如何解析 xml
 */
public class WebHandler extends DefaultHandler {
    //这是用来，存放 web.xml 中<servlet></servlet>里的信息的，因为多条信息，所以将其存入 List 中
    private List<Entity> entities;
    //这是用来，存放 web.xml 中<servlet>...</servlet>里的信息的
    private Entity entity;
    //这里用来，存放 web.xml 中<servlet-mapping>...</servlet-mapping>里的信息的
    private Mapping mapping;
    //同理
    private List<Mapping> mappings;
    //这个是用来在  public void characters(char[] ch, int start, int length)，执行这个方法的时候，需要知道他的标签值，因为我们只是要存储<servlet-name>...</servlet-name>
    //<servlet-class>...</servlet-class> 和 <servlet-name>...</servlet-name><url-pattern>...</url-pattern>，需要判断，在这里至于为什么 在endElement()是需要将flag = null，请看类之前的注释
    private String flag;
    //判断是 servlet 还是 servlet-mapping
    private boolean isMapping;

    @Override
    public void startDocument() throws SAXException {
        super.startDocument();
        //文档开始，进行初始化
        entities = new ArrayList<>();
        mappings = new ArrayList<>();
        System.out.println("文档开始");
    }

    @Override
    public void startElement(String uri, String localName, String qName, Attributes attributes) throws SAXException {
        super.startElement(uri, localName, qName, attributes);
        //元素开始，判断
        System.out.println("元素开始" + qName);
        if(null != qName){
            flag = qName;

            if(qName.equals("servlet")){
                isMapping = false;
                entity = new Entity();
            }else if(qName.equals("servlet-mapping")){
                isMapping = true;
                mapping = new Mapping();
            }
        }
    }

    @Override
    public void characters(char[] ch, int start, int length) throws SAXException {
        super.characters(ch, start, length);
        //元素
        String str = new String(ch, start, length);
        //开始 存储 信息了
        if(null != flag){
            if(isMapping){
                if(flag.equals("servlet-name")){
                    mapping.setServletName(str);
                }else if(flag.equals("url-pattern")){
                    mapping.getUrlList().add(str);
                }
            }
            if(!isMapping){
                if(flag.equals("servlet-name")){
                    entity.setServletName(str);
                }else if(flag.equals("servlet-class")){
                    entity.setServletClass(str);
                }
            }
        }
    }

    @Override
    public void endElement(String uri, String localName, String qName) throws SAXException {
        super.endElement(uri, localName, qName);
        //元素结束，将 对象 add进 List中去
        if(null != qName){
            if(qName.equals("servlet")){
                entities.add(entity);
            }else if(qName.equals("servlet-mapping")){
                mappings.add(mapping);
            }
        }

        flag = null;
    }

    @Override
    public void endDocument() throws SAXException {
        super.endDocument();
        //文档结束
    }

/*    public static void main(String[] args) throws ParserConfigurationException, SAXException, IOException {
        SAXParserFactory saxParserFactory = SAXParserFactory.newInstance();
        SAXParser saxParser = saxParserFactory.newSAXParser();

        WebHandler webHandler = new WebHandler();
        saxParser.parse(Thread.currentThread().getContextClassLoader().getResourceAsStream("web.xml"), webHandler);
        System.out.println(webHandler.getEntities());
        for(Entity temp : webHandler.getEntities()){
            System.out.println(temp.toString());
        }
    }*/

    //你需要 在 WebApp 中将这里的 List 存入到 它里面的 Map 中去
    public List<Entity> getEntities() {
        return entities;
    }

    public List<Mapping> getMappings() {
        return mappings;
    }
}