- * 由 {@link com.github.houbb.sensitive.word.bs.SensitiveWordBs} 在 init() 时回调，
- * 因此 CRUD 完成后调用 sensitiveWordBs.init() 即可实现热更新。
- *
- * @author 芋道源码
- */
-@Component
-public class ImSensitiveWordDeny implements IWordDeny {
-
- @Resource
- private ImSensitiveWordMapper sensitiveWordMapper;
-
- @Override
- public List
- * 加 synchronized 是因为 SensitiveWordBs.init() 内部对 wordData 做就地写入，
- * 若与并发的 contains() 重叠会读到半构建的 trie。互斥后短暂阻塞 contains 即可保证可见性。
+ * 每分钟触发一次异步 reload，先读 max(update_time)，没变就复用旧实例（避免 trie 重建），变了才重新读词库 + 重建。
+ * 单实例 CRUD 后另外通过 {@link #invalidateSensitiveWordBsCaches()} 立即让本机失效，多实例靠定时刷新最长 1 分钟内收敛。
+ */
+ @SuppressWarnings({"Convert2Diamond", "NullableProblems"})
+ private final LoadingCache
- * init() 本身幂等，单实例可以直接调；多实例部署时需配合 Redis pub/sub 广播让各实例 init。
+ * 有租户上下文：仅失效该租户。无租户上下文（如系统级 / 跨租户清理）：兜底失效所有租户。
*/
- private synchronized void reloadSensitiveWords() {
- sensitiveWordBs.init();
- log.info("[reloadSensitiveWords][敏感词库已 reload]");
+ private void invalidateSensitiveWordBsCaches() {
+ Long tenantId = TenantContextHolder.getTenantId();
+ if (tenantId != null) {
+ sensitiveWordBsCaches.invalidate(tenantId);
+ return;
+ }
+ sensitiveWordBsCaches.invalidateAll();
}
- // TODO @AI：需要按照租户来。
@Override
public void validateText(String text) {
if (StrUtil.isBlank(text)) {
return;
}
- if (sensitiveWordBs.contains(text)) {
+ SensitiveWordBs bs = sensitiveWordBsCaches.getUnchecked(TenantContextHolder.getRequiredTenantId()).getBs();
+ if (bs.contains(text)) {
throw exception(MESSAGE_SENSITIVE_WORD_BLOCKED);
}
}
@@ -89,9 +161,8 @@ public class ImSensitiveWordServiceImpl implements ImSensitiveWordService {
// 2.1 入库
ImSensitiveWordDO word = BeanUtils.toBean(reqVO, ImSensitiveWordDO.class);
sensitiveWordMapper.insert(word);
- // TODO @AI：houbb 这个注释有点奇怪！
- // 2.2 触发 houbb 词库 reload
- reloadSensitiveWords();
+ // 2.2 强制失效本机缓存（多实例靠定时刷新收敛）
+ invalidateSensitiveWordBsCaches();
return word.getId();
}
@@ -105,9 +176,8 @@ public class ImSensitiveWordServiceImpl implements ImSensitiveWordService {
// 2.1 更新
ImSensitiveWordDO updateObj = BeanUtils.toBean(reqVO, ImSensitiveWordDO.class);
sensitiveWordMapper.updateById(updateObj);
- // 2.2 TODO @AI：houbb 这个注释有点奇怪！
- // 2.2 触发 houbb 词库 reload
- reloadSensitiveWords();
+ // 2.2 强制失效本机缓存
+ invalidateSensitiveWordBsCaches();
}
@Override
@@ -117,8 +187,8 @@ public class ImSensitiveWordServiceImpl implements ImSensitiveWordService {
// 2.1 删除
sensitiveWordMapper.deleteById(id);
- // 2.2 触发 houbb 词库 reload
- reloadSensitiveWords();
+ // 2.2 强制失效本机缓存
+ invalidateSensitiveWordBsCaches();
}
@@ -127,11 +197,10 @@ public class ImSensitiveWordServiceImpl implements ImSensitiveWordService {
if (CollUtil.isEmpty(ids)) {
return;
}
-
// 1. 删除
sensitiveWordMapper.deleteByIds(ids);
- // 触发 houbb 词库 reload
- reloadSensitiveWords();
+ // 2. 强制失效本机缓存
+ invalidateSensitiveWordBsCaches();
}
private void validateSensitiveWordExists(Long id) {
@@ -148,7 +217,6 @@ public class ImSensitiveWordServiceImpl implements ImSensitiveWordService {
if (exist == null) {
return;
}
- // TODO @AI：是不是 id 不用判断
if (id == null || ObjUtil.notEqual(exist.getId(), id)) {
throw exception(SENSITIVE_WORD_DUPLICATED, word);
}