如何使用@PostConstruct初始化敏感词库和hutool过滤敏感词信息

• 文档地址:https://hutool.cn/docs/#/dfa/DFA%E6%9F%A5%E6%89%BE

准备工作

敏感词存放的数据库

依赖

import cn.hutool.dfa.WordTree;

开搞

工具类

• 存放hutool用来匹配敏感词的tree

public class WordTreeUtil {
    public static WordTree tree = new WordTree();
}

初始化

• 配置类,做初始化(项目启动执行),将敏感词数据库的数据放进hutool的tree内

@Configuration
@Slf4j
public class SensitiveWordInit {
    @Autowired
    private SensitiveWordService sensitiveWordService;
    @PostConstruct
    public void init(){
        log.info("开始初始化敏感词");
        QueryWrapper<SensitiveWord> queryWrapper = new QueryWrapper<>();
        LambdaQueryWrapper<SensitiveWord> wrapper = queryWrapper.lambda();
        wrapper.eq(SensitiveWord::getStatus,0);//启用的敏感词
        List<SensitiveWord> list =  sensitiveWordService.list(wrapper);
        List<String> wordList = list.stream().map(SensitiveWord::getWord).collect(Collectors.toList());
        WordTreeUtil.tree.addWords(wordList);
        log.info("初始化敏感词成功");
    }
}

业务层部分

• 直接将传入的数据和hutool的tree内数据匹配即可,若含有则返回对应的数据

 public String checkWord(SensitiveWordStringReq req) {
        String word = req.getWord();
        List<String> matchAll = WordTreeUtil.tree.matchAll(word, -1, false, false);
        if (CollectionUtils.isNotEmpty(matchAll)){
            return matchAll.toString();
        }
        return null;
    }

• ps: 匹配的规则