Java tutorial
/* * APDPlat - Application Product Development Platform * Copyright (c) 2013, ??, yang-shangchuan@qq.com * * This program is free software: you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation, either version 3 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program. If not, see <http://www.gnu.org/licenses/>. */ package org.apdplat.superword.tools; import org.apache.commons.lang.StringUtils; import org.apdplat.superword.model.Word; import org.jsoup.Connection; import org.jsoup.Jsoup; import org.jsoup.nodes.Document; import org.jsoup.nodes.Element; import org.jsoup.select.Elements; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import java.io.*; import java.nio.file.FileSystem; import java.nio.file.*; import java.nio.file.attribute.BasicFileAttributes; import java.util.*; import java.util.concurrent.ConcurrentHashMap; import java.util.concurrent.atomic.AtomicInteger; import java.util.stream.Collectors; /** * ????? * ??????? * ??????63789????HTML?origin_html_youdao.zip * ?http://pan.baidu.com/s/1pJH4ugj * @author ?? */ public class WordClassifierForYouDao { private WordClassifierForYouDao() { } private static final Logger LOGGER = LoggerFactory.getLogger(WordClassifierForYouDao.class); private static final String YOUDAO = WordLinker.YOUDAO; private static final String TYPE_CSS_PATH = "span.via.rank"; private static final String UNFOUND_CSS_PATH = "div.error-typo h4"; private static final String ACCEPT = "text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8"; private static final String ENCODING = "gzip, deflate"; private static final String LANGUAGE = "zh-cn,zh;q=0.8,en-us;q=0.5,en;q=0.3"; private static final String CONNECTION = "keep-alive"; private static final String HOST = "dict.youdao.com"; private static final String REFERER = "http://dict.youdao.com/"; private static final String USER_AGENT = "Mozilla/5.0 (Macintosh; Intel Mac OS X 10.10; rv:36.0) Gecko/20100101 Firefox/36.0"; private static final Set<String> NOT_FOUND_WORDS = Collections.newSetFromMap(new ConcurrentHashMap<>()); private static final Set<String> ORIGIN_HTML = Collections.newSetFromMap(new ConcurrentHashMap<>()); private static final AtomicInteger COUNT = new AtomicInteger(); public static void classify(Set<Word> words) { LOGGER.debug("??" + words.size()); AtomicInteger i = new AtomicInteger(); Map<String, List<String>> data = new ConcurrentHashMap<>(); words.parallelStream().forEach(word -> { if (i.get() % 1000 == 999) { save(data); } showStatus(data, i.incrementAndGet(), words.size(), word.getWord()); String html = getContent(word.getWord()); //LOGGER.debug("?HTML" +html); int times = 0; while (StringUtils.isNotBlank(html) && html.contains("??ip?")) { //IP? DynamicIp.toNewIp(); html = getContent(word.getWord()); if (++times > 2) { break; } } if (StringUtils.isNotBlank(html)) { parse(word.getWord(), html, data); if (!NOT_FOUND_WORDS.contains(word.getWord())) { ORIGIN_HTML.add(word.getWord() + "??" + html); } } else { NOT_FOUND_WORDS.add(word.getWord()); } }); //? save(data); LOGGER.debug("??" + words.size()); } public static void parse(String path) { if (path.endsWith(".zip")) { parseZip(path); } else if (Files.isDirectory(Paths.get(path))) { parseDir(path); } else { parseFile(path); } } public static void parseDir(String dir) { LOGGER.info("?" + dir); try { Files.walkFileTree(Paths.get(dir), new SimpleFileVisitor<Path>() { @Override public FileVisitResult visitFile(Path file, BasicFileAttributes attrs) throws IOException { parseFile(file.toFile().getAbsolutePath()); return FileVisitResult.CONTINUE; } }); } catch (IOException e) { LOGGER.error("?", e); } } public static void parseZip(String zipFile) { LOGGER.info("?ZIP" + zipFile); try (FileSystem fs = FileSystems.newFileSystem(Paths.get(zipFile), WordClassifierForYouDao.class.getClassLoader())) { for (Path path : fs.getRootDirectories()) { LOGGER.info("?" + path); Files.walkFileTree(path, new SimpleFileVisitor<Path>() { @Override public FileVisitResult visitFile(Path file, BasicFileAttributes attrs) throws IOException { LOGGER.info("?" + file); // ? Path temp = Paths.get("target/origin-html-temp.txt"); Files.copy(file, temp, StandardCopyOption.REPLACE_EXISTING); parseFile(temp.toFile().getAbsolutePath()); return FileVisitResult.CONTINUE; } }); } } catch (Exception e) { LOGGER.error("?", e); } } public static void parseFile(String file) { LOGGER.info("?" + file); try (BufferedReader reader = new BufferedReader( new InputStreamReader(new BufferedInputStream(new FileInputStream(file))))) { Map<String, List<String>> data = new ConcurrentHashMap<>(); String line = null; while ((line = reader.readLine()) != null) { parse(line, data); } save(data); } catch (IOException e) { LOGGER.error("?", e); } } public static void parse(String html, Map<String, List<String>> data) { LOGGER.debug("html:" + html); String[] attr = html.split("??"); if (attr == null || attr.length != 2) { LOGGER.error( "?'??'???????" + html); return; } String word = attr[0]; LOGGER.info(COUNT.incrementAndGet() + "????" + word); String htm = attr[1]; parse(word, htm, data); } public static void showStatus(Map<String, List<String>> data, int current, int total, String word) { LOGGER.debug("?? " + current + "/" + total + " ? " + current / (float) total * 100 + "% " + word); data.entrySet().forEach(e -> { LOGGER.debug(e.getKey() + "\t" + e.getValue().size()); }); } public static synchronized void save(Map<String, List<String>> data) { LOGGER.info("??"); data.keySet().forEach(key -> { try { String path = "src/main/resources/word_" + key + ".txt"; LOGGER.info("??" + path); List<String> existWords = Files.readAllLines(Paths.get(path)); Set<String> allWords = new HashSet<>(); existWords.forEach(line -> { String[] attr = line.split("\\s+"); if (attr != null) { String w = ""; if (attr.length == 1) { w = attr[0]; } if (attr.length == 2) { w = attr[1]; } allWords.add(w); } }); if (data.get(key) != null) { allWords.addAll(data.get(key)); } AtomicInteger i = new AtomicInteger(); List<String> list = allWords.stream().sorted().map(word -> i.incrementAndGet() + "\t" + word) .collect(Collectors.toList()); Files.write(Paths.get(path), list); if (data.get(key) != null) { data.get(key).clear(); } existWords.clear(); allWords.clear(); list.clear(); } catch (Exception e) { LOGGER.error("??", e); } }); data.clear(); try { if (!NOT_FOUND_WORDS.isEmpty()) { String path = "src/main/resources/word_not_found.txt"; LOGGER.info("??" + path); AtomicInteger i = new AtomicInteger(); //NOT_FOUND_WORDS List<String> list = NOT_FOUND_WORDS.stream().sorted().map(word -> i.incrementAndGet() + "\t" + word) .collect(Collectors.toList()); Files.write(Paths.get(path), list); list.clear(); } //?HTML if (!ORIGIN_HTML.isEmpty()) { String path = "src/main/resources/origin_html_" + System.currentTimeMillis() + ".txt"; LOGGER.info("??" + path); Files.write(Paths.get(path), ORIGIN_HTML); ORIGIN_HTML.clear(); } } catch (Exception e) { LOGGER.error("??", e); } } public static String getContent(String word) { String url = YOUDAO + word + "&renovate=" + (new Random(System.currentTimeMillis()).nextInt(899999) + 100000); LOGGER.debug("url:" + url); Connection conn = Jsoup.connect(url).header("Accept", ACCEPT).header("Accept-Encoding", ENCODING) .header("Accept-Language", LANGUAGE).header("Connection", CONNECTION).header("Referer", REFERER) .header("Host", HOST).header("User-Agent", USER_AGENT).timeout(60000).ignoreContentType(true); String html = ""; try { html = conn.post().html(); html = html.replaceAll("[\n\r]", ""); } catch (Exception e) { LOGGER.error("?URL" + url + "?", e); } return html; } public static void parse(String word, String html, Map<String, List<String>> data) { Document doc = Jsoup.parse(html); Elements es = doc.select(TYPE_CSS_PATH); for (Element e : es) { String types = e.text(); LOGGER.debug("?" + types); for (String type : types.split("\\s+")) { if (StringUtils.isNotBlank(type)) { data.putIfAbsent(type, new ArrayList<>()); data.get(type).add(word); } } } es = doc.select(UNFOUND_CSS_PATH); for (Element e : es) { String notFound = e.text(); LOGGER.debug("?" + notFound); if (StringUtils.isNotBlank(notFound) && (notFound.contains("?") || notFound.contains("??"))) { NOT_FOUND_WORDS.add(word); } } } public static void main(String[] args) { //Set<Word> words = new HashSet<>(); //words.add(new Word("time", "")); //words.add(new Word("yangshangchuan", "")); //classify(words); //classify(WordSources.getAll()); //parse("src/main/resources/origin_html_1449054801557.txt"); //origin_html_youdao.zip?63789?????HTML??http://pan.baidu.com/s/1pJH4ugj parse("/Users/apple/?/origin_html_youdao.zip"); } }