123333

ruoyi-sensitive-word/src/main/java/com/ruoyi/sensitive/cache/LastUpdateTimeCache.java

@@ -0,0 +1,28 @@
+package com.ruoyi.sensitive.cache;
+
+import org.redisson.api.RBucket;
+import org.redisson.api.RedissonClient;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Component;
+
+import java.time.LocalDateTime;
+
+@Component
+public class LastUpdateTimeCache {
+    @Autowired
+    private RedissonClient redissonClient;
+    private String getKey(){
+        return "sensitive-lastRefreshTime";
+    }
+    public LocalDateTime getTime(){
+        String key = getKey();
+        RBucket<LocalDateTime> bucket = redissonClient.getBucket(key);
+        return bucket.get();
+    }
+
+    public void setTime(){
+        String key = getKey();
+        RBucket<LocalDateTime> bucket = redissonClient.getBucket(key);
+        bucket.set(LocalDateTime.now());
+    }
+}

ruoyi-sensitive-word/src/main/java/com/ruoyi/sensitive/config/MyDdWordAllow.java

@@ -0,0 +1,38 @@
+package com.ruoyi.sensitive.config;
+
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
+import com.github.houbb.heaven.support.handler.IHandler;
+import com.github.houbb.heaven.util.util.CollectionUtil;
+import com.github.houbb.sensitive.word.api.IWordAllow;
+import com.ruoyi.sensitive.domain.Word;
+import com.ruoyi.sensitive.service.WordService;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Component;
+
+import java.util.List;
+
+/**
+ * @author binbin.hou
+ * @since 1.1.0
+ */
+@Component
+public class MyDdWordAllow implements IWordAllow {
+
+    @Autowired
+    private WordService wordService;
+
+    @Override
+    public List<String> allow() {
+        List<Word> wordList = wordService.list(Wrappers.lambdaQuery(Word.class)
+                .select(Word::getWord)
+                .eq(Word::getEnableStatus,1)
+                .eq(Word::getType, 1));
+        return CollectionUtil.toList(wordList, new IHandler<Word, String>() {
+            @Override
+            public String handle(Word word) {
+                return word.getWord();
+            }
+        });
+    }
+
+}

ruoyi-sensitive-word/src/main/java/com/ruoyi/sensitive/config/MyDdWordDeny.java

@@ -0,0 +1,39 @@
+package com.ruoyi.sensitive.config;
+
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
+import com.github.houbb.heaven.support.handler.IHandler;
+import com.github.houbb.heaven.util.util.CollectionUtil;
+import com.github.houbb.sensitive.word.api.IWordDeny;
+import com.ruoyi.sensitive.domain.Word;
+import com.ruoyi.sensitive.service.WordService;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Component;
+
+import java.util.List;
+
+/**
+ * @author binbin.hou
+ * @since 1.1.0
+ */
+@Component
+public class MyDdWordDeny implements IWordDeny {
+
+    @Autowired
+    private WordService wordService;
+
+    @Override
+    public List<String> deny() {
+        List<Word> wordList = wordService.list(Wrappers.lambdaQuery(Word.class)
+                .select(Word::getWord)
+                .eq(Word::getEnableStatus,1)
+                .eq(Word::getType, 2));
+
+        return CollectionUtil.toList(wordList, new IHandler<Word, String>() {
+            @Override
+            public String handle(Word word) {
+                return word.getWord();
+            }
+        });
+    }
+
+}

ruoyi-sensitive-word/src/main/java/com/ruoyi/sensitive/config/MySensitiveWordScheduleRefresh.java

@@ -0,0 +1,47 @@
+package com.ruoyi.sensitive.config;
+
+import com.github.houbb.sensitive.word.bs.SensitiveWordBs;
+import com.ruoyi.sensitive.cache.LastUpdateTimeCache;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.scheduling.annotation.Scheduled;
+import org.springframework.stereotype.Component;
+
+import javax.annotation.PostConstruct;
+import java.time.LocalDateTime;
+import java.util.concurrent.TimeUnit;
+
+@Component
+@Slf4j
+public class MySensitiveWordScheduleRefresh {
+
+    @Autowired
+    private SensitiveWordBs sensitiveWordBs;
+    @Autowired
+    private LastUpdateTimeCache lastUpdateTimeCache;
+
+    @Scheduled(fixedDelay = 20,timeUnit = TimeUnit.SECONDS)
+    public void job(){
+        try {
+            log.info("MySensitiveWordScheduleRefresh start");
+            LocalDateTime time = lastUpdateTimeCache.getTime();
+            LocalDateTime now = LocalDateTime.now();
+            if(time == null || time.isBefore(now)){
+                return;
+            }
+            refresh();
+            log.info("MySensitiveWordScheduleRefresh end");
+        } catch (Exception e) {
+            log.error("MySensitiveWordScheduleRefresh meet ex", e);
+        }
+    }
+
+    @PostConstruct
+    public void init() {
+    }
+
+    private void refresh() {
+        sensitiveWordBs.init();
+    }
+
+}

ruoyi-sensitive-word/src/main/java/com/ruoyi/sensitive/config/SensitiveWordConfig.java

