linkedList实现高亮逻辑

阿超

发布于 2022-08-21 15:01:05

3380

发布于 2022-08-21 15:01:05

这里是基于hutool的dfa查找法得到的结果集进行封装，不一定非要依赖hutool，可以自定义FoundWord对象，里面就只用到了一个起始下标，以及对应需要高亮的词汇

public static String highlight(String text, List<FoundWord> fondWords, UnaryOperator<String> highlightOperator) {
        return highlight(text, fondWords, highlightOperator, false);
    }

    public static String highlight(String text, List<FoundWord> fondWords, UnaryOperator<String> highlightOperator, boolean needSort) {
        if (CollUtil.isEmpty(fondWords)) {
            return text;
        }
        if (needSort) {
            fondWords = Steam.of(fondWords)
                    .sorted(Comparator.comparing(FoundWord::getStartIndex)
                            .thenComparingInt(w -> w.getFoundWord().length()))
                    .toList();
        }
        LinkedList<FoundWord> linkedList = new LinkedList<>();
        // 记录历史下标(当前拼接到整体string的哪个位置了)
        int lastIdx = 0;
        // 进行遍历所有结果(理论上已按照 startIndex和fondWord.length排序)
        for (int i = 0; i < fondWords.size(); i++) {
            FoundWord fondWord = fondWords.get(i);
            String word = null;
            // 如果本次匹配发现历史下标已经超过了当前匹配值，说明这次词语和上次词语下标重复或者是上次的一部分
            if (i > 0 && lastIdx > fondWord.getStartIndex()) {
                // 移除拼接的额外部分
                FoundWord last = Objects.requireNonNull(linkedList.pollLast());
                // 判断上次和这次的长度，如果上次长度大于这次长度，说明这次词语是上次的一部分
                lastIdx -= last.getWord().length();
                if (last.getWord().length() > fondWord.getWord().length()) {
                    // 直接使用上次的词汇以及重置历史下标
                    word = last.getWord();
                }
            } else {
                // 否则根据历史下标到当前词汇下标进行查找额外部分
                String partOne = text.substring(lastIdx, fondWord.getStartIndex());
                // 并且将历史下标往前推进
                lastIdx += partOne.length();
                // 将额外部分拼接到链表中
                linkedList.add(new FoundWord(partOne, partOne, lastIdx, -1));
            }
            // 获取本次需要高亮的词汇
            word = Opt.ofNullable(word).orElseGet(fondWord::getWord);
            // 历史下标往前推进
            lastIdx += word.length();
            // 执行高亮操作
            linkedList.add(new FoundWord(word, highlightOperator.apply(word), lastIdx, lastIdx + word.length()));
        }
        // 别忘了加上最后一截
        linkedList.add(new FoundWord(text.substring(lastIdx), text.substring(lastIdx), lastIdx, lastIdx + text.substring(lastIdx).length()));
        return Steam.of(linkedList).map(FoundWord::getFoundWord).join();
    }

使用：

WordTree tree = new WordTree();
tree.addWord("大");
tree.addWord("大土豆");
tree.addWord("土豆");
tree.addWord("刚出锅");
tree.addWord("出锅");
String text = "我有一颗大土豆，刚出锅的";
List<FoundWord> foundWords = tree.matchAllWords(text, -1, true, true);
String result = CommonUtils.highlight(text, foundWords, s -> "<span style='color:red'>" + s + "</span>");
System.out.println(result);

效果：