admin
2022-04-16 04f09e52ffd4681bdfd85e51acd3da0d1280c3d3
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
package com.yeshi.buwan.util;
 
import com.yeshi.buwan.vo.baidu.BaiDuCPUTypeVO;
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.io.InputStream;
import java.util.ArrayList;
import java.util.List;
 
public class BaiDuCPUUtil {
 
    /**
     * 获取咨询的所有类型
     *
     * @return
     */
    public static List<BaiDuCPUTypeVO> getNewsTotalTypes() {
        InputStream inputStream = BaiDuCPUUtil.class.getClassLoader().getResourceAsStream("baidu/news.xml");
        return parse(inputStream, "total");
    }
 
    /**
     * 获取咨询安全类型
     *
     * @return
     */
    public static List<BaiDuCPUTypeVO> getNewsSafeTypes() {
 
        InputStream inputStream = BaiDuCPUUtil.class.getClassLoader().getResourceAsStream("baidu/news.xml");
        return parse(inputStream, "safe");
    }
 
    /**
     * 获取视频的所有类型
     *
     * @return
     */
    public static List<BaiDuCPUTypeVO> getVideosTotalTypes() {
        InputStream inputStream = BaiDuCPUUtil.class.getClassLoader().getResourceAsStream("baidu/videos.xml");
        return parse(inputStream, "total");
    }
 
    /**
     * 获取视频安全类型
     *
     * @return
     */
    public static List<BaiDuCPUTypeVO> getVideoSafeTypes() {
 
        InputStream inputStream = BaiDuCPUUtil.class.getClassLoader().getResourceAsStream("baidu/videos.xml");
        return parse(inputStream, "safe");
    }
 
    private static List<BaiDuCPUTypeVO> parse(InputStream inputStream, String startKey) {
        List<BaiDuCPUTypeVO> typeVOList = new ArrayList<>();
        SAXParserFactory factory = SAXParserFactory.newInstance();
        SAXParser saxParser = null;
        try {
            saxParser = factory.newSAXParser();
 
            saxParser.parse(inputStream, new DefaultHandler() {
 
                String tagName = "";
                BaiDuCPUTypeVO vo = null;
                boolean start = false;
 
                @Override
                public void startElement(String uri, String localName, String qName, Attributes attributes) throws SAXException {
                    tagName = qName;
                    if (qName.equalsIgnoreCase(startKey)) {
                        typeVOList.clear();
                        start = true;
                    } else if (qName.equalsIgnoreCase("name")) {
 
                    } else if (qName.equalsIgnoreCase("id")) {
 
                    } else if (qName.equalsIgnoreCase("item")) {
                        vo = new BaiDuCPUTypeVO();
                    }
                }
 
                @Override
                public void endElement(String uri, String localName, String qName) throws SAXException {
                    super.endElement(uri, localName, qName);
                    tagName = "";
                    if (qName.equalsIgnoreCase(startKey)) {
                        start = false;
                    } else if (qName.equalsIgnoreCase("item") && vo != null) {
                        if (start)
                            typeVOList.add(vo);
                    }
                }
 
                @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 (tagName.equalsIgnoreCase("name")) {
                        vo.setName(str);
                    } else if (tagName.equalsIgnoreCase("id")) {
                        vo.setId(Integer.parseInt(str));
                    }
                }
            });
 
        } catch (ParserConfigurationException e) {
            e.printStackTrace();
        } catch (SAXException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
        return typeVOList;
    }
 
 
}