更新添加敏感词过滤

ruoyi-modules/ruoyi-system/src/main/java/util/BadWordsUtil.java

@@ -0,0 +1,147 @@
+package util;
+
+import org.apache.commons.io.FileUtils;
+import org.apache.commons.lang3.StringUtils;
+import java.io.File;
+import java.io.IOException;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+/**
+ * 敏感词审核工具类
+ * @author tjf
+ */
+
+public class BadWordsUtil {
+
+    public static final int WORDS_MAX_LENGTH = 10;
+
+    // 我们自己的敏感词库
+    public static final String BAD_WORDS_LIB_FILE_NAME = "badWord.txt";
+
+    //敏感词列表
+    public static Map<String, String>[] badWordsList = null;
+
+    //敏感词索引
+    public static Map<String, Integer> wordIndex = new HashMap<>();
+
+    /**
+
+     * 初始化敏感词库
+
+     */
+
+    public static void initbadWordsList() throws IOException {
+
+        if (badWordsList == null) {
+
+            badWordsList = new Map[WORDS_MAX_LENGTH];
+
+            for (int i = 0; i < badWordsList.length; i++) {
+
+                badWordsList[i] = new HashMap<>();
+
+            }
+
+        }
+
+        //敏感词词库所在目录，这里为txt文本，一个敏感词一行
+        String path = BadWordsUtil.class.getClassLoader()
+
+            .getResource(BAD_WORDS_LIB_FILE_NAME)
+
+            .getPath();
+
+        System.out.println(path);
+
+        List<String> words = FileUtils.readLines(new File(path), "UTF-8");
+
+        for (String w : words) {
+
+            if (StringUtils.isNotBlank(w)) {
+
+                //将敏感词按长度存入map
+
+                badWordsList[w.length()].put(w, "");
+
+                Integer index = wordIndex.get(w.substring(0, 1));
+
+                //生成敏感词索引，存入map
+
+                if (index == null) {
+
+                    index = 0;
+
+                }
+
+                int x = (int) Math.pow(2, w.length());
+
+                index = (index | x);
+
+                wordIndex.put(w.substring(0, 1), index);
+
+            }
+
+        }
+
+    }
+
+    /**
+     * 检索敏感词
+     *
+     * @param content 你是大傻叉
+     * @return
+     */
+
+    public static String searchBanWords(String content) {
+
+        if (badWordsList == null) {
+
+            try {
+
+                initbadWordsList();
+
+            } catch (IOException e) {
+
+                throw new RuntimeException(e);
+
+            }
+
+        }
+
+        for (int i = 0; i < content.length(); i++) {
+
+            Integer index = wordIndex.get(content.substring(i, i + 1));
+
+            int p = 0;
+
+            while ((index != null) && (index > 0)) {
+
+                p++;
+
+                index = index >> 1;
+
+                String sub = "";
+
+                if ((i + p) < (content.length() - 1)) {
+
+                    sub = content.substring(i, i + p);
+
+                } else {
+
+                    sub = content.substring(i);
+
+                }
+
+                if (((index % 2) == 1) && badWordsList[p].containsKey(sub)) {
+                    content = content.replace(sub, "**");
+                }
+            }
+
+        }
+
+        return content;
+
+    }
+}

ruoyi-modules/ruoyi-system/src/main/resources/badWord.txt

@@ -0,0 +1,4 @@
+sb
+SB
+傻逼
+自杀