Java实现检查文章是否包含敏感词

导语

敏感词过滤是一个常见的需求，在很多应用场景中都需要对用户输入的内容进行敏感词过滤，以保证信息的安全性和用户体验。本文将介绍如何使用Java实现检查文章是否包含敏感词，并提供代码示例。

敏感词过滤的原理

敏感词过滤的原理一般是基于文本匹配的方法，即将待检查的文章与一批敏感词进行对比，如果文章中包含了任何一个敏感词，就判断为包含敏感词。

敏感词过滤的流程

下面是敏感词过滤的流程图：

flowchart TD
    A[开始] --> B(读取敏感词库)
    B --> C(加载敏感词)
    C --> D(从文章中提取词汇)
    D --> E(检查敏感词)
    E --> F(输出检查结果)
    F --> G[结束]

在流程图中，我们可以看到敏感词过滤的大致流程，下面将详细介绍每一步的实现。

读取敏感词库

首先，我们需要将敏感词库的内容读取到内存中，以便后续的检查。敏感词库可以是一个文件，也可以是一个数据库表。我们可以使用Java的文件读取或数据库查询的方式来实现。

下面是一个从文件中读取敏感词库的示例代码：

public List<String> readSensitiveWords(String filePath) {
    List<String> sensitiveWords = new ArrayList<>();
    try (BufferedReader reader = new BufferedReader(new FileReader(filePath))) {
        String line;
        while ((line = reader.readLine()) != null) {
            sensitiveWords.add(line);
        }
    } catch (IOException e) {
        e.printStackTrace();
    }
    return sensitiveWords;
}

加载敏感词

读取到敏感词库后，我们需要将敏感词进行加载，以便后续的检查。一种常见的方式是使用Trie树来存储敏感词，以提高检查的效率。

下面是一个使用Trie树加载敏感词的示例代码：

public void buildTrieTree(List<String> sensitiveWords) {
    for (String word : sensitiveWords) {
        TrieNode node = root;
        for (char c : word.toCharArray()) {
            if (!node.containsKey(c)) {
                node.put(c, new TrieNode());
            }
            node = node.get(c);
        }
        node.setEnd(true);
    }
}

class TrieNode {
    private Map<Character, TrieNode> children;
    private boolean isEnd;

    public TrieNode() {
        children = new HashMap<>();
        isEnd = false;
    }

    public boolean containsKey(char c) {
        return children.containsKey(c);
    }

    public TrieNode get(char c) {
        return children.get(c);
    }

    public void put(char c, TrieNode node) {
        children.put(c, node);
    }

    public boolean isEnd() {
        return isEnd;
    }

    public void setEnd(boolean end) {
        isEnd = end;
    }
}

检查敏感词

加载敏感词库后，我们就可以使用Trie树来检查文章中是否包含敏感词了。我们将文章中的每一个字符与Trie树进行匹配，如果匹配到了一个敏感词的结尾，就说明文章中包含敏感词。

下面是一个检查敏感词的示例代码：

public boolean checkSensitiveWords(String text) {
    TrieNode node = root;
    int start = 0;
    int position = 0;
    while (position < text.length()) {
        char c = text.charAt(position);
        if (node.containsKey(c)) {
            node = node.get(c);
            if (node.isEnd()) {
                return true;
            }
            position++;
        } else {
            node = root;
            start++;
            position = start;
        }
    }
    return false;
}

输出检查结果

最后，我们需要根据检查的结果来输出相应的信息，可以是直接返回结果，也可以是记录日志等。

下面是一个输出检查结果的示例代码：

public void