shayu/cai-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

123333

Browse Source
This commit is contained in:
张良(004796)
2024-02-19 17:21:30 +08:00
parent 083fac84bb
commit 1a7c01e5cc
14 changed files with 284 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
ruoyi-admin/src/main/java/com/ruoyi/web/controller/sensitive/WordController.java
View File

@@ -3,6 +3,7 @@ package com.ruoyi.web.controller.sensitive;
import cn.dev33.satoken.annotation.SaCheckPermission;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.github.houbb.sensitive.word.bs.SensitiveWordBs;
import com.ruoyi.common.annotation.Log;
import com.ruoyi.common.annotation.RepeatSubmit;
import com.ruoyi.common.core.controller.BaseController;
@@ -13,15 +14,21 @@ import com.ruoyi.common.core.validate.AddGroup;
import com.ruoyi.common.core.validate.EditGroup;
import com.ruoyi.common.enums.BusinessType;
import com.ruoyi.common.utils.poi.ExcelUtil;
import com.ruoyi.sensitive.cache.LastUpdateTimeCache;
import com.ruoyi.sensitive.domain.Word;
import com.ruoyi.sensitive.dto.SensitiveConfig;
import com.ruoyi.sensitive.dto.WordTestResp;
import com.ruoyi.sensitive.service.SensitiveWordManager;
import com.ruoyi.sensitive.service.WordService;
import lombok.RequiredArgsConstructor;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;
import javax.servlet.http.HttpServletResponse;
import javax.validation.constraints.NotEmpty;
import javax.validation.constraints.NotNull;
import java.time.LocalDateTime;
import java.util.Arrays;
import java.util.List;
@@ -38,6 +45,30 @@ import java.util.List;
public class WordController extends BaseController {
private final WordService wordService;
@Autowired
private SensitiveWordBs sensitiveWordBs;
@Autowired
private LastUpdateTimeCache cache;
@Autowired
private SensitiveWordManager sensitiveWordManager;
@PostMapping("/updateSensitiveConfig")
public R<Void> updateSensitiveConfig(@RequestBody SensitiveConfig sensitiveConfig){
sensitiveWordManager.updateSensitiveConfig(sensitiveConfig);
return R.ok();
}
@GetMapping("/getSensitiveConfig")
public R<SensitiveConfig> getSensitiveConfig(){
return R.ok(sensitiveWordManager.getSensitiveConfig());
}
@GetMapping("/lastRefresh")
public R<LocalDateTime> lastRefresh() {
LocalDateTime time = cache.getRefreshTime();
return R.ok(time);
}
/**
* 查询敏感词列表
@@ -106,4 +137,19 @@ public class WordController extends BaseController {
@PathVariable Integer[] ids) {
return toAjax(wordService.removeWordBatchByIds(Arrays.asList(ids)));
}
@Log(title = "刷新敏感词", businessType = BusinessType.OTHER)
@GetMapping("/refresh")
public R<Void> refresh() {
sensitiveWordBs.init();
cache.setRefreshTime();
return R.ok();
}
@PostMapping("/test")
public R<WordTestResp> test(@RequestBody Word word) {
return R.ok(sensitiveWordManager.test(word.getWord()));
}
}

4
ruoyi-cai/pom.xml
View File

@@ -45,6 +45,10 @@
<artifactId>IJPay-AliPay</artifactId>
<version>${ijapy.version}</version>
</dependency>
<dependency>
<groupId>com.ruoyi</groupId>
<artifactId>ruoyi-sensitive-word</artifactId>
</dependency>
<dependency>
<groupId>com.aliyun</groupId>
<artifactId>aliyun-java-sdk-core</artifactId>

3
ruoyi-cai/src/main/java/com/ruoyi/cai/domain/UserChatFilter.java
View File

@@ -1,5 +1,6 @@
package com.ruoyi.cai.domain;
import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableName;
import lombok.Data;
@@ -22,7 +23,7 @@ public class UserChatFilter implements Serializable {
/**
*
*/
@TableId(value = "id")
@TableId(value = "id",type = IdType.AUTO)
private Long id;
/**
*

23
ruoyi-cai/src/main/java/com/ruoyi/cai/manager/ImService.java
View File

@@ -3,6 +3,7 @@ package com.ruoyi.cai.manager;
import cn.hutool.core.lang.UUID;
import com.alibaba.fastjson.JSON;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.github.houbb.sensitive.word.bs.SensitiveWordBs;
import com.ruoyi.cai.domain.Account;
import com.ruoyi.cai.domain.User;
import com.ruoyi.cai.domain.UserChatRecord;
@@ -52,6 +53,10 @@ public class ImService {
private AmqpProducer amqpProducer;
@Autowired
private ISysOssService sysOssService;
@Autowired
private SensitiveWordBs sensitiveWordBs;
@Autowired
private UserChatFilterService userChatFilterService;
public ImResp sendMessage(Long fromUserId, ImMessageDTO message) {
ChatTypeEnum typeEnum = ChatTypeEnum.getByType(message.getType());
@@ -129,15 +134,21 @@ public class ImService {
if (fromUser.getIsAnchor().equals(0) && toUser.getIsAnchor().equals(0)) {
throw new ServiceException("目前只能和女神私信!");
}
int filter = 0;
String content = message.getContent();
// 正则判断违规数据替换
if(typeEnum == ChatTypeEnum.MESSAGE){
boolean contains = sensitiveWordBs.contains(message.getContent());
if(contains){
filter = 1;
content = sensitiveWordBs.replace(message.getContent());
}
}
if(fromUser.getIsAnchor().equals(1)){ // 女神发消息不要钱
Account account = accountService.getByUserId(fromUserId);
ImResp resp = new ImResp();
resp.setCoin(account.getCoin()+ account.getIncomeCoin());
resp.setContent(message.getContent());
resp.setContent(content);
return resp;
}
// 判断VIP获取价格
@@ -150,15 +161,21 @@ public class ImService {
// 存储聊天记录
Account account = accountService.getByUserId(fromUserId);
UserChatRecord record = userChatRecordService.saveRecord(fromUser, toUser, traceId, message);
if(filter == 1){
userChatFilterService.saveFilter(fromUser, toUser, content);
}
ImResp resp = new ImResp();
resp.setCut(imPrice > 0);
resp.setCutCoin(imPrice);
resp.setCoin(account.getCoin()+ account.getIncomeCoin());
resp.setRecordId(record.getId());
resp.setContent(message.getContent());
resp.setFilter(filter);
resp.setContent(content);
return resp;
}
private Long getByImPrice(Long userId){
UserMemberTypeEnum userMemberType = userMemberService.checkUserIsMember(userId);
if(userMemberType == null){

2
ruoyi-cai/src/main/java/com/ruoyi/cai/service/UserChatFilterService.java
View File

@@ -1,6 +1,7 @@
package com.ruoyi.cai.service;
import com.baomidou.mybatisplus.extension.service.IService;
import com.ruoyi.cai.domain.User;
import com.ruoyi.cai.domain.UserChatFilter;
/**
@@ -11,4 +12,5 @@ import com.ruoyi.cai.domain.UserChatFilter;
*/
public interface UserChatFilterService extends IService<UserChatFilter> {
void saveFilter(User fromUser, User toUser, String content);
}

9
ruoyi-cai/src/main/java/com/ruoyi/cai/service/impl/UserChatFilterServiceImpl.java
View File

@@ -1,6 +1,7 @@
package com.ruoyi.cai.service.impl;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.ruoyi.cai.domain.User;
import com.ruoyi.cai.domain.UserChatFilter;
import com.ruoyi.cai.mapper.UserChatFilterMapper;
import com.ruoyi.cai.service.UserChatFilterService;
@@ -14,4 +15,12 @@ import org.springframework.stereotype.Service;
*/
@Service
public class UserChatFilterServiceImpl extends ServiceImpl<UserChatFilterMapper,UserChatFilter> implements UserChatFilterService {
@Override
public void saveFilter(User fromUser, User toUser, String content) {
UserChatFilter chatFilter = new UserChatFilter();
chatFilter.setFromUid(fromUser.getId());
chatFilter.setToUid(toUser.getId());
chatFilter.setContent(content);
this.save(chatFilter);
}
}

42
ruoyi-sensitive-word/src/main/java/com/ruoyi/sensitive/cache/LastUpdateTimeCache.java vendored
View File

@@ -1,26 +1,54 @@
package com.ruoyi.sensitive.cache;
import com.ruoyi.common.constant.DateConstant;
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;
import java.time.format.DateTimeFormatter;
@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();
return "sensitive-lastUpdateTime";
}
public void setTime(){
private String getRefreshKey(){
return "sensitive-lastRefreshTime";
}
public void setRefreshTime(){
String key = getRefreshKey();
RBucket<LocalDateTime> bucket = redissonClient.getBucket(key);
bucket.set(LocalDateTime.now());
}
public LocalDateTime getRefreshTime(){
String key = getRefreshKey();
RBucket<String> bucket = redissonClient.getBucket(key);
String s = bucket.get();
if(s == null){
return null;
}
return LocalDateTime.parse(s, DateTimeFormatter.ofPattern(DateConstant.PATTERN_DATETIME));
}
public LocalDateTime getUpdateTime(){
String key = getKey();
RBucket<String> bucket = redissonClient.getBucket(key);
String s = bucket.get();
if(s == null){
return null;
}
return LocalDateTime.parse(s, DateTimeFormatter.ofPattern(DateConstant.PATTERN_DATETIME));
}
public void setUpdateTime(){
String key = getKey();
RBucket<LocalDateTime> bucket = redissonClient.getBucket(key);
bucket.set(LocalDateTime.now());

28
ruoyi-sensitive-word/src/main/java/com/ruoyi/sensitive/cache/SensitiveWordConfigCache.java vendored Normal file
View File

@@ -0,0 +1,28 @@
package com.ruoyi.sensitive.cache;
import com.ruoyi.sensitive.dto.SensitiveConfig;
import org.redisson.api.RBucket;
import org.redisson.api.RedissonClient;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
@Component
public class SensitiveWordConfigCache {
@Autowired
private RedissonClient redissonClient;
private String getKey(){
return "sensitive-config";
}
public void setConfig(SensitiveConfig config){
String key = getKey();
RBucket<SensitiveConfig> bucket = redissonClient.getBucket(key);
bucket.set(config);
}
public SensitiveConfig getConfig(){
String key = getKey();
RBucket<SensitiveConfig> bucket = redissonClient.getBucket(key);
return bucket.get();
}
}

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

@@ -18,19 +18,17 @@ public class MySensitiveWordScheduleRefresh {
@Autowired
private SensitiveWordBs sensitiveWordBs;
@Autowired
private LastUpdateTimeCache lastUpdateTimeCache;
private LastUpdateTimeCache cache;
@Scheduled(fixedDelay = 20,timeUnit = TimeUnit.SECONDS)
public void job(){
try {
log.info("MySensitiveWordScheduleRefresh start");
LocalDateTime time = lastUpdateTimeCache.getTime();
LocalDateTime time = cache.getUpdateTime();
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);
}
@@ -42,6 +40,7 @@ public class MySensitiveWordScheduleRefresh {
private void refresh() {
sensitiveWordBs.init();
cache.setRefreshTime();
}
}

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

@@ -3,6 +3,8 @@ 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 com.ruoyi.sensitive.cache.SensitiveWordConfigCache;
import com.ruoyi.sensitive.dto.SensitiveConfig;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
@@ -19,6 +21,8 @@ public class SensitiveWordConfig {
@Autowired
private MyDdWordDeny myDdWordDeny;
@Autowired
private SensitiveWordConfigCache sensitiveWordConfigCache;
/**
* 初始化引导类
@@ -27,11 +31,25 @@ public class SensitiveWordConfig {
*/
@Bean
public SensitiveWordBs sensitiveWordBs() {
SensitiveConfig sensitiveConfig = sensitiveWordConfigCache.getConfig();
if(sensitiveConfig == null){
sensitiveConfig = new SensitiveConfig();
}
return SensitiveWordBs.newInstance()
.wordAllow(WordAllows.chains(WordAllows.defaults(), myDdWordAllow))
.wordDeny(WordDenys.chains(WordDenys.defaults(), myDdWordDeny))
.ignoreRepeat(false)
// 各种其他配置
.ignoreCase(sensitiveConfig.isIgnoreCase())
.ignoreWidth(sensitiveConfig.isIgnoreWidth())
.ignoreNumStyle(sensitiveConfig.isIgnoreNumStyle())
.ignoreChineseStyle(sensitiveConfig.isIgnoreChineseStyle())
.ignoreEnglishStyle(sensitiveConfig.isIgnoreEnglishStyle())
.ignoreRepeat(sensitiveConfig.isIgnoreRepeat())
.enableNumCheck(sensitiveConfig.isEnableNumCheck())
.enableEmailCheck(sensitiveConfig.isEnableEmailCheck())
.enableUrlCheck(sensitiveConfig.isEnableUrlCheck())
.enableWordCheck(sensitiveConfig.isEnableWordCheck())
.numCheckLen(sensitiveConfig.getNumCheckLen())
.init();
}

57
ruoyi-sensitive-word/src/main/java/com/ruoyi/sensitive/dto/SensitiveConfig.java Normal file
View File

@@ -0,0 +1,57 @@
package com.ruoyi.sensitive.dto;
import lombok.Data;
@Data
public class SensitiveConfig {
/**
* 是否忽略大小写
*/
private boolean ignoreCase = true;
/**
* 是否忽略全角、半角
*/
private boolean ignoreWidth = true;
/**
* 是否忽略数字样式
*/
private boolean ignoreNumStyle = true;
/**
* 是否忽略中文样式
*/
private boolean ignoreChineseStyle = true;
/**
* 是否忽略英文样式
*/
private boolean ignoreEnglishStyle = true;
/**
* 是否忽略重复
*/
private boolean ignoreRepeat = false;
// 开启校验
/**
* 启用数字检测
*/
private boolean enableNumCheck = true;
/**
* 启用邮箱检测
*/
private boolean enableEmailCheck = true;
/**
* 启用 URL 检测
*/
private boolean enableUrlCheck = true;
/**
* 单词校验
* @since 0.4.0
*/
private boolean enableWordCheck = true;
// 额外配置
/**
* 检测数字时的长度
*/
private int numCheckLen = 8;
}

9
ruoyi-sensitive-word/src/main/java/com/ruoyi/sensitive/dto/WordTestResp.java Normal file
View File

@@ -0,0 +1,9 @@
package com.ruoyi.sensitive.dto;
import lombok.Data;
@Data
public class WordTestResp {
private String sourceText;
private String targetText;
}

47
ruoyi-sensitive-word/src/main/java/com/ruoyi/sensitive/service/SensitiveWordManager.java Normal file
View File

@@ -0,0 +1,47 @@
package com.ruoyi.sensitive.service;
import com.github.houbb.sensitive.word.bs.SensitiveWordBs;
import com.ruoyi.sensitive.cache.SensitiveWordConfigCache;
import com.ruoyi.sensitive.dto.SensitiveConfig;
import com.ruoyi.sensitive.dto.WordTestResp;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
@Component
public class SensitiveWordManager {
@Autowired
private SensitiveWordBs sensitiveWordBs;
@Autowired
private SensitiveWordConfigCache sensitiveWordConfigCache;
public WordTestResp test(String text) {
String results = sensitiveWordBs.replace(text);
WordTestResp wordTestResp = new WordTestResp();
wordTestResp.setSourceText(text);
wordTestResp.setTargetText(results);
return wordTestResp;
}
public void updateSensitiveConfig(SensitiveConfig sensitiveConfig) {
sensitiveWordBs
.ignoreCase(sensitiveConfig.isIgnoreCase())
.ignoreWidth(sensitiveConfig.isIgnoreWidth())
.ignoreNumStyle(sensitiveConfig.isIgnoreNumStyle())
.ignoreChineseStyle(sensitiveConfig.isIgnoreChineseStyle())
.ignoreEnglishStyle(sensitiveConfig.isIgnoreEnglishStyle())
.ignoreRepeat(sensitiveConfig.isIgnoreRepeat())
.enableNumCheck(sensitiveConfig.isEnableNumCheck())
.enableEmailCheck(sensitiveConfig.isEnableEmailCheck())
.enableUrlCheck(sensitiveConfig.isEnableUrlCheck())
.enableWordCheck(sensitiveConfig.isEnableWordCheck())
.numCheckLen(sensitiveConfig.getNumCheckLen())
.init();
sensitiveWordConfigCache.setConfig(sensitiveConfig);
}
public SensitiveConfig getSensitiveConfig(){
return sensitiveWordConfigCache.getConfig();
}
}

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

@@ -1,7 +1,6 @@
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;
@@ -26,21 +25,21 @@ public class WordServiceImpl extends ServiceImpl<WordMapper,Word> implements Wor
@Override
public boolean removeWordBatchByIds(List<Integer> asList) {
boolean re = this.removeBatchByIds(asList);
lastUpdateTimeCache.setTime();
lastUpdateTimeCache.setUpdateTime();
return re;
}
@Override
public boolean updateWordById(Word bo) {
boolean re = this.updateById(bo);
lastUpdateTimeCache.setTime();
lastUpdateTimeCache.setUpdateTime();
return re;
}
@Override
public boolean saveWord(Word bo) {
boolean re = this.save(bo);
lastUpdateTimeCache.setTime();
lastUpdateTimeCache.setUpdateTime();
return re;
}
}