@@ -0,0 +1,38 @@
+package com.ruoyi.sensitive.config;
+
+import com.github.houbb.sensitive.word.bs.SensitiveWordBs;
+import com.github.houbb.sensitive.word.support.allow.WordAllows;
+import com.github.houbb.sensitive.word.support.deny.WordDenys;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+
+/**
+ * @author binbin.hou
+ * @since 1.1.0
+ */
+@Configuration
+public class SensitiveWordConfig {
+
+    @Autowired
+    private MyDdWordAllow myDdWordAllow;
+
+    @Autowired
+    private MyDdWordDeny myDdWordDeny;
+
+    /**
+     * 初始化引导类
+     * @return 初始化引导类
+     * @since 1.0.0
+     */
+    @Bean
+    public SensitiveWordBs sensitiveWordBs() {
+        return SensitiveWordBs.newInstance()
+                .wordAllow(WordAllows.chains(WordAllows.defaults(), myDdWordAllow))
+                .wordDeny(WordDenys.chains(WordDenys.defaults(), myDdWordDeny))
+                .ignoreRepeat(false)
+                // 各种其他配置
+                .init();
+    }
+
+}

ruoyi-sensitive-word/src/main/java/com/ruoyi/sensitive/domain/Word.java

@@ -0,0 +1,51 @@
+package com.ruoyi.sensitive.domain;
+
+import com.baomidou.mybatisplus.annotation.TableId;
+import com.baomidou.mybatisplus.annotation.TableName;
+import lombok.Data;
+
+import java.io.Serializable;
+import java.time.LocalDateTime;
+
+/**
+ * 敏感词对象 cai_word
+ *
+ * @author 77
+ * @date 2024-02-19
+ */
+@Data
+@TableName("cai_word")
+public class Word implements Serializable {
+
+    private static final long serialVersionUID=1L;
+
+    /**
+     * 应用自增主键
+     */
+    @TableId(value = "id")
+    private Integer id;
+    /**
+     * 单词
+     */
+    private String word;
+    /**
+     * 1-白名单 2-黑名单
+     */
+    private Integer type;
+    /**
+     * 启用状态
+     */
+    private Integer enableStatus;
+    /**
+     * 配置描述
+     */
+    private String remark;
+    /**
+     * 操作员名称
+     */
+    private String operatorId;
+
+    private LocalDateTime updateTime;
+    private LocalDateTime createTime;
+
+}

ruoyi-sensitive-word/src/main/java/com/ruoyi/sensitive/mapper/WordMapper.java

@@ -0,0 +1,14 @@
+package com.ruoyi.sensitive.mapper;
+
+import com.baomidou.mybatisplus.core.mapper.BaseMapper;
+import com.ruoyi.sensitive.domain.Word;
+
+/**
+ * 敏感词Mapper接口
+ *
+ * @author 77
+ * @date 2024-02-19
+ */
+public interface WordMapper extends BaseMapper<Word> {
+
+}

ruoyi-sensitive-word/src/main/java/com/ruoyi/sensitive/service/WordService.java

@@ -0,0 +1,21 @@
+package com.ruoyi.sensitive.service;
+
+import com.baomidou.mybatisplus.extension.service.IService;
+import com.ruoyi.sensitive.domain.Word;
+
+import java.util.List;
+
+/**
+ * 敏感词Service接口
+ *
+ * @author 77
+ * @date 2024-02-19
+ */
+public interface WordService extends IService<Word> {
+
+    boolean removeWordBatchByIds(List<Integer> asList);
+
+    boolean updateWordById(Word bo);
+
+    boolean saveWord(Word bo);
+}

ruoyi-sensitive-word/src/main/java/com/ruoyi/sensitive/service/impl/WordServiceImpl.java

@@ -0,0 +1,46 @@
+package com.ruoyi.sensitive.service.impl;
+
+import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.github.houbb.sensitive.word.support.allow.WordAllows;
+import com.ruoyi.sensitive.cache.LastUpdateTimeCache;
+import com.ruoyi.sensitive.domain.Word;
+import com.ruoyi.sensitive.mapper.WordMapper;
+import com.ruoyi.sensitive.service.WordService;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Service;
+
+import java.util.List;
+
+/**
+ * 敏感词Service业务层处理
+ *
+ * @author 77
+ * @date 2024-02-19
+ */
+@Service
+public class WordServiceImpl extends ServiceImpl<WordMapper,Word> implements WordService {
+
+    @Autowired
+    private LastUpdateTimeCache lastUpdateTimeCache;
+
+    @Override
+    public boolean removeWordBatchByIds(List<Integer> asList) {
+        boolean re = this.removeBatchByIds(asList);
+        lastUpdateTimeCache.setTime();
+        return re;
+    }
+
+    @Override
+    public boolean updateWordById(Word bo) {
+        boolean re = this.updateById(bo);
+        lastUpdateTimeCache.setTime();
+        return re;
+    }
+
+    @Override
+    public boolean saveWord(Word bo) {
+        boolean re = this.save(bo);
+        lastUpdateTimeCache.setTime();
+        return re;
+    }
+}