最近接到一个需求,要添加一个敏感词管理模块,一如既往的CURD,敏感词我们添加到了自己的库里。然后进行一个自定义敏感词过滤,话不多说直接贴代码
1、工具类
这里只是最简单的得到敏感词进行转换,可以根据自己的业务需求进行填充
package com.zylc.bixiang.business.keywords.web;import com.zylc.bixiang.business.keywords.domain.repository.BXKeyWordsMapper;
import com.zylc.bixiang.business.keywords.domain.vo.BXKeyWordsVO;
import org.springframework.beans.factory.annotation.Autowired;import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.UnsupportedEncodingException;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.List;/*** @author admin* @Description: 敏感词过滤 工具类* @Date : 2021/10/28 10:21* @Modified By :*/
public class SensitiveWordFilter {@AutowiredBXKeyWordsMapper bxKeyWordsMapper;//初始化private StringBuilder replaceAll;private String encoding = "UTF-8";private String replceStr = "*";private int replceSize = 500;private List<BXKeyWordsVO> keyWordsVOS;/*** @param str 将要被过滤信息* @return 过滤后的信息*/public String filterInfo(String str) {StringBuilder buffer = new StringBuilder(str);HashMap<Integer, Integer> hash = new HashMap<Integer, Integer>(keyWordsVOS.size());String temp;for (int x = 0; x < keyWordsVOS.size(); x++) {temp = keyWordsVOS.get(x).getValue();int findIndexSize = 0;for (int start = -1; (start = buffer.indexOf(temp, findIndexSize)) > -1; ) {//从已找到的后面开始找findIndexSize = start + temp.length();//起始位置Integer mapStart = hash.get(start);//满足1个,即可更新mapif (mapStart == null || (mapStart != null && findIndexSize > mapStart)) {hash.put(start, findIndexSize);}}}Collection<Integer> values = hash.keySet();for (Integer startIndex : values) {Integer endIndex = hash.get(startIndex);buffer.replace(startIndex, endIndex, replaceAll.substring(0, endIndex - startIndex));}hash.clear();return buffer.toString();}/*** 初始化敏感词库*/public void InitializationWork() {replaceAll = new StringBuilder(replceSize);for (int x = 0; x < replceSize; x++) {replaceAll.append(replceStr);}//加载词库keyWordsVOS = bxKeyWordsMapper.findAllKeywords();}public StringBuilder getReplaceAll() {return replaceAll;}public void setReplaceAll(StringBuilder replaceAll) {this.replaceAll = replaceAll;}public String getReplceStr() {return replceStr;}public void setReplceStr(String replceStr) {this.replceStr = replceStr;}public int getReplceSize() {return replceSize;}public void setReplceSize(int replceSize) {this.replceSize = replceSize;}public void setEncoding(String encoding) {this.encoding = encoding;}
}
3、因为我们业务有敏感级别,我要得到每个敏感词和相对应的级别,所以上方初始化词库用的集合对象