
基于DFA算法实现文章敏感词过滤
发布日期:2021-05-06 19:32:57
浏览次数:11
分类:技术文章
本文共 11312 字,大约阅读时间需要 37 分钟。
最近公司要出一个论坛系统
因为最近貌似xxx查的也比较严,所以图片和文字安全一样要注意 其中文字就涉及到敏感字过滤的问题 目前大概流传两种解决办法: 1、利用分词器分词实现过滤 比如见得比较多的 IKAnalyzer 2、利用一些效率优秀的算法 比如DFA算法 DFA算法 中文称作有穷自动机 解释:有一个有限状态集合和一些从一个状态通向另一个状态的边,每条边上标记有一个符号,其中一个状态是初态,某些状态是终态。 它是是通过event和当前的state得到下一个state,即event+state=nextstate。 至于为什么说DFA算法比较优秀,其实他也没做什么运算,有的只是状态的变化 ps:那好奇的宝宝说肯定就有无穷,详情见xxxdu。。。 下面用画图表示 比如现在有几个词需要被过滤 我是好人、我是坏蛋、我是学生package com.hqjl.communityserv.filter;import com.hqjl.communityserv.util.SensitiveWordInit;import java.io.File;import java.util.HashSet;import java.util.Iterator;import java.util.Map;import java.util.Set;/** * @author chunying * @Date: 2019/9/26 0026 */public class SensitivewordFilter { @SuppressWarnings("rawtypes") private Map sensitiveWordMap = null; public static int minMatchTYpe = 1; //最小匹配规则 public static int maxMatchType = 2; //最大匹配规则 /** * 构造函数,初始化敏感词库 */ public SensitivewordFilter(File file){ sensitiveWordMap = new SensitiveWordInit().initKeyWord(file); } /** * 判断文字是否包含敏感字符 * @param txt 文字 * @param matchType 匹配规则 1:最小匹配规则,2:最大匹配规则 * @return 若包含返回true,否则返回false */ public boolean isContaintSensitiveWord(String txt,int matchType){ boolean flag = false; for(int i = 0 ; i < txt.length() ; i++){ int matchFlag = this.CheckSensitiveWord(txt, i, matchType); //判断是否包含敏感字符 if(matchFlag > 0){ //大于0存在,返回true flag = true; } } return flag; } /** * 获取文字中的敏感词 * @param txt 文字 * @param matchType 匹配规则 1:最小匹配规则,2:最大匹配规则 * @return */ public SetgetSensitiveWord(String txt , int matchType){ Set sensitiveWordList = new HashSet (); for(int i = 0 ; i < txt.length() ; i++){ int length = CheckSensitiveWord(txt, i, matchType); //判断是否包含敏感字符 if(length > 0){ //存在,加入list中 sensitiveWordList.add(txt.substring(i, i+length)); i = i + length - 1; //减1的原因,是因为for会自增 } } return sensitiveWordList; } /** * 替换敏感字字符 * @param txt * @param matchType * @param replaceChar 替换字符,默认* */ public String replaceSensitiveWord(String txt,int matchType,String replaceChar){ String resultTxt = txt; Set set = getSensitiveWord(txt, matchType); //获取所有的敏感词 Iterator iterator = set.iterator(); String word = null; String replaceString = null; while (iterator.hasNext()) { word = iterator.next(); replaceString = getReplaceChars(replaceChar, word.length()); resultTxt = resultTxt.replaceAll(word, replaceString); } return resultTxt; } /** * 获取替换字符串 * @param replaceChar * @param length * @return */ private String getReplaceChars(String replaceChar,int length){ String resultReplace = replaceChar; for(int i = 1 ; i < length ; i++){ resultReplace += replaceChar; } return resultReplace; } /** * 检查文字中是否包含敏感字符,检查规则如下: * @param txt * @param beginIndex * @param matchType * @return,如果存在,则返回敏感词字符的长度,不存在返回0 */ @SuppressWarnings({ "rawtypes"}) public int CheckSensitiveWord(String txt,int beginIndex,int matchType){ boolean flag = false; //敏感词结束标识位:用于敏感词只有1位的情况 int matchFlag = 0; //匹配标识数默认为0 char word = 0; Map nowMap = sensitiveWordMap; for(int i = beginIndex; i < txt.length() ; i++){ word = txt.charAt(i); nowMap = (Map) nowMap.get(word); //获取指定key if(nowMap != null){ //存在,则判断是否为最后一个 matchFlag++; //找到相应key,匹配标识+1 if("1".equals(nowMap.get("isEnd"))){ //如果为最后一个匹配规则,结束循环,返回匹配标识数 flag = true; //结束标志位为true if(SensitivewordFilter.minMatchTYpe == matchType){ //最小规则,直接返回,最大规则还需继续查找 break; } } } else{ //不存在,直接返回 break; } } if(matchFlag < 2 || !flag){ //长度必须大于等于1,为词 matchFlag = 0; } return matchFlag; }}这里要用到一个初始化敏感词库的工具package com.hqjl.communityserv.util;import org.slf4j.Logger;import org.slf4j.LoggerFactory;import java.io.BufferedReader;import java.io.File;import java.io.FileInputStream;import java.io.InputStreamReader;import java.util.*;/** * @author chunying * @Date: 2019/9/26 0026 */public class SensitiveWordInit { private final Logger logger = LoggerFactory.getLogger(SensitiveWordInit.class); public String ENCODING = "UTF-8"; //字符编码 @SuppressWarnings("rawtypes") public HashMap sensitiveWordMap; public SensitiveWordInit(){ super(); } @SuppressWarnings("rawtypes") public Map initKeyWord(File file){ try { //读取敏感词库 Set keyWordSet = readSensitiveWordFile(file); //将敏感词库加入到HashMap中 addSensitiveWordToHashMap(keyWordSet); //spring获取application,然后application.setAttribute("sensitiveWordMap",sensitiveWordMap); } catch (Exception e) { logger.error(e.getMessage(), e); } return sensitiveWordMap; } /** * 读取敏感词库,将敏感词放入HashSet中,构建一个DFA算法模型: * 我 = { * isEnd = 0 * 是 = { * isEnd = 1 * 坏= {isEnd = 0 * 人 = {isEnd = 1} * } * 好 = { * isEnd = 0 * 人 = { * isEnd = 1 * } * } * } * } * @param keyWordSet 敏感词库 */ @SuppressWarnings({ "rawtypes", "unchecked" }) private void addSensitiveWordToHashMap(Set keyWordSet) { sensitiveWordMap = new HashMap(keyWordSet.size()); //初始化敏感词容器,减少扩容操作 String key = null; Map nowMap = null; Map newWorMap = null; //迭代keyWordSet Iterator iterator = keyWordSet.iterator(); while(iterator.hasNext()){ key = iterator.next(); //关键字 nowMap = sensitiveWordMap; for(int i = 0 ; i < key.length() ; i++){ char keyChar = key.charAt(i); //转换成char型 Object wordMap = nowMap.get(keyChar); //获取 if(wordMap != null){ //如果存在该key,直接赋值 nowMap = (Map) wordMap; } else{ //不存在则,则构建一个map,同时将isEnd设置为0,因为他不是最后一个 newWorMap = new HashMap (); newWorMap.put("isEnd", "0"); //不是最后一个 nowMap.put(keyChar, newWorMap); nowMap = newWorMap; } if(i == key.length() - 1){ nowMap.put("isEnd", "1"); //最后一个 } } } } /** * 读取敏感词库中的内容,将内容添加到set集合中 */ @SuppressWarnings("resource") private Set readSensitiveWordFile(File file) throws Exception{ Set set = null; InputStreamReader read = new InputStreamReader(new FileInputStream(file),ENCODING); try { if(file.isFile() && file.exists()){ //文件流是否存在 set = new HashSet (); BufferedReader bufferedReader = new BufferedReader(read); String txt = null; while((txt = bufferedReader.readLine()) != null){ //读取文件,将文件内容放入到set中 set.add(txt); } } else{ //不存在抛出异常信息 throw new Exception("敏感词库文件不存在"); } } catch (Exception e) { throw e; }finally{ read.close(); //关闭文件流 } return set; }}
这里说下最小匹配原则 用于发现敏感词就算违规的系统,当发现含有敏感词时直接返回
最大匹配原则 当发现敏感词时继续查看 适用于后续需要替换敏感词然后看下文章过滤是怎么用的
首先是baseFilterpackage com.hqjl.communityserv.filter;import org.slf4j.Logger;import org.slf4j.LoggerFactory;import org.springframework.context.annotation.Scope;import org.springframework.stereotype.Component;import javax.annotation.PostConstruct;import java.io.File;/** * @author chunying * @Date: 2019/10/8 0008 */@Componentpublic class BaseFilter{ private final Logger LOG = LoggerFactory.getLogger(BaseFilter.class); private static File file; private static SensitivewordFilter secondFilter; private static SensitivewordFilter firetFilter; private T t; protected SensitivewordFilter getSecondFilter() { return secondFilter; } protected SensitivewordFilter getFiretFilter() { return firetFilter; } public void setT(T t) { this.t = t; } protected T getT() { return t; } public BaseFilter() { } @PostConstruct private synchronized void init() { if (firetFilter== null) { LOG.warn("正在初始化敏感词库..."); file = new File(BaseFilter.class.getClassLoader().getResource("file/SensitiveWord.txt").getFile()); firetFilter = new SensitivewordFilter(file); } } public FilterCheckResult checkViolation() { return null; }}
创建baseFilter的原因是 我的程序中有好多需过滤的地方,其他filter需要继承它
其中SensitiveWord.txt 是放在resource下的敏感词文件 这个类的作用是项目启动时初始化了敏感词库,初始化以后不用再次读取敏感词文件了 这里我预留了两个filter,可以用作敏感1级 、2级的区分然后是articleFilter
package com.hqjl.communityserv.filter;import com.hqjl.communityserv.bean.po.Article;import com.hqjl.communityserv.imageCheck.BaseRequest;import com.hqjl.communityserv.imageCheck.ImageSyncScanRequest;import org.apache.commons.lang3.StringUtils;import org.slf4j.Logger;import org.slf4j.LoggerFactory;import org.springframework.beans.factory.annotation.Autowired;import org.springframework.context.annotation.Scope;import org.springframework.stereotype.Component;/** * @author chunying * @Date: 2019/10/8 0008 * * @Description:帖子的过滤器 用于帖子的内容校验以及违规检查 */@Componentpublic class ArticleFilterextends BaseFilter{ private final Logger LOG = LoggerFactory.getLogger(ArticleFilter.class); public ArticleFilter() { } public FilterCheckResult checkFormat() { FilterCheckResult filterCheckResult = new FilterCheckResult(); Boolean result; T t = (T)super.getT(); String content = t.getContent(); String title = t.getTitle(); if (StringUtils.isEmpty(content.trim())) { LOG.warn("该帖子内容为空" + t.getTitle()); result = false; } result = true; filterCheckResult.setResult(result); return filterCheckResult; } @Override public FilterCheckResult checkViolation() { FilterCheckResult filterCheckResult = new FilterCheckResult(); String detail = ""; Boolean result = true; filterCheckResult.setResult(result); filterCheckResult.setDetail(detail); if(!(checkFormat().getResult())) { result = false; return filterCheckResult; } T t = (T)getT(); String content = t.getContent(); String title = t.getTitle(); String label = t.getLabel(); boolean b3 = false; //敏感词过滤 TODO 双层filter过滤 SensitivewordFilter filter = getFiretFilter(); boolean b1 = filter.isContaintSensitiveWord(content, 1); if (label != null) { b3 = filter.isContaintSensitiveWord(label, 1); } //替换敏感字 if (b1) { content = filter.replaceSensitiveWord(content, 1, "*") ; detail += "帖子正文有敏感词"; } if (b3 && label != null) { filter.replaceSensitiveWord(label, 1, "*") ; detail += "话题有敏感词"; } t.setContent(content); t.setLabel(label); if (b1 || b3) { result = false; } filterCheckResult.setResult(result); filterCheckResult.setDetail(detail); return filterCheckResult; }}这里写的有点乱 不要介意,具体的article类我就不往这里放了 仅供参考我这里是只要看到敏感词直接返回false 前端发帖失败
发表评论
最新留言
网站不错 人气很旺了 加油
[***.192.178.218]2025年03月10日 18时49分04秒
关于作者

喝酒易醉,品茶养心,人生如梦,品茶悟道,何以解忧?唯有杜康!
-- 愿君每日到此一游!
推荐文章
整型关键字的散列映射
2019-03-03
多位水仙花数-python(出现运行超时?不妨用减法计算)
2019-03-03
地下迷宫探索(后两个测试点无法通过?这里有你想要的答案)
2019-03-03
小白看完都会了!阿里云大师深入拆解Java虚拟机,看完这一篇你就懂了
2019-03-03
【IT之路】FAQ-Hibernate报错:表不存在
2019-03-03
VBA之正则表达式(19)-- 相对引用转绝对引用
2019-03-03
巧用VBA统一数字单位
2019-03-03
Transpose实现数组行列转置的限制
2019-03-03
用float/double作为中转类型的“雷区”
2019-03-03
golang中interface的一些语法缺陷的改进
2019-03-03
vue-router路由 学习笔记
2019-03-03
【数据库】第七章课后题
2019-03-03
第四章 串、数组和广义表 —— BF算法和KMP算法
2019-03-03
[选拔赛1]花园(矩阵快速幂),JM的月亮神树(最短路),保护出题人(斜率优化)
2019-03-03
DLA:一种深度网络特征融合方法
2019-03-03
leetcode114(二叉树展开为链表)
2019-03-03
java —— static 关键字
2019-03-03
在 Python 调试过程中设置不中断的断点 | Linux 中国
2019-03-03
使用开源可视化工具来理解你的 Python 代码 | Linux 中国
2019-03-03
硬核观察 | 有人在比特币骗局中损失了 10 个比特币
2019-03-03