Compare commits

...

19 Commits

Author SHA1 Message Date
94729e5170 refactor(note):优化笔记点赞功能,使用 Roaring Bitmap 替代布隆过滤器
- 修改消费者组名称,统一命名规范
- 更新 HTTP 客户端测试用例中的授权令牌和笔记 ID
- 引入 NoteLikeDOMapper 并替换原有的 service 查询方式
- 将布隆过滤器相关逻辑全部替换为 Roaring Bitmap 实现
- 新增多个 Lua 脚本支持 Roaring Bitmap 的操作与初始化
- 添加 Roaring Bitmap 相关的 Redis Key 构建方法
- 删除旧有的布隆过滤器校验逻辑及冗余代码
- 更新 Redis Key 常量类,增加 Roaring Bitmap 相关定义
- 日志字典文件中新增 rbitmap 关键词
- 优化点赞和取消点赞流程,提升性能与准确性
2025-11-09 22:09:23 +08:00
6e0f226b42 feat(comment): 实现评论删除功能及相关缓存更新
- 新增删除一级评论及其子评论的逻辑
- 支持批量删除评论及递归查找回复评论
- 添加评论计数更新机制,包括 Redis 和数据库同步
- 实现评论热度值异步更新消息队列发送- 新增本地缓存删除消费者,支持广播模式清理缓存
- 扩展 CommentLevelEnum 枚举,增加 valueOf 方法用于类型转换- 在 NoteCountDOMapper 中新增更新评论总数的方法- 完善注释和日志记录,提升代码可读性和维护性
2025-11-09 15:30:04 +08:00
e0cf96edbf Revert "feat(comment): 新增评论删除及缓存清理功能"
This reverts commit 6985431236.
2025-11-09 15:19:48 +08:00
d9a960e265 Merge remote-tracking branch 'all/master'
# Conflicts:
#	han-note-comment/han-note-comment-biz/src/main/java/com/hanserwei/hannote/comment/biz/enums/ResponseCodeEnum.java
#	han-note-comment/han-note-comment-biz/src/main/java/com/hanserwei/hannote/comment/biz/service/impl/CommentServiceImpl.java
#	http-client/gateApi.http
2025-11-09 15:18:17 +08:00
6985431236 feat(comment): 新增评论删除及缓存清理功能
- 新增删除一级评论及其子评论的接口与实现- 新增批量删除评论的功能支持
- 新增根据回复评论ID查询评论的方法
- 为CommentLevelEnum添加通过code获取枚举值的方法
- 实现评论本地缓存删除服务接口
- 新增删除评论后的MQ消费者处理逻辑
- 新增删除评论本地缓存的MQ广播消费逻辑
- 扩展NoteCountDOMapper以支持评论总数更新操作
2025-11-09 15:16:20 +08:00
85e0238857 feat(comment): 新增删除评论功能
- 新增删除评论接口,支持物理删除评论及关联内容
- 添加权限校验,仅允许评论创建者删除评论- 使用编程式事务保证删除操作的原子性- 删除评论后清理 Redis 缓存(ZSet 和 String 类型)
- 发送 MQ 消息异步更新计数、删除关联数据及本地缓存
- 新增 DeleteCommentReqVO 请求参数类校验评论 ID
- 补充 KeyValueRpcService 删除评论内容方法
- 新增相关 MQ Topic 常量及响应码枚举
- 更新 HTTP 接口测试用例
2025-11-09 14:12:24 +08:00
93ca81a15b feat(kv): 新增删除评论内容功能
- 在 CommentContentController 中新增删除评论内容的接口
- 定义 DeleteCommentContentReqDTO 用于接收删除请求参数
- 在 CommentContentRepository 中新增删除评论正文的方法
- 在 CommentContentService 及其实现类中新增删除评论内容的业务逻辑
- 在 KeyValueFeignApi 中新增删除评论内容的 Feign 接口
- 在 gateApi.http 中添加删除评论内容的测试用例
2025-11-09 14:01:14 +08:00
f74397ed1e feat(comment): 计数服务:评论点赞数更新,取消点赞接口
- 新增取消点赞接口 /comment/unlike
- 添加布隆过滤器校验评论是否已点赞
- 实现取消点赞时从布隆过滤器中移除记录
- 发送取消点赞消息到 RocketMQ 进行异步处理
- 新增取消点赞相关枚举和异常码
- 更新计数服务消费点赞/取消点赞消息逻辑
- 支持评论点赞数的增减与持久化更新
- 添加 HTTP 客户端测试用例
2025-11-09 13:53:07 +08:00
f90e36f7d6 feat(comment): 实现评论点赞与取消点赞功能,评论点赞、取消点赞批量写库
- 新增评论点赞布隆过滤器,提升点赞判断性能
- 实现评论点赞与取消点赞的批量操作消费者
- 添加评论点赞状态查询接口及异常处理
- 优化点赞操作合并逻辑,减少数据库访问频率
- 增加评论点赞相关 Lua 脚本支持过期时间设置
- 完善评论点赞 Mapper 层批量插入与删除方法
- 添加评论已点赞业务异常状态码
- 新增测试类用于验证评论点赞 MQ 消费逻辑
- 调整 MQ 消费者 Bean 名称避免冲突
- 更新 HTTP 测试文件中的评论 ID便于调试
2025-11-08 22:55:09 +08:00
a8d5c7f9b7 feat(comment): 实现评论点赞功能
- 新增评论点赞接口,支持用户对评论进行点赞操作
- 集成 Redis 布隆过滤器,用于快速校验评论是否已被点赞
- 编写 Lua 脚本实现布隆过滤器的检查与添加逻辑
- 定义点赞相关枚举类,包括点赞状态和操作类型- 新增点赞请求 VO 和 MQ 消息 DTO,规范数据传输结构
- 通过 RocketMQ 异步处理点赞记录落库,提升接口响应速度
- 添加 HTTP 客户端测试用例,便于接口调试和验证
- 补充 Redis Key 构建方法及常量定义,统一缓存键管理
2025-11-08 22:16:15 +08:00
51cebf6215 feat(count): 新增笔记评论数缓存更新逻辑
- 在 CountNoteCommentConsumer 中引入 RedisTemplate依赖
- 消费评论消息时,更新 Redis 中笔记评论总数
- 新增 RedisKeyConstants.FIELD_COMMENT_TOTAL 常量定义
- 实现基于 Hash 的评论数累加更新机制
- 优化评论数更新流程,支持批量处理与缓存同步
2025-11-08 22:06:26 +08:00
8be6719be8 feat(comment): 实现子评论分页查询与缓存优化
- 新增根据父评论 ID 和限制数量查询子评论的数据库方法
- 实现子评论分页缓存机制,使用 ZSET + String 结构提升查询性能
- 添加子评论详情批量同步到 Redis 的功能
-优化子评论计数数据的缓存读取与数据库同步逻辑
- 新增父评论不存在时的业务异常处理
- 完善子评论缓存失效后的数据库查询与数据回填机制
- 提取公共方法用于批量操作 Redis 数据,提升代码复用性
2025-11-08 21:58:29 +08:00
e3f9b6a5b5 feat(comment): 一级评论:子评论总数更新与查询
- 新增批量查询评论计数的数据库接口及SQL实现
- 优化本地缓存中评论ID失效判断逻辑,修正变量命名
- 增加从Redis中获取评论计数数据的功能,并支持缺失时回源数据库
- 实现评论计数数据异步同步至Redis的逻辑,包括子评论总数和点赞数
- 在消费端增加更新Redis中评论子评论总数的逻辑
- 添加评论计数相关的Redis Key和Field常量定义
- 更新HTTP测试用例中的评论内容和回复ID,验证计数同步功能
2025-11-08 21:01:15 +08:00
6f22c2b50d feat(comment): 实现二级评论分页查询功能
- 新增子评论分页查询接口 /comment/child/list- 添加查询一级评论下子评论总数的 Mapper 方法
- 实现二级评论分页数据查询的 Mapper 方法
- 补充对应的 XML 查询语句,支持按 parent_id 查询子评论
- 创建 FindChildCommentItemRspVO 和 FindChildCommentPageListReqVO VO 类
- 在 CommentServiceImpl 中实现子评论分页查询业务逻辑
- 支持批量查询子评论内容及用户信息并组装返回数据
- 添加 HTTP 客户端测试用例用于验证接口功能
2025-11-08 20:29:24 +08:00
bd775b805c feat(comment): 实现评论热度同步到 Redis ZSet
- 新增 Lua 脚本支持热点评论添加与更新
- 在评论消费端同步一级评论至 Redis 热点评论 ZSet
- 支持批量更新评论热度并维护 Redis 中的 Top 500 热点评论
- 修改 CommentDO 和 CommentHeatBO 模型,增加 noteId 字段以支持按笔记分组
- 调整 Mapper XML 查询字段,补充 note_id 字段用于构建 Redis Key
- 优化 Redis 脚本执行逻辑,确保线程安全及数据一致性
- 更新 HTTP 测试用例内容,验证 Redis 同步功能正确性
2025-11-08 15:48:18 +08:00
85e6bab079 feat(comment): 引入本地缓存优化评论查询性能
- 添加 Caffeine 依赖以支持本地缓存
- 实现评论详情的本地缓存机制,减少 Redis 查询压力
-优化分页查询逻辑,优先从本地缓存获取评论数据
- 异步同步评论详情至本地缓存,提升响应速度
- 简化 Redis 操作代码,提高可读性和维护性
2025-11-08 15:34:27 +08:00
6fbe8eed25 feat(comment): 实现评论热度排序及缓存优化
- 修改 CommentDO 中 heat 字段类型从 BigDecimal为 Double
- 新增 selectHeatComments 方法用于查询热门评论- 优化评论分页查询逻辑,引入 Redis 缓存提升性能
- 新增评论总数与热门评论的 Redis 缓存同步机制
- 实现评论详情的 Redis 批量缓存与过期策略
- 添加 COMMENT_NOT_FOUND 业务异常码
- 更新 RedisKeyConstants 增加相关键构建方法
- 调整 XML 映射文件以支持新的查询与字段类型- 引入 RedisTemplate 和线程池异步处理缓存操作
- 在 FindCommentItemRspVO 中新增 heat 字段返回热度值
2025-11-08 11:43:32 +08:00
fdee4dc2b4 feat(comment): 实现评论分页查询功能
- 新增评论分页查询接口与实现逻辑
- 支持查询一级评论及其最早回复的二级评论
- 支持从KV服务批量获取评论内容
- 支持从用户服务批量获取用户信息并组装- 新增评论热度字段用于排序
- 修改MyBatis代码生成配置至comment模块
- 调整评论表结构,优化字段定义
- 完善相关DTO、VO及Mapper文件
- 添加HTTP客户端测试用例
2025-11-08 11:07:50 +08:00
2b06ca0300 feat(kv): 新增批量查询评论内容功能
- 新增 BatchFindCommentContentReqDTO 用于批量查询请求参数校验
- 新增 FindCommentContentReqDTO 和 FindCommentContentRspDTO 用于查询参数与响应封装
- 在 CommentContentController 中添加 /comment/content/batchFind 接口
- 实现 CommentContentRepository 的批量查询方法
- 在 CommentContentServiceImpl 中完成批量查询逻辑,包括参数解析与数据转换
- 更新 gateApi.http 添加批量查询接口测试用例
2025-11-08 09:54:34 +08:00
73 changed files with 3885 additions and 94 deletions

View File

@@ -9,11 +9,11 @@
<option name="deleteByPrimayKeyEnabled" value="false" /> <option name="deleteByPrimayKeyEnabled" value="false" />
<option name="insertMethodEnabled" value="false" /> <option name="insertMethodEnabled" value="false" />
<option name="insertSelectiveMethodEnabled" value="false" /> <option name="insertSelectiveMethodEnabled" value="false" />
<option name="javaMapperPackage" value="com.hanserwei.hannote.count.biz.domain.mapper" /> <option name="javaMapperPackage" value="com.hanserwei.hannote.comment.biz.domain.mapper" />
<option name="javaMapperPath" value="$PROJECT_DIR$/han-note-count/han-note-count-biz/src/main/java" /> <option name="javaMapperPath" value="$PROJECT_DIR$/han-note-comment/han-note-comment-biz/src/main/java" />
<option name="javaModelPackage" value="com.hanserwei.hannote.count.biz.domain.dataobject" /> <option name="javaModelPackage" value="com.hanserwei.hannote.comment.biz.domain.dataobject" />
<option name="javaModelPath" value="$PROJECT_DIR$/han-note-count/han-note-count-biz/src/main/java" /> <option name="javaModelPath" value="$PROJECT_DIR$/han-note-comment/han-note-comment-biz/src/main/java" />
<option name="lastDatabaseCrudChooseModuleName" value="han-note-count-biz" /> <option name="lastDatabaseCrudChooseModuleName" value="han-note-comment-biz" />
<option name="lombokAllArgConstructor" value="true" /> <option name="lombokAllArgConstructor" value="true" />
<option name="lombokDataAnnotation" value="true" /> <option name="lombokDataAnnotation" value="true" />
<option name="lombokNoArgsConstructor" value="true" /> <option name="lombokNoArgsConstructor" value="true" />
@@ -94,7 +94,7 @@
</entry> </entry>
</map> </map>
</option> </option>
<option name="mybatisPlusIdType" value="AUTO" /> <option name="mybatisPlusIdType" value="ASSIGN_ID" />
<option name="selectByPrimaryKeyEnabled" value="false" /> <option name="selectByPrimaryKeyEnabled" value="false" />
<option name="tableGenerateConfigs"> <option name="tableGenerateConfigs">
<map> <map>
@@ -144,7 +144,7 @@
<option name="insertMethodEnabled" value="false" /> <option name="insertMethodEnabled" value="false" />
<option name="insertSelectiveMethodEnabled" value="false" /> <option name="insertSelectiveMethodEnabled" value="false" />
<option name="javaModelName" value="CommentDO" /> <option name="javaModelName" value="CommentDO" />
<option name="moduleName" value="han-note-count-biz" /> <option name="moduleName" value="han-note-comment-biz" />
<option name="mybatisplusIdType" value="AUTO" /> <option name="mybatisplusIdType" value="AUTO" />
<option name="selectByPrimaryKeyEnabled" value="false" /> <option name="selectByPrimaryKeyEnabled" value="false" />
<option name="sequenceColumn" value="" /> <option name="sequenceColumn" value="" />
@@ -258,7 +258,7 @@
<option name="insertMethodEnabled" value="false" /> <option name="insertMethodEnabled" value="false" />
<option name="insertSelectiveMethodEnabled" value="false" /> <option name="insertSelectiveMethodEnabled" value="false" />
<option name="javaModelName" value="NoteCountDO" /> <option name="javaModelName" value="NoteCountDO" />
<option name="moduleName" value="han-note-count-biz" /> <option name="moduleName" value="han-note-comment-biz" />
<option name="mybatisplusIdType" value="ASSIGN_ID" /> <option name="mybatisplusIdType" value="ASSIGN_ID" />
<option name="selectByPrimaryKeyEnabled" value="false" /> <option name="selectByPrimaryKeyEnabled" value="false" />
<option name="sequenceColumn" value="" /> <option name="sequenceColumn" value="" />
@@ -427,7 +427,7 @@
<option name="updateByPrimaykeyEnabled" value="false" /> <option name="updateByPrimaykeyEnabled" value="false" />
<option name="userMybatisPlus" value="true" /> <option name="userMybatisPlus" value="true" />
<option name="xmlMapperPackage" value="mapperxml" /> <option name="xmlMapperPackage" value="mapperxml" />
<option name="xmlMapperPath" value="$PROJECT_DIR$/han-note-count/han-note-count-biz/src/main/resources" /> <option name="xmlMapperPath" value="$PROJECT_DIR$/han-note-comment/han-note-comment-biz/src/main/resources" />
</ProjectProfile> </ProjectProfile>
</option> </option>
</component> </component>

View File

@@ -6,8 +6,10 @@
<w>hannote</w> <w>hannote</w>
<w>hanserwei</w> <w>hanserwei</w>
<w>jobhandler</w> <w>jobhandler</w>
<w>mget</w>
<w>nacos</w> <w>nacos</w>
<w>operationlog</w> <w>operationlog</w>
<w>rbitmap</w>
<w>rustfs</w> <w>rustfs</w>
<w>zadd</w> <w>zadd</w>
<w>zrevrangebyscore</w> <w>zrevrangebyscore</w>

View File

@@ -125,6 +125,17 @@
<artifactId>spring-boot-starter-data-redis</artifactId> <artifactId>spring-boot-starter-data-redis</artifactId>
</dependency> </dependency>
<dependency>
<groupId>com.hanserwei</groupId>
<artifactId>han-note-user-api</artifactId>
</dependency>
<!-- Caffeine 本地缓存 -->
<dependency>
<groupId>com.github.ben-manes.caffeine</groupId>
<artifactId>caffeine</artifactId>
</dependency>
</dependencies> </dependencies>

View File

@@ -17,4 +17,29 @@ public interface MQConstants {
*/ */
String TOPIC_COMMENT_HEAT_UPDATE = "CommentHeatUpdateTopic"; String TOPIC_COMMENT_HEAT_UPDATE = "CommentHeatUpdateTopic";
/**
* Topic: 评论点赞、取消点赞共用一个 Topic
*/
String TOPIC_COMMENT_LIKE_OR_UNLIKE = "CommentLikeUnlikeTopic";
/**
* Topic: 删除本地缓存 —— 评论详情
*/
String TOPIC_DELETE_COMMENT_LOCAL_CACHE = "DeleteCommentDetailLocalCacheTopic";
/**
* Topic: 删除评论
*/
String TOPIC_DELETE_COMMENT = "DeleteCommentTopic";
/**
* Tag 标签:点赞
*/
String TAG_LIKE = "Like";
/**
* Tag 标签:取消点赞
*/
String TAG_UNLIKE = "UnLike";
} }

View File

@@ -7,6 +7,78 @@ public class RedisKeyConstants {
*/ */
private static final String HAVE_FIRST_REPLY_COMMENT_KEY_PREFIX = "comment:havaFirstReplyCommentId:"; private static final String HAVE_FIRST_REPLY_COMMENT_KEY_PREFIX = "comment:havaFirstReplyCommentId:";
/**
* Key 前缀:布隆过滤器 - 用户点赞的评论
*/
private static final String BLOOM_COMMENT_LIKES_KEY_PREFIX = "bloom:comment:likes:";
/**
* Key 前缀:二级评论分页 ZSET
*/
private static final String CHILD_COMMENT_LIST_KEY_PREFIX = "comment:childList:";
/**
* Hash Field: 子评论总数
*/
public static final String FIELD_CHILD_COMMENT_TOTAL = "childCommentTotal";
/**
* Hash Field: 点赞总数
*/
public static final String FIELD_LIKE_TOTAL = "likeTotal";
/**
* 评论维度计数 Key 前缀
*/
private static final String COUNT_COMMENT_KEY_PREFIX = "count:comment:";
/**
* Hash Field 键:评论总数
*/
public static final String FIELD_COMMENT_TOTAL = "commentTotal";
/**
* Key 前缀:笔记评论总数
*/
private static final String COUNT_COMMENT_TOTAL_KEY_PREFIX = "count:note:";
/**
* Key 前缀:评论分页 ZSET
*/
private static final String COMMENT_LIST_KEY_PREFIX = "comment:list:";
/**
* Key 前缀:评论详情 JSON
*/
private static final String COMMENT_DETAIL_KEY_PREFIX = "comment:detail:";
/**
* 构建 布隆过滤器 - 用户点赞的评论 完整 KEY
*
* @param userId 用户 ID
* @return 布隆过滤器 - 用户点赞的评论 完整 KEY
*/
public static String buildBloomCommentLikesKey(Long userId) {
return BLOOM_COMMENT_LIKES_KEY_PREFIX + userId;
}
/**
* 构建子评论分页 ZSET 完整 KEY
*
* @param commentId 一级评论 ID
* @return 子评论分页 ZSET 完整 KEY
*/
public static String buildChildCommentListKey(Long commentId) {
return CHILD_COMMENT_LIST_KEY_PREFIX + commentId;
}
/**
* 构建评论维度计数 Key
*
* @param commentId 评论 ID
* @return 评论维度计数 Key
*/
public static String buildCountCommentKey(Long commentId) {
return COUNT_COMMENT_KEY_PREFIX + commentId;
}
/** /**
* 构建完整 KEY * 构建完整 KEY
@@ -18,4 +90,34 @@ public class RedisKeyConstants {
return HAVE_FIRST_REPLY_COMMENT_KEY_PREFIX + commentId; return HAVE_FIRST_REPLY_COMMENT_KEY_PREFIX + commentId;
} }
/**
* 构建笔记评论总数完整 KEY
*
* @param noteId 笔记 ID
* @return 笔记评论总数完整 KEY
*/
public static String buildNoteCommentTotalKey(Long noteId) {
return COUNT_COMMENT_TOTAL_KEY_PREFIX + noteId;
}
/**
* 构建评论分页 ZSET 完整 KEY
*
* @param noteId 笔记 ID
* @return 评论分页 ZSET 完整 KEY
*/
public static String buildCommentListKey(Long noteId) {
return COMMENT_LIST_KEY_PREFIX + noteId;
}
/**
* 构建评论详情完整 KEY
*
* @param commentId 评论 ID
* @return 评论详情完整 KEY
*/
public static String buildCommentDetailKey(Object commentId) {
return COMMENT_DETAIL_KEY_PREFIX + commentId;
}
} }

View File

@@ -6,6 +6,7 @@ import com.google.common.collect.Maps;
import com.google.common.util.concurrent.RateLimiter; import com.google.common.util.concurrent.RateLimiter;
import com.hanserwei.framework.common.utils.JsonUtils; import com.hanserwei.framework.common.utils.JsonUtils;
import com.hanserwei.hannote.comment.biz.constants.MQConstants; import com.hanserwei.hannote.comment.biz.constants.MQConstants;
import com.hanserwei.hannote.comment.biz.constants.RedisKeyConstants;
import com.hanserwei.hannote.comment.biz.domain.dataobject.CommentDO; import com.hanserwei.hannote.comment.biz.domain.dataobject.CommentDO;
import com.hanserwei.hannote.comment.biz.domain.mapper.CommentDOMapper; import com.hanserwei.hannote.comment.biz.domain.mapper.CommentDOMapper;
import com.hanserwei.hannote.comment.biz.enums.CommentLevelEnum; import com.hanserwei.hannote.comment.biz.enums.CommentLevelEnum;
@@ -29,14 +30,15 @@ import org.apache.rocketmq.remoting.protocol.heartbeat.MessageModel;
import org.apache.rocketmq.spring.core.RocketMQTemplate; import org.apache.rocketmq.spring.core.RocketMQTemplate;
import org.springframework.beans.factory.annotation.Value; import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Bean;
import org.springframework.core.io.ClassPathResource;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.script.DefaultRedisScript;
import org.springframework.messaging.support.MessageBuilder; import org.springframework.messaging.support.MessageBuilder;
import org.springframework.scripting.support.ResourceScriptSource;
import org.springframework.stereotype.Component; import org.springframework.stereotype.Component;
import org.springframework.transaction.support.TransactionTemplate; import org.springframework.transaction.support.TransactionTemplate;
import java.util.List; import java.util.*;
import java.util.Map;
import java.util.Objects;
import java.util.UUID;
import java.util.stream.Collectors; import java.util.stream.Collectors;
@SuppressWarnings("UnstableApiUsage") @SuppressWarnings("UnstableApiUsage")
@@ -55,13 +57,15 @@ public class Comment2DBConsumer {
private KeyValueRpcService keyValueRpcService; private KeyValueRpcService keyValueRpcService;
@Resource @Resource
private RocketMQTemplate rocketMQTemplate; private RocketMQTemplate rocketMQTemplate;
@Resource
private RedisTemplate<String, Object> redisTemplate;
private DefaultMQPushConsumer consumer; private DefaultMQPushConsumer consumer;
// 每秒创建 1000 个令牌 // 每秒创建 1000 个令牌
private final RateLimiter rateLimiter = RateLimiter.create(1000); private final RateLimiter rateLimiter = RateLimiter.create(1000);
@Bean @Bean(name = "Comment2DBConsumer")
public DefaultMQPushConsumer mqPushConsumer() throws MQClientException { public DefaultMQPushConsumer mqPushConsumer() throws MQClientException {
// Group组 // Group组
String group = "han_note_group_" + MQConstants.TOPIC_PUBLISH_COMMENT; String group = "han_note_group_" + MQConstants.TOPIC_PUBLISH_COMMENT;
@@ -183,7 +187,10 @@ public class Comment2DBConsumer {
.toList(); .toList();
if (CollUtil.isNotEmpty(commentContentNotEmptyBOS)) { if (CollUtil.isNotEmpty(commentContentNotEmptyBOS)) {
// 批量存入评论内容 // 批量存入评论内容
keyValueRpcService.batchSaveCommentContent(commentContentNotEmptyBOS); boolean result = keyValueRpcService.batchSaveCommentContent(commentContentNotEmptyBOS);
if (!result) {
throw new RuntimeException("批量保存评论内容失败");
}
} }
return count; return count;
@@ -210,6 +217,9 @@ public class Comment2DBConsumer {
org.springframework.messaging.Message<String> message = MessageBuilder.withPayload(JsonUtils.toJsonString(countPublishCommentMqDTOS)) org.springframework.messaging.Message<String> message = MessageBuilder.withPayload(JsonUtils.toJsonString(countPublishCommentMqDTOS))
.build(); .build();
// 同步一级评论到 Redis 热点评论 ZSET 中
syncOneLevelComment2RedisZSet(commentBOS);
// 异步发送 MQ 消息 // 异步发送 MQ 消息
rocketMQTemplate.asyncSend(MQConstants.TOPIC_COUNT_NOTE_COMMENT, message, new SendCallback() { rocketMQTemplate.asyncSend(MQConstants.TOPIC_COUNT_NOTE_COMMENT, message, new SendCallback() {
@Override @Override
@@ -238,6 +248,40 @@ public class Comment2DBConsumer {
return consumer; return consumer;
} }
/**
* 同步一级评论到 Redis 热点评论 ZSET 中
*
* @param commentBOS 评论 BO 列表
*/
private void syncOneLevelComment2RedisZSet(List<CommentBO> commentBOS) {
// 过滤出一级评论,并按所属笔记进行分组,转换为一个 Map 字典
Map<Long, List<CommentBO>> commentIdAndBOListMap = commentBOS.stream()
.filter(commentBO -> Objects.equals(commentBO.getLevel(), CommentLevelEnum.ONE.getCode())) // 仅过滤一级评论
.collect(Collectors.groupingBy(CommentBO::getNoteId));
// 循环字典
commentIdAndBOListMap.forEach((noteId, commentBOList) -> {
// 构建 Redis 热点评论 ZSET Key
String key = RedisKeyConstants.buildCommentListKey(noteId);
DefaultRedisScript<Long> script = new DefaultRedisScript<>();
// Lua 脚本路径
script.setScriptSource(new ResourceScriptSource(new ClassPathResource("/lua/add_hot_comments.lua")));
// 返回值类型
script.setResultType(Long.class);
// 构建执行 Lua 脚本所需的 ARGS 参数
List<Object> args = Lists.newArrayList();
commentBOList.forEach(commentBO -> {
args.add(commentBO.getId()); // Member: 评论ID
args.add(0); // Score: 热度值,初始值为 0
});
// 执行 Lua 脚本
redisTemplate.execute(script, Collections.singletonList(key), args.toArray());
});
}
@PreDestroy @PreDestroy
public void destroy() { public void destroy() {
if (Objects.nonNull(consumer)) { if (Objects.nonNull(consumer)) {

View File

@@ -5,6 +5,7 @@ import com.google.common.collect.Lists;
import com.google.common.collect.Sets; import com.google.common.collect.Sets;
import com.hanserwei.framework.common.utils.JsonUtils; import com.hanserwei.framework.common.utils.JsonUtils;
import com.hanserwei.hannote.comment.biz.constants.MQConstants; import com.hanserwei.hannote.comment.biz.constants.MQConstants;
import com.hanserwei.hannote.comment.biz.constants.RedisKeyConstants;
import com.hanserwei.hannote.comment.biz.domain.dataobject.CommentDO; import com.hanserwei.hannote.comment.biz.domain.dataobject.CommentDO;
import com.hanserwei.hannote.comment.biz.domain.mapper.CommentDOMapper; import com.hanserwei.hannote.comment.biz.domain.mapper.CommentDOMapper;
import com.hanserwei.hannote.comment.biz.model.bo.CommentHeatBO; import com.hanserwei.hannote.comment.biz.model.bo.CommentHeatBO;
@@ -13,12 +14,19 @@ import jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.apache.rocketmq.spring.annotation.RocketMQMessageListener; import org.apache.rocketmq.spring.annotation.RocketMQMessageListener;
import org.apache.rocketmq.spring.core.RocketMQListener; import org.apache.rocketmq.spring.core.RocketMQListener;
import org.springframework.core.io.ClassPathResource;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.script.DefaultRedisScript;
import org.springframework.scripting.support.ResourceScriptSource;
import org.springframework.stereotype.Component; import org.springframework.stereotype.Component;
import java.math.BigDecimal; import java.math.BigDecimal;
import java.time.Duration; import java.time.Duration;
import java.util.Collections;
import java.util.List; import java.util.List;
import java.util.Map;
import java.util.Set; import java.util.Set;
import java.util.stream.Collectors;
@Component @Component
@RocketMQMessageListener(consumerGroup = "han_note_group_" + MQConstants.TOPIC_COMMENT_HEAT_UPDATE, // Group 组 @RocketMQMessageListener(consumerGroup = "han_note_group_" + MQConstants.TOPIC_COMMENT_HEAT_UPDATE, // Group 组
@@ -29,6 +37,8 @@ public class CommentHeatUpdateConsumer implements RocketMQListener<String> {
@Resource @Resource
private CommentDOMapper commentDOMapper; private CommentDOMapper commentDOMapper;
@Resource
private RedisTemplate<String, Object> redisTemplate;
private final BufferTrigger<String> bufferTrigger = BufferTrigger.<String>batchBlocking() private final BufferTrigger<String> bufferTrigger = BufferTrigger.<String>batchBlocking()
.bufferSize(50000) // 缓存队列的最大容量 .bufferSize(50000) // 缓存队列的最大容量
@@ -82,9 +92,47 @@ public class CommentHeatUpdateConsumer implements RocketMQListener<String> {
commentBOS.add(CommentHeatBO.builder() commentBOS.add(CommentHeatBO.builder()
.id(commentId) .id(commentId)
.heat(heatNum.doubleValue()) .heat(heatNum.doubleValue())
.noteId(commentDO.getNoteId())
.build()); .build());
}); });
// 批量更新评论热度值 // 批量更新评论热度值
commentDOMapper.batchUpdateHeatByCommentIds(ids, commentBOS); commentDOMapper.batchUpdateHeatByCommentIds(ids, commentBOS);
// 更新 Redis 中热度评论 ZSET
updateRedisHotComments(commentBOS);
}
/**
* 更新 Redis 中热点评论 ZSET
*
* @param commentHeatBOList 热度值 BO 列表
*/
private void updateRedisHotComments(List<CommentHeatBO> commentHeatBOList) {
// 过滤出热度值大于 0 的,并按所属笔记 ID 分组若热度等于0则不进行更新
Map<Long, List<CommentHeatBO>> noteIdAndBOListMap = commentHeatBOList.stream()
.filter(commentHeatBO -> commentHeatBO.getHeat() > 0)
.collect(Collectors.groupingBy(CommentHeatBO::getNoteId));
// 循环
noteIdAndBOListMap.forEach((noteId, commentHeatBOS) -> {
// 构建热点评论 Redis Key
String key = RedisKeyConstants.buildCommentListKey(noteId);
DefaultRedisScript<Long> script = new DefaultRedisScript<>();
// Lua 脚本路径
script.setScriptSource(new ResourceScriptSource(new ClassPathResource("/lua/update_hot_comments.lua")));
// 返回值类型
script.setResultType(Long.class);
// 构建执行 Lua 脚本所需的 ARGS 参数
List<Object> args = Lists.newArrayList();
commentHeatBOS.forEach(commentHeatBO -> {
args.add(commentHeatBO.getId()); // Member: 评论ID
args.add(commentHeatBO.getHeat()); // Score: 热度值
});
// 执行 Lua 脚本
redisTemplate.execute(script, Collections.singletonList(key), args.toArray());
});
} }
} }

View File

@@ -0,0 +1,184 @@
package com.hanserwei.hannote.comment.biz.consumer;
import cn.hutool.core.collection.CollUtil;
import com.google.common.collect.Lists;
import com.google.common.collect.Sets;
import com.google.common.util.concurrent.RateLimiter;
import com.hanserwei.framework.common.utils.JsonUtils;
import com.hanserwei.hannote.comment.biz.constants.MQConstants;
import com.hanserwei.hannote.comment.biz.constants.RedisKeyConstants;
import com.hanserwei.hannote.comment.biz.domain.dataobject.CommentDO;
import com.hanserwei.hannote.comment.biz.domain.mapper.CommentDOMapper;
import com.hanserwei.hannote.comment.biz.domain.mapper.NoteCountDOMapper;
import com.hanserwei.hannote.comment.biz.enums.CommentLevelEnum;
import jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import org.apache.rocketmq.client.producer.SendCallback;
import org.apache.rocketmq.client.producer.SendResult;
import org.apache.rocketmq.spring.annotation.RocketMQMessageListener;
import org.apache.rocketmq.spring.core.RocketMQListener;
import org.apache.rocketmq.spring.core.RocketMQTemplate;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.messaging.support.MessageBuilder;
import org.springframework.stereotype.Component;
import java.util.List;
import java.util.Objects;
import java.util.Set;
@SuppressWarnings("UnstableApiUsage")
@Component
@Slf4j
@RocketMQMessageListener(consumerGroup = "han_note_group_" + MQConstants.TOPIC_DELETE_COMMENT, // Group
topic = MQConstants.TOPIC_DELETE_COMMENT // 消费的主题 Topic
)
public class DeleteCommentConsumer implements RocketMQListener<String> {
// 每秒创建 1000 个令牌
private final RateLimiter rateLimiter = RateLimiter.create(1000);
@Resource
private CommentDOMapper commentDOMapper;
@Resource
private NoteCountDOMapper noteCountDOMapper;
@Resource
private RedisTemplate<String, Object> redisTemplate;
@Resource
private RocketMQTemplate rocketMQTemplate;
@Override
public void onMessage(String body) {
// 令牌桶流控
rateLimiter.acquire();
log.info("## 【删除评论 - 后续业务处理】消费者消费成功, body: {}", body);
CommentDO commentDO = JsonUtils.parseObject(body, CommentDO.class);
// 评论级别
Integer level = null;
if (commentDO != null) {
level = commentDO.getLevel();
}
CommentLevelEnum commentLevelEnum = CommentLevelEnum.valueOf(level);
if (commentLevelEnum != null) {
switch (commentLevelEnum) {
case ONE -> // 一级评论
handleOneLevelComment(commentDO);
case TWO -> // 二级评论
handleTwoLevelComment(commentDO);
}
}
}
/**
* 一级评论处理
*
* @param commentDO 评论
*/
private void handleOneLevelComment(CommentDO commentDO) {
Long commentId = commentDO.getId();
Long noteId = commentDO.getNoteId();
// 1. 关联评论删除(一级评论下所有子评论,都需要删除)
int count = commentDOMapper.deleteByParentId(commentId);
// 2. 计数更新(笔记下总评论数)
// 更新 Redis 缓存
String redisKey = RedisKeyConstants.buildNoteCommentTotalKey(noteId);
boolean hasKey = redisTemplate.hasKey(redisKey);
if (hasKey) {
// 笔记评论总数 -1
redisTemplate.opsForHash().increment(redisKey, RedisKeyConstants.FIELD_COMMENT_TOTAL, -(count + 1));
}
// 更新 t_note_count 计数表
noteCountDOMapper.updateCommentTotalByNoteId(noteId, -(count + 1));
}
/**
* 二级评论处理
*
* @param commentDO 评论
*/
private void handleTwoLevelComment(CommentDO commentDO) {
Long commentId = commentDO.getId();
// 1. 批量删除关联评论(递归查询回复评论,并批量删除)
List<Long> replyCommentIds = Lists.newArrayList();
recurrentGetReplyCommentId(replyCommentIds, commentId);
// 被删除的行数
int count = 0;
if (CollUtil.isNotEmpty(replyCommentIds)) {
count = commentDOMapper.deleteByIds(replyCommentIds);
}
// 2. 更新一级评论的计数
Long parentCommentId = commentDO.getParentId();
String redisKey = RedisKeyConstants.buildCountCommentKey(parentCommentId);
boolean hasKey = redisTemplate.hasKey(redisKey);
if (hasKey) {
redisTemplate.opsForHash().increment(redisKey, RedisKeyConstants.FIELD_CHILD_COMMENT_TOTAL, -(count + 1));
}
// 3. 若是最早的发布的二级评论被删除,需要更新一级评论的 first_reply_comment_id
// 查询一级评论
CommentDO oneLevelCommentDO = commentDOMapper.selectById(parentCommentId);
Long firstReplyCommentId = oneLevelCommentDO.getFirstReplyCommentId();
// 若删除的是最早回复的二级评论
if (Objects.equals(firstReplyCommentId, commentId)) {
// 查询数据库,重新获取一级评论最早回复的评论
CommentDO earliestCommentDO = commentDOMapper.selectEarliestByParentId(parentCommentId);
// 最早回复的那条评论 ID。若查询结果为 null, 则最早回复的评论 ID 为 null
Long earliestCommentId = Objects.nonNull(earliestCommentDO) ? earliestCommentDO.getId() : null;
// 更新其一级评论的 first_reply_comment_id
commentDOMapper.updateFirstReplyCommentIdByPrimaryKey(earliestCommentId, parentCommentId);
}
// 4. 重新计算一级评论的热度值
Set<Long> commentIds = Sets.newHashSetWithExpectedSize(1);
commentIds.add(parentCommentId);
// 异步发送计数 MQ, 更新评论热度值
org.springframework.messaging.Message<String> message = MessageBuilder.withPayload(JsonUtils.toJsonString(commentIds))
.build();
// 异步发送 MQ 消息
rocketMQTemplate.asyncSend(MQConstants.TOPIC_COMMENT_HEAT_UPDATE, message, new SendCallback() {
@Override
public void onSuccess(SendResult sendResult) {
log.info("==> 【评论热度值更新】MQ 发送成功SendResult: {}", sendResult);
}
@Override
public void onException(Throwable throwable) {
log.error("==> 【评论热度值更新】MQ 发送异常: ", throwable);
}
});
}
/**
* 递归获取全部回复的评论 ID
*
* @param commentIds 评论 ID 列表
* @param commentId 评论 ID
*/
private void recurrentGetReplyCommentId(List<Long> commentIds, Long commentId) {
CommentDO replyCommentDO = commentDOMapper.selectByReplyCommentId(commentId);
if (Objects.isNull(replyCommentDO)) return;
commentIds.add(replyCommentDO.getId());
Long replyCommentId = replyCommentDO.getId();
// 递归调用
recurrentGetReplyCommentId(commentIds, replyCommentId);
}
}

View File

@@ -0,0 +1,29 @@
package com.hanserwei.hannote.comment.biz.consumer;
import com.hanserwei.hannote.comment.biz.constants.MQConstants;
import com.hanserwei.hannote.comment.biz.service.CommentService;
import jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import org.apache.rocketmq.spring.annotation.MessageModel;
import org.apache.rocketmq.spring.annotation.RocketMQMessageListener;
import org.apache.rocketmq.spring.core.RocketMQListener;
import org.springframework.stereotype.Component;
@Component
@Slf4j
@RocketMQMessageListener(consumerGroup = "han_note_group_" + MQConstants.TOPIC_DELETE_COMMENT_LOCAL_CACHE, // Group
topic = MQConstants.TOPIC_DELETE_COMMENT_LOCAL_CACHE, // 消费的主题 Topic
messageModel = MessageModel.BROADCASTING) // 广播模式
public class DeleteCommentLocalCacheConsumer implements RocketMQListener<String> {
@Resource
private CommentService commentService;
@Override
public void onMessage(String body) {
Long commentId = Long.valueOf(body);
log.info("## 消费者消费成功, commentId: {}", commentId);
commentService.deleteCommentLocalCache(commentId);
}
}

View File

@@ -0,0 +1,156 @@
package com.hanserwei.hannote.comment.biz.consumer;
import cn.hutool.core.collection.CollUtil;
import com.google.common.collect.Lists;
import com.google.common.util.concurrent.RateLimiter;
import com.hanserwei.framework.common.utils.JsonUtils;
import com.hanserwei.hannote.comment.biz.constants.MQConstants;
import com.hanserwei.hannote.comment.biz.domain.mapper.CommentLikeDOMapper;
import com.hanserwei.hannote.comment.biz.enums.LikeUnlikeCommentTypeEnum;
import com.hanserwei.hannote.comment.biz.model.dto.LikeUnlikeCommentMqDTO;
import jakarta.annotation.PreDestroy;
import jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import org.apache.rocketmq.client.consumer.DefaultMQPushConsumer;
import org.apache.rocketmq.client.consumer.listener.ConsumeOrderlyStatus;
import org.apache.rocketmq.client.consumer.listener.MessageListenerOrderly;
import org.apache.rocketmq.client.exception.MQClientException;
import org.apache.rocketmq.common.consumer.ConsumeFromWhere;
import org.apache.rocketmq.remoting.protocol.heartbeat.MessageModel;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.stereotype.Component;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.function.Function;
import java.util.stream.Collectors;
@SuppressWarnings("UnstableApiUsage")
@Component
@Slf4j
public class LikeUnlikeComment2DBConsumer {
// 每秒创建 5000 个令牌
private final RateLimiter rateLimiter = RateLimiter.create(5000);
@Value("${rocketmq.name-server}")
private String nameServer;
@Resource
private CommentLikeDOMapper commentLikeDOMapper;
private DefaultMQPushConsumer consumer;
@Bean(name = "LikeUnlikeComment2DBConsumer")
public DefaultMQPushConsumer mqPushConsumer() throws MQClientException {
// Group 组
String group = "han_note_group_" + MQConstants.TOPIC_COMMENT_LIKE_OR_UNLIKE;
// 创建一个新的 DefaultMQPushConsumer 实例,并指定消费者的消费组名
consumer = new DefaultMQPushConsumer(group);
// 设置 RocketMQ 的 NameServer 地址
consumer.setNamesrvAddr(nameServer);
// 订阅指定的主题,并设置主题的订阅规则("*" 表示订阅所有标签的消息)
consumer.subscribe(MQConstants.TOPIC_COMMENT_LIKE_OR_UNLIKE, "*");
// 设置消费者消费消息的起始位置,如果队列中没有消息,则从最新的消息开始消费。
consumer.setConsumeFromWhere(ConsumeFromWhere.CONSUME_FROM_LAST_OFFSET);
// 设置消息消费模式,这里使用集群模式 (CLUSTERING)
consumer.setMessageModel(MessageModel.CLUSTERING);
// 最大重试次数, 以防消息重试过多次仍然没有成功,避免消息卡在消费队列中。
consumer.setMaxReconsumeTimes(3);
// 设置每批次消费的最大消息数量,这里设置为 30表示每次拉取时最多消费 30 条消息。
consumer.setConsumeMessageBatchMaxSize(30);
// 注册消息监听器
consumer.registerMessageListener((MessageListenerOrderly) (msgs, context) -> {
log.info("==> 【评论点赞、取消点赞】本批次消息大小: {}", msgs.size());
try {
// 令牌桶流控, 以控制数据库能够承受的 QPS
rateLimiter.acquire();
// 将批次 Json 消息体转换 DTO 集合
List<LikeUnlikeCommentMqDTO> likeUnlikeCommentMqDTOS = Lists.newArrayList();
msgs.forEach(msg -> {
String tag = msg.getTags(); // Tag 标签
String msgJson = new String(msg.getBody()); // 消息体 Json 字符串
log.info("==> 【评论点赞、取消点赞】Consumer - Tag: {}, Received message: {}", tag, msgJson);
// Json 转 DTO
likeUnlikeCommentMqDTOS.add(JsonUtils.parseObject(msgJson, LikeUnlikeCommentMqDTO.class));
});
// 按评论 ID 分组
Map<Long, List<LikeUnlikeCommentMqDTO>> commentIdAndListMap = likeUnlikeCommentMqDTOS.stream()
.collect(Collectors.groupingBy(LikeUnlikeCommentMqDTO::getCommentId));
List<LikeUnlikeCommentMqDTO> finalLikeUnlikeCommentMqDTOS = Lists.newArrayList();
commentIdAndListMap.forEach((commentId, ops) -> {
// 优化:若某个用户对某评论,多次操作,如点赞 -> 取消点赞 -> 点赞,需进行操作合并,只提取最后一次操作,进一步降低操作数据库的频率
Map<Long, LikeUnlikeCommentMqDTO> userLastOp = ops.stream()
.collect(Collectors.toMap(
LikeUnlikeCommentMqDTO::getUserId, // 以发布评论的用户 ID 作为 Map 的键
Function.identity(), // 直接使用 DTO 对象本身作为 Map 的值
// 合并策略:当出现重复键(同一用户多次操作)时,保留时间更晚的记录
(oldValue, newValue) ->
oldValue.getCreateTime().isAfter(newValue.getCreateTime()) ? oldValue : newValue
));
finalLikeUnlikeCommentMqDTOS.addAll(userLastOp.values());
});
// 批量操作数据库
executeBatchSQL(finalLikeUnlikeCommentMqDTOS);
// 手动 ACK告诉 RocketMQ 这批次消息消费成功
return ConsumeOrderlyStatus.SUCCESS;
} catch (Exception e) {
log.error("", e);
// 这样 RocketMQ 会暂停当前队列的消费一段时间,再重试
return ConsumeOrderlyStatus.SUSPEND_CURRENT_QUEUE_A_MOMENT;
}
});
// 启动消费者
consumer.start();
return consumer;
}
private void executeBatchSQL(List<LikeUnlikeCommentMqDTO> values) {
// 过滤出点赞操作
List<LikeUnlikeCommentMqDTO> likes = values.stream()
.filter(op -> Objects.equals(op.getType(), LikeUnlikeCommentTypeEnum.LIKE.getCode()))
.toList();
// 过滤出取消点赞操作
List<LikeUnlikeCommentMqDTO> unlikes = values.stream()
.filter(op -> Objects.equals(op.getType(), LikeUnlikeCommentTypeEnum.UNLIKE.getCode()))
.toList();
// 取消点赞:批量删除
if (CollUtil.isNotEmpty(unlikes)) {
commentLikeDOMapper.batchDelete(unlikes);
}
// 点赞:批量新增
if (CollUtil.isNotEmpty(likes)) {
commentLikeDOMapper.batchInsert(likes);
}
}
@PreDestroy
public void destroy() {
if (Objects.nonNull(consumer)) {
try {
consumer.shutdown(); // 关闭消费者
} catch (Exception e) {
log.error("", e);
}
}
}
}

View File

@@ -1,8 +1,9 @@
package com.hanserwei.hannote.comment.biz.controller; package com.hanserwei.hannote.comment.biz.controller;
import com.hanserwei.framework.biz.operationlog.aspect.ApiOperationLog; import com.hanserwei.framework.biz.operationlog.aspect.ApiOperationLog;
import com.hanserwei.framework.common.response.PageResponse;
import com.hanserwei.framework.common.response.Response; import com.hanserwei.framework.common.response.Response;
import com.hanserwei.hannote.comment.biz.model.vo.PublishCommentReqVO; import com.hanserwei.hannote.comment.biz.model.vo.*;
import com.hanserwei.hannote.comment.biz.service.CommentService; import com.hanserwei.hannote.comment.biz.service.CommentService;
import jakarta.annotation.Resource; import jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
@@ -26,4 +27,34 @@ public class CommentController {
return commentService.publishComment(publishCommentReqVO); return commentService.publishComment(publishCommentReqVO);
} }
@PostMapping("/list")
@ApiOperationLog(description = "评论分页查询")
public PageResponse<FindCommentItemRspVO> findCommentPageList(@Validated @RequestBody FindCommentPageListReqVO findCommentPageListReqVO) {
return commentService.findCommentPageList(findCommentPageListReqVO);
}
@PostMapping("/child/list")
@ApiOperationLog(description = "二级评论分页查询")
public PageResponse<FindChildCommentItemRspVO> findChildCommentPageList(@Validated @RequestBody FindChildCommentPageListReqVO findChildCommentPageListReqVO) {
return commentService.findChildCommentPageList(findChildCommentPageListReqVO);
}
@PostMapping("/like")
@ApiOperationLog(description = "评论点赞")
public Response<?> likeComment(@Validated @RequestBody LikeCommentReqVO likeCommentReqVO) {
return commentService.likeComment(likeCommentReqVO);
}
@PostMapping("/unlike")
@ApiOperationLog(description = "评论取消点赞")
public Response<?> unlikeComment(@Validated @RequestBody UnLikeCommentReqVO unLikeCommentReqVO) {
return commentService.unlikeComment(unLikeCommentReqVO);
}
@PostMapping("/delete")
@ApiOperationLog(description = "删除评论")
public Response<?> deleteComment(@Validated @RequestBody DeleteCommentReqVO deleteCommentReqVO) {
return commentService.deleteComment(deleteCommentReqVO);
}
} }

View File

@@ -104,12 +104,6 @@ public class CommentDO {
@TableField(value = "create_time") @TableField(value = "create_time")
private LocalDateTime createTime; private LocalDateTime createTime;
/**
* 下级评论总数
*/
@TableField(value = "child_comment_total")
private Long childCommentTotal;
/** /**
* 更新时间 * 更新时间
*/ */
@@ -117,7 +111,19 @@ public class CommentDO {
private LocalDateTime updateTime; private LocalDateTime updateTime;
/** /**
* 级评论的第一个回复的评论ID * 级评论总数(只有一级评论才需要统计)
*/
@TableField(value = "child_comment_total")
private Long childCommentTotal;
/**
* 评论热度
*/
@TableField(value = "heat")
private Double heat;
/**
* 最早回复的评论ID (只有一级评论需要)
*/ */
@TableField(value = "first_reply_comment_id") @TableField(value = "first_reply_comment_id")
private Long firstReplyCommentId; private Long firstReplyCommentId;

View File

@@ -0,0 +1,50 @@
package com.hanserwei.hannote.comment.biz.domain.dataobject;
import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableName;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;
/**
* 笔记计数表
*/
@Data
@Builder
@AllArgsConstructor
@NoArgsConstructor
@TableName(value = "t_note_count")
public class NoteCountDO {
/**
* 主键ID
*/
@TableId(value = "id", type = IdType.ASSIGN_ID)
private Long id;
/**
* 笔记ID
*/
@TableField(value = "note_id")
private Long noteId;
/**
* 获得点赞总数
*/
@TableField(value = "like_total")
private Long likeTotal;
/**
* 获得收藏总数
*/
@TableField(value = "collect_total")
private Long collectTotal;
/**
* 被评论总数
*/
@TableField(value = "comment_total")
private Long commentTotal;
}

View File

@@ -11,7 +11,6 @@ import java.util.List;
@Mapper @Mapper
public interface CommentDOMapper extends BaseMapper<CommentDO> { public interface CommentDOMapper extends BaseMapper<CommentDO> {
/** /**
* 根据评论 ID 批量查询 * 根据评论 ID 批量查询
* *
@@ -55,4 +54,96 @@ public interface CommentDOMapper extends BaseMapper<CommentDO> {
*/ */
int updateFirstReplyCommentIdByPrimaryKey(@Param("firstReplyCommentId") Long firstReplyCommentId, int updateFirstReplyCommentIdByPrimaryKey(@Param("firstReplyCommentId") Long firstReplyCommentId,
@Param("id") Long id); @Param("id") Long id);
/**
* 查询评论分页数据
*
* @param noteId 笔记 ID
* @param offset 偏移量
* @param pageSize 页大小
* @return 评论分页数据
*/
List<CommentDO> selectPageList(@Param("noteId") Long noteId,
@Param("offset") long offset,
@Param("pageSize") long pageSize);
/**
* 批量查询二级评论
*
* @param commentIds 评论 ID 列表
* @return 二级评论
*/
List<CommentDO> selectTwoLevelCommentByIds(@Param("commentIds") List<Long> commentIds);
/**
* 查询热门评论
*
* @param noteId 笔记 ID
* @return 热门评论
*/
List<CommentDO> selectHeatComments(Long noteId);
/**
* 查询一级评论下子评论总数
*
* @param commentId 一级评论 ID
* @return 一级评论下子评论总数
*/
Long selectChildCommentTotalById(Long commentId);
/**
* 查询二级评论分页数据
*
* @param parentId 一级评论 ID
* @param offset 偏移量
* @param pageSize 页大小
* @return 二级评论分页数据
*/
List<CommentDO> selectChildPageList(@Param("parentId") Long parentId,
@Param("offset") long offset,
@Param("pageSize") long pageSize);
/**
* 批量查询计数数据
*
* @param commentIds 评论 ID 列表
* @return 计数数据
*/
List<CommentDO> selectCommentCountByIds(@Param("commentIds") List<Long> commentIds);
/**
* 查询子评论
*
* @param parentId 一级评论 ID
* @param limit 子评论数量限制
* @return 子评论
*/
List<CommentDO> selectChildCommentsByParentIdAndLimit(@Param("parentId") Long parentId,
@Param("limit") int limit);
/**
* 删除一级评论下,所有二级评论
*
* @param commentId 一级评论 ID
* @return 删除数量
*/
int deleteByParentId(Long commentId);
/**
* 批量删除评论
*
* @param commentIds 评论 ID 列表
* @return 删除数量
*/
int deleteByIds(@Param("commentIds") List<Long> commentIds);
/**
* 根据 reply_comment_id 查询
*
* @param commentId 回复的评论 ID
* @return 评论
*/
CommentDO selectByReplyCommentId(Long commentId);
} }

View File

@@ -2,8 +2,46 @@ package com.hanserwei.hannote.comment.biz.domain.mapper;
import com.baomidou.mybatisplus.core.mapper.BaseMapper; import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.hanserwei.hannote.comment.biz.domain.dataobject.CommentLikeDO; import com.hanserwei.hannote.comment.biz.domain.dataobject.CommentLikeDO;
import com.hanserwei.hannote.comment.biz.model.dto.LikeUnlikeCommentMqDTO;
import org.apache.ibatis.annotations.Mapper; import org.apache.ibatis.annotations.Mapper;
import org.apache.ibatis.annotations.Param;
import java.util.List;
@Mapper @Mapper
public interface CommentLikeDOMapper extends BaseMapper<CommentLikeDO> { public interface CommentLikeDOMapper extends BaseMapper<CommentLikeDO> {
/**
* 查询某个评论是否被点赞
*
* @param userId 用户 ID
* @param commentId 评论 ID
* @return 1 表示已点赞0 表示未点赞
*/
int selectCountByUserIdAndCommentId(@Param("userId") Long userId,
@Param("commentId") Long commentId);
/**
* 查询对应用户点赞的所有评论
*
* @param userId 用户 ID
* @return 评论点赞列表
*/
List<CommentLikeDO> selectByUserId(@Param("userId") Long userId);
/**
* 批量删除点赞记录
*
* @param unlikes 删除点赞记录
* @return 删除数量
*/
int batchDelete(@Param("unlikes") List<LikeUnlikeCommentMqDTO> unlikes);
/**
* 批量添加点赞记录
*
* @param likes 添加点赞记录
* @return 添加数量
*/
int batchInsert(@Param("likes") List<LikeUnlikeCommentMqDTO> likes);
} }

View File

@@ -0,0 +1,28 @@
package com.hanserwei.hannote.comment.biz.domain.mapper;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.hanserwei.hannote.comment.biz.domain.dataobject.NoteCountDO;
import org.apache.ibatis.annotations.Mapper;
import org.apache.ibatis.annotations.Param;
@Mapper
public interface NoteCountDOMapper extends BaseMapper<NoteCountDO> {
/**
* 查询笔记评论总数
*
* @param noteId 笔记ID
* @return 笔记评论总数
*/
Long selectCommentTotalByNoteId(Long noteId);
/**
* 更新评论总数
*
* @param noteId 笔记 ID
* @param count 评论总数
* @return 更新数量
*/
int updateCommentTotalByNoteId(@Param("noteId") Long noteId,
@Param("count") int count);
}

View File

@@ -3,6 +3,8 @@ package com.hanserwei.hannote.comment.biz.enums;
import lombok.AllArgsConstructor; import lombok.AllArgsConstructor;
import lombok.Getter; import lombok.Getter;
import java.util.Objects;
@Getter @Getter
@AllArgsConstructor @AllArgsConstructor
public enum CommentLevelEnum { public enum CommentLevelEnum {
@@ -14,4 +16,18 @@ public enum CommentLevelEnum {
private final Integer code; private final Integer code;
/**
* 根据类型 code 获取对应的枚举
*
* @param code 类型 code
* @return 枚举
*/
public static CommentLevelEnum valueOf(Integer code) {
for (CommentLevelEnum commentLevelEnum : CommentLevelEnum.values()) {
if (Objects.equals(code, commentLevelEnum.getCode())) {
return commentLevelEnum;
}
}
return null;
}
} }

View File

@@ -0,0 +1,37 @@
package com.hanserwei.hannote.comment.biz.enums;
import lombok.AllArgsConstructor;
import lombok.Getter;
import java.util.Objects;
@Getter
@AllArgsConstructor
public enum CommentLikeLuaResultEnum {
// 布隆过滤器不存在
NOT_EXIST(-1L),
// 评论已点赞
COMMENT_LIKED(1L),
// 评论点赞成功
COMMENT_LIKE_SUCCESS(0L),
;
private final Long code;
/**
* 根据类型 code 获取对应的枚举
*
* @param code 类型 code
* @return 枚举
*/
public static CommentLikeLuaResultEnum valueOf(Long code) {
for (CommentLikeLuaResultEnum commentLikeLuaResultEnum : CommentLikeLuaResultEnum.values()) {
if (Objects.equals(code, commentLikeLuaResultEnum.getCode())) {
return commentLikeLuaResultEnum;
}
}
return null;
}
}

View File

@@ -0,0 +1,35 @@
package com.hanserwei.hannote.comment.biz.enums;
import lombok.AllArgsConstructor;
import lombok.Getter;
import java.util.Objects;
@Getter
@AllArgsConstructor
public enum CommentUnlikeLuaResultEnum {
// 布隆过滤器不存在
NOT_EXIST(-1L),
// 评论已点赞
COMMENT_LIKED(1L),
// 评论未点赞
COMMENT_NOT_LIKED(0L),
;
private final Long code;
/**
* 根据类型 code 获取对应的枚举
*
* @param code 类型 code
* @return 枚举
*/
public static CommentUnlikeLuaResultEnum valueOf(Long code) {
for (CommentUnlikeLuaResultEnum commentUnlikeLuaResultEnum : CommentUnlikeLuaResultEnum.values()) {
if (Objects.equals(code, commentUnlikeLuaResultEnum.getCode())) {
return commentUnlikeLuaResultEnum;
}
}
return null;
}
}

View File

@@ -0,0 +1,17 @@
package com.hanserwei.hannote.comment.biz.enums;
import lombok.AllArgsConstructor;
import lombok.Getter;
@Getter
@AllArgsConstructor
public enum LikeUnlikeCommentTypeEnum {
// 点赞
LIKE(1),
// 取消点赞
UNLIKE(0),
;
private final Integer code;
}

View File

@@ -13,6 +13,11 @@ public enum ResponseCodeEnum implements BaseExceptionInterface {
PARAM_NOT_VALID("COMMENT-10001", "参数错误"), PARAM_NOT_VALID("COMMENT-10001", "参数错误"),
// ----------- 业务异常状态码 ----------- // ----------- 业务异常状态码 -----------
COMMENT_NOT_FOUND("COMMENT-20001", "此评论不存在"),
PARENT_COMMENT_NOT_FOUND("COMMENT-20000", "此父评论不存在"),
COMMENT_ALREADY_LIKED("COMMENT-20002", "您已经点赞过该评论"),
COMMENT_NOT_LIKED("COMMENT-20003", "您未点赞该评论,无法取消点赞"),
COMMENT_CANT_OPERATE("COMMENT-20004", "您无法操作该评论"),
; ;
// 异常码 // 异常码

View File

@@ -19,4 +19,9 @@ public class CommentHeatBO {
* 热度值 * 热度值
*/ */
private Double heat; private Double heat;
/**
* 笔记 ID
*/
private Long noteId;
} }

View File

@@ -0,0 +1,26 @@
package com.hanserwei.hannote.comment.biz.model.dto;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;
import java.time.LocalDateTime;
@Data
@AllArgsConstructor
@NoArgsConstructor
@Builder
public class LikeUnlikeCommentMqDTO {
private Long userId;
private Long commentId;
/**
* 0: 取消点赞, 1点赞
*/
private Integer type;
private LocalDateTime createTime;
}

View File

@@ -0,0 +1,18 @@
package com.hanserwei.hannote.comment.biz.model.vo;
import jakarta.validation.constraints.NotNull;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;
@Data
@AllArgsConstructor
@NoArgsConstructor
@Builder
public class DeleteCommentReqVO {
@NotNull(message = "评论 ID 不能为空")
private Long commentId;
}

View File

@@ -0,0 +1,63 @@
package com.hanserwei.hannote.comment.biz.model.vo;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;
@Data
@AllArgsConstructor
@NoArgsConstructor
@Builder
public class FindChildCommentItemRspVO {
/**
* 评论 ID
*/
private Long commentId;
/**
* 发布者用户 ID
*/
private Long userId;
/**
* 头像
*/
private String avatar;
/**
* 昵称
*/
private String nickname;
/**
* 评论内容
*/
private String content;
/**
* 评论内容
*/
private String imageUrl;
/**
* 发布时间
*/
private String createTime;
/**
* 被点赞数
*/
private Long likeTotal;
/**
* 回复的用户昵称
*/
private String replyUserName;
/**
* 回复的用户 ID
*/
private Long replyUserId;
}

View File

@@ -0,0 +1,20 @@
package com.hanserwei.hannote.comment.biz.model.vo;
import jakarta.validation.constraints.NotNull;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;
@Data
@AllArgsConstructor
@NoArgsConstructor
@Builder
public class FindChildCommentPageListReqVO {
@NotNull(message = "父评论 ID 不能为空")
private Long parentCommentId;
@NotNull(message = "页码不能为空")
private Integer pageNo = 1;
}

View File

@@ -0,0 +1,69 @@
package com.hanserwei.hannote.comment.biz.model.vo;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;
@Data
@AllArgsConstructor
@NoArgsConstructor
@Builder
public class FindCommentItemRspVO {
/**
* 评论 ID
*/
private Long commentId;
/**
* 发布者用户 ID
*/
private Long userId;
/**
* 头像
*/
private String avatar;
/**
* 昵称
*/
private String nickname;
/**
* 评论内容
*/
private String content;
/**
* 评论内容
*/
private String imageUrl;
/**
* 发布时间
*/
private String createTime;
/**
* 被点赞数
*/
private Long likeTotal;
/**
* 二级评论总数
*/
private Long childCommentTotal;
/**
* 最早回复的评论
*/
private FindCommentItemRspVO firstReplyComment;
/**
* 热度值
*/
private Double heat;
}

View File

@@ -0,0 +1,20 @@
package com.hanserwei.hannote.comment.biz.model.vo;
import jakarta.validation.constraints.NotNull;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;
@Data
@AllArgsConstructor
@NoArgsConstructor
@Builder
public class FindCommentPageListReqVO {
@NotNull(message = "笔记 ID 不能为空")
private Long noteId;
@NotNull(message = "页码不能为空")
private Integer pageNo = 1;
}

View File

@@ -0,0 +1,18 @@
package com.hanserwei.hannote.comment.biz.model.vo;
import jakarta.validation.constraints.NotNull;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;
@Data
@AllArgsConstructor
@NoArgsConstructor
@Builder
public class LikeCommentReqVO {
@NotNull(message = "评论 ID 不能为空")
private Long commentId;
}

View File

@@ -0,0 +1,18 @@
package com.hanserwei.hannote.comment.biz.model.vo;
import jakarta.validation.constraints.NotNull;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;
@Data
@AllArgsConstructor
@NoArgsConstructor
@Builder
public class UnLikeCommentReqVO {
@NotNull(message = "评论 ID 不能为空")
private Long commentId;
}

View File

@@ -1,16 +1,19 @@
package com.hanserwei.hannote.comment.biz.rpc; package com.hanserwei.hannote.comment.biz.rpc;
import cn.hutool.core.collection.CollUtil;
import com.google.common.collect.Lists; import com.google.common.collect.Lists;
import com.hanserwei.framework.common.constant.DateConstants; import com.hanserwei.framework.common.constant.DateConstants;
import com.hanserwei.framework.common.response.Response; import com.hanserwei.framework.common.response.Response;
import com.hanserwei.hannote.comment.biz.model.bo.CommentBO; import com.hanserwei.hannote.comment.biz.model.bo.CommentBO;
import com.hanserwei.hannote.kv.api.KeyValueFeignApi; import com.hanserwei.hannote.kv.api.KeyValueFeignApi;
import com.hanserwei.hannote.kv.dto.req.BatchAddCommentContentReqDTO; import com.hanserwei.hannote.kv.dto.req.*;
import com.hanserwei.hannote.kv.dto.req.CommentContentReqDTO; import com.hanserwei.hannote.kv.dto.resp.FindCommentContentRspDTO;
import jakarta.annotation.Resource; import jakarta.annotation.Resource;
import org.springframework.stereotype.Component; import org.springframework.stereotype.Component;
import java.time.LocalDateTime;
import java.util.List; import java.util.List;
import java.util.Objects;
@Component @Component
public class KeyValueRpcService { public class KeyValueRpcService {
@@ -54,4 +57,51 @@ public class KeyValueRpcService {
return true; return true;
} }
/**
* 批量查询评论内容
*
* @param noteId 笔记ID
* @param findCommentContentReqDTOS 查询参数
* @return 批量查询结果
*/
public List<FindCommentContentRspDTO> batchFindCommentContent(Long noteId, List<FindCommentContentReqDTO> findCommentContentReqDTOS) {
BatchFindCommentContentReqDTO bathFindCommentContentReqDTO = BatchFindCommentContentReqDTO.builder()
.noteId(noteId)
.commentContentKeys(findCommentContentReqDTOS)
.build();
Response<List<FindCommentContentRspDTO>> response = keyValueFeignApi.batchFindCommentContent(bathFindCommentContentReqDTO);
if (!response.isSuccess() || Objects.isNull(response.getData()) || CollUtil.isEmpty(response.getData())) {
return null;
}
return response.getData();
}
/**
* 删除评论内容
*
* @param noteId 笔记ID
* @param createTime 创建时间
* @param contentId 评论内容ID
* @return 是否成功
*/
public boolean deleteCommentContent(Long noteId, LocalDateTime createTime, String contentId) {
DeleteCommentContentReqDTO deleteCommentContentReqDTO = DeleteCommentContentReqDTO.builder()
.noteId(noteId)
.yearMonth(DateConstants.DATE_FORMAT_Y_M.format(createTime))
.contentId(contentId)
.build();
// 调用 KV 存储服务
Response<?> response = keyValueFeignApi.deleteCommentContent(deleteCommentContentReqDTO);
if (!response.isSuccess()) {
throw new RuntimeException("删除评论内容失败");
}
return true;
}
} }

View File

@@ -0,0 +1,45 @@
package com.hanserwei.hannote.comment.biz.rpc;
import cn.hutool.core.collection.CollUtil;
import com.hanserwei.framework.common.response.Response;
import com.hanserwei.hannote.user.api.UserFeignApi;
import com.hanserwei.hannote.user.dto.req.FindUsersByIdsReqDTO;
import com.hanserwei.hannote.user.dto.resp.FindUserByIdRspDTO;
import jakarta.annotation.Resource;
import org.springframework.stereotype.Component;
import java.util.List;
import java.util.Objects;
import java.util.stream.Collectors;
@Component
public class UserRpcService {
@Resource
private UserFeignApi userFeignApi;
/**
* 批量查询用户信息
*
* @param userIds 用户 ID集合
* @return 用户信息集合
*/
public List<FindUserByIdRspDTO> findByIds(List<Long> userIds) {
if (CollUtil.isEmpty(userIds)) {
return null;
}
FindUsersByIdsReqDTO findUsersByIdsReqDTO = new FindUsersByIdsReqDTO();
// 去重, 并设置用户 ID 集合
findUsersByIdsReqDTO.setIds(userIds.stream().distinct().collect(Collectors.toList()));
Response<List<FindUserByIdRspDTO>> response = userFeignApi.findByIds(findUsersByIdsReqDTO);
if (!response.isSuccess() || Objects.isNull(response.getData()) || CollUtil.isEmpty(response.getData())) {
return null;
}
return response.getData();
}
}

View File

@@ -1,9 +1,10 @@
package com.hanserwei.hannote.comment.biz.service; package com.hanserwei.hannote.comment.biz.service;
import com.baomidou.mybatisplus.extension.service.IService; import com.baomidou.mybatisplus.extension.service.IService;
import com.hanserwei.framework.common.response.PageResponse;
import com.hanserwei.framework.common.response.Response; import com.hanserwei.framework.common.response.Response;
import com.hanserwei.hannote.comment.biz.domain.dataobject.CommentDO; import com.hanserwei.hannote.comment.biz.domain.dataobject.CommentDO;
import com.hanserwei.hannote.comment.biz.model.vo.PublishCommentReqVO; import com.hanserwei.hannote.comment.biz.model.vo.*;
public interface CommentService extends IService<CommentDO> { public interface CommentService extends IService<CommentDO> {
/** /**
@@ -13,4 +14,51 @@ public interface CommentService extends IService<CommentDO> {
* @return 响应 * @return 响应
*/ */
Response<?> publishComment(PublishCommentReqVO publishCommentReqVO); Response<?> publishComment(PublishCommentReqVO publishCommentReqVO);
/**
* 评论列表分页查询
*
* @param findCommentPageListReqVO 评论列表分页查询参数
* @return 响应
*/
PageResponse<FindCommentItemRspVO> findCommentPageList(FindCommentPageListReqVO findCommentPageListReqVO);
/**
* 二级评论分页查询
*
* @param findChildCommentPageListReqVO 二级评论分页查询参数
* @return 响应
*/
PageResponse<FindChildCommentItemRspVO> findChildCommentPageList(FindChildCommentPageListReqVO findChildCommentPageListReqVO);
/**
* 评论点赞
*
* @param likeCommentReqVO 评论点赞请求
* @return 响应
*/
Response<?> likeComment(LikeCommentReqVO likeCommentReqVO);
/**
* 取消评论点赞
*
* @param unLikeCommentReqVO 取消评论点赞请求
* @return 响应
*/
Response<?> unlikeComment(UnLikeCommentReqVO unLikeCommentReqVO);
/**
* 删除评论
*
* @param deleteCommentReqVO 删除评论请求
* @return 响应
*/
Response<?> deleteComment(DeleteCommentReqVO deleteCommentReqVO);
/**
* 删除本地评论缓存
*
* @param commentId 评论ID
*/
void deleteCommentLocalCache(Long commentId);
} }

View File

@@ -0,0 +1,33 @@
-- 操作的 Key
local zsetKey = KEYS[1]
-- 获取传入的成员和分数列表
local membersScores = ARGV
-- ZSet 最多缓存 500 条评论
local sizeLimit = 500
-- 检查 ZSet 是否存在
if redis.call('EXISTS', zsetKey) == 0 then
return -1 -- 若不存在,直接 return
end
-- 获取当前 ZSet 的大小
local currentSize = redis.call('ZCARD', zsetKey)
-- 遍历传入的成员和分数,添加到 ZSet 中
for i = 1, #membersScores, 2 do
-- 评论 ID
local member = membersScores[i]
-- 热度值
local score = membersScores[i + 1]
-- 检查当前 ZSet 的大小是否小于 500 条
if currentSize < sizeLimit then
-- 若是,则添加缓存
redis.call('ZADD', zsetKey, score, member)
currentSize = currentSize + 1 -- 更新 ZSet 大小
else
break -- 否则,则达到最大限制,停止添加
end
end
return 0

View File

@@ -0,0 +1,10 @@
-- 操作的 Key
local key = KEYS[1]
local commentId = ARGV[1] -- 评论ID
local expireSeconds = ARGV[2] -- 过期时间(秒)
redis.call("BF.ADD", key, commentId)
-- 设置过期时间
redis.call("EXPIRE", key, expireSeconds)
return 0

View File

@@ -0,0 +1,12 @@
-- 操作的 Key
local key = KEYS[1]
for i = 1, #ARGV - 1 do
redis.call("BF.ADD", key, ARGV[i])
end
---- 最后一个参数为过期时间
local expireTime = ARGV[#ARGV]
-- 设置过期时间
redis.call("EXPIRE", key, expireTime)
return 0

View File

@@ -0,0 +1,20 @@
-- LUA 脚本:评论点赞布隆过滤器
local key = KEYS[1] -- 操作的 Redis Key
local commentId = ARGV[1] -- 笔记ID
-- 使用 EXISTS 命令检查布隆过滤器是否存在
local exists = redis.call('EXISTS', key)
if exists == 0 then
return -1
end
-- 校验该评论是否被点赞过(1 表示已经点赞0 表示未点赞)
local isLiked = redis.call('BF.EXISTS', key, commentId)
if isLiked == 1 then
return 1
end
-- 未被点赞,添加点赞数据
redis.call('BF.ADD', key, commentId)
return 0

View File

@@ -0,0 +1,11 @@
local key = KEYS[1] -- 操作的 Redis Key
local commentId = ARGV[1] -- 评论ID
-- 使用 EXISTS 命令检查布隆过滤器是否存在
local exists = redis.call('EXISTS', key)
if exists == 0 then
return -1
end
-- 校验该评论是否被点赞过(1 表示已经点赞0 表示未点赞)
return redis.call('BF.EXISTS', key, commentId)

View File

@@ -0,0 +1,40 @@
-- 入参说明:
-- KEYS[1]: ZSet 的键
-- ARGV: 每个评论的数据,格式为 member1, score1, member2, score2 ...
local zsetKey = KEYS[1]
local maxSize = 500 -- 最多缓存 500 条热点评论
local batchSize = #ARGV / 2 -- 有多少条评论
-- 确认 ZSet 是否存在
if redis.call("EXISTS", zsetKey) == 0 then
return -1 -- 如果 ZSet 不存在,直接返回
end
for i = 1, batchSize do
local member = ARGV[(i - 1) * 2 + 1] -- 获取当前评论 ID
local score = ARGV[(i - 1) * 2 + 2] -- 获取当前评论的热度
-- 获取 ZSet 的大小
local currentSize = redis.call("ZCARD", zsetKey)
if currentSize < maxSize then
-- 如果 ZSet 的大小小于 maxSize直接添加
redis.call("ZADD", zsetKey, score, member)
else
-- 若已缓存 500 条热点评论
-- 获取当前 ZSet 中热度值最小的评论
local minEntry = redis.call("ZRANGE", zsetKey, 0, 0, "WITHSCORES")
-- 热度最小评论的值
local minScore = minEntry[2]
if score > minScore then
-- 如果当前评论的热度大于最小热度,替换掉最小的;否则无视
redis.call("ZREM", zsetKey, minEntry[1])
redis.call("ZADD", zsetKey, score, member)
end
end
end
return 0

View File

@@ -20,6 +20,7 @@
<result column="create_time" jdbcType="TIMESTAMP" property="createTime" /> <result column="create_time" jdbcType="TIMESTAMP" property="createTime" />
<result column="update_time" jdbcType="TIMESTAMP" property="updateTime" /> <result column="update_time" jdbcType="TIMESTAMP" property="updateTime" />
<result column="child_comment_total" jdbcType="BIGINT" property="childCommentTotal"/> <result column="child_comment_total" jdbcType="BIGINT" property="childCommentTotal"/>
<result column="heat" jdbcType="DOUBLE" property="heat"/>
<result column="first_reply_comment_id" jdbcType="BIGINT" property="firstReplyCommentId"/> <result column="first_reply_comment_id" jdbcType="BIGINT" property="firstReplyCommentId"/>
</resultMap> </resultMap>
<sql id="Base_Column_List"> <sql id="Base_Column_List">
@@ -40,17 +41,25 @@
create_time, create_time,
update_time, update_time,
child_comment_total, child_comment_total,
heat,
first_reply_comment_id first_reply_comment_id
</sql> </sql>
<select id="selectByCommentIds" resultMap="BaseResultMap" parameterType="list"> <select id="selectByCommentIds" resultMap="BaseResultMap" parameterType="list">
select id, select id,
user_id,
content_uuid,
is_content_empty,
image_url,
like_total,
is_top,
create_time,
first_reply_comment_id,
child_comment_total,
level, level,
parent_id, parent_id,
user_id, heat,
child_comment_total, note_id
like_total,
first_reply_comment_id
from t_comment from t_comment
where id in where id in
<foreach collection="commentIds" open="(" separator="," close=")" item="commentId"> <foreach collection="commentIds" open="(" separator="," close=")" item="commentId">
@@ -58,6 +67,7 @@
</foreach> </foreach>
</select> </select>
<insert id="batchInsert" parameterType="list"> <insert id="batchInsert" parameterType="list">
insert IGNORE into t_comment (id, note_id, user_id, insert IGNORE into t_comment (id, note_id, user_id,
content_uuid, is_content_empty, image_url, content_uuid, is_content_empty, image_url,
@@ -81,12 +91,12 @@
</foreach> </foreach>
ELSE heat END ELSE heat END
WHERE id IN WHERE id IN
<foreach collection="commentIds" item="commentId" open="(" close=")" separator=","> <foreach close=")" collection="commentIds" item="commentId" open="(" separator=",">
#{commentId} #{commentId}
</foreach> </foreach>
</update> </update>
<select id="selectEarliestByParentId" resultMap="BaseResultMap" parameterType="map"> <select id="selectEarliestByParentId" parameterType="map" resultMap="BaseResultMap">
select id select id
from t_comment from t_comment
where parent_id = #{parentId} where parent_id = #{parentId}
@@ -100,4 +110,117 @@
set first_reply_comment_id = #{firstReplyCommentId} set first_reply_comment_id = #{firstReplyCommentId}
where id = #{id} where id = #{id}
</update> </update>
<select id="selectPageList" resultMap="BaseResultMap" parameterType="map">
select id,
user_id,
content_uuid,
is_content_empty,
image_url,
like_total,
is_top,
create_time,
first_reply_comment_id,
child_comment_total,
heat
from t_comment
where note_id = #{noteId}
and level = 1
order by heat desc
limit #{offset}, #{pageSize}
</select>
<select id="selectTwoLevelCommentByIds" resultMap="BaseResultMap" parameterType="list">
select id,
user_id,
content_uuid,
is_content_empty,
image_url,
like_total,
create_time,
heat
from t_comment
where id in
<foreach collection="commentIds" open="(" separator="," close=")" item="commentId">
#{commentId}
</foreach>
</select>
<select id="selectHeatComments" resultMap="BaseResultMap">
select id, heat
from t_comment
where note_id = #{noteId}
and level = 1
order by heat desc
limit 500
</select>
<select id="selectChildCommentTotalById" resultType="long">
select child_comment_total
from t_comment
where id = #{commentId}
and level = 1
</select>
<select id="selectChildPageList" resultMap="BaseResultMap" parameterType="map">
select id,
user_id,
note_id,
content_uuid,
is_content_empty,
image_url,
like_total,
create_time,
reply_user_id,
parent_id,
reply_comment_id
from t_comment
where parent_id = #{parentId}
and level = 2
order by id
limit #{offset}, #{pageSize}
</select>
<select id="selectCommentCountByIds" resultMap="BaseResultMap" parameterType="list">
select id,
child_comment_total,
like_total,
level
from t_comment
where id in
<foreach collection="commentIds" open="(" separator="," close=")" item="commentId">
#{commentId}
</foreach>
</select>
<select id="selectChildCommentsByParentIdAndLimit" resultMap="BaseResultMap" parameterType="map">
select id, create_time
from t_comment
where parent_id = #{parentId}
and level = 2
order by create_time
limit #{limit}
</select>
<delete id="deleteByParentId" parameterType="long">
delete
from t_comment
where parent_id = #{commentId}
</delete>
<delete id="deleteByIds" parameterType="map">
delete
from t_comment
where id in
<foreach collection="commentIds" item="commentId" open="(" separator="," close=")">
#{commentId}
</foreach>
</delete>
<select id="selectByReplyCommentId" resultMap="BaseResultMap" parameterType="long">
select
<include refid="Base_Column_List"/>
from t_comment
where reply_comment_id = #{commentId}
</select>
</mapper> </mapper>

View File

@@ -13,4 +13,36 @@
<!--@mbg.generated--> <!--@mbg.generated-->
id, user_id, comment_id, create_time id, user_id, comment_id, create_time
</sql> </sql>
<select id="selectCountByUserIdAndCommentId" resultType="int" parameterType="map">
select count(1)
from t_comment_like
where user_id = #{userId}
and comment_id = #{commentId}
limit 1
</select>
<select id="selectByUserId" resultMap="BaseResultMap" parameterType="map">
select comment_id
from t_comment_like
where user_id = #{userId}
</select>
<delete id="batchDelete" parameterType="map">
DELETE
FROM t_comment_like
WHERE (comment_id, user_id) IN
<foreach collection="unlikes" item="unlike" open="(" separator="," close=")">
(#{unlike.commentId}, #{unlike.userId})
</foreach>
</delete>
<insert id="batchInsert" parameterType="list">
INSERT INTO t_comment_like (comment_id, user_id, create_time)
VALUES
<foreach collection="likes" item="like" separator=",">
(#{like.commentId}, #{like.userId}, #{like.createTime})
</foreach>
ON DUPLICATE KEY UPDATE id=id
</insert>
</mapper> </mapper>

View File

@@ -0,0 +1,29 @@
<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="com.hanserwei.hannote.comment.biz.domain.mapper.NoteCountDOMapper">
<resultMap id="BaseResultMap" type="com.hanserwei.hannote.comment.biz.domain.dataobject.NoteCountDO">
<!--@mbg.generated-->
<!--@Table t_note_count-->
<id column="id" jdbcType="BIGINT" property="id"/>
<result column="note_id" jdbcType="BIGINT" property="noteId"/>
<result column="like_total" jdbcType="BIGINT" property="likeTotal"/>
<result column="collect_total" jdbcType="BIGINT" property="collectTotal"/>
<result column="comment_total" jdbcType="BIGINT" property="commentTotal"/>
</resultMap>
<sql id="Base_Column_List">
<!--@mbg.generated-->
id, note_id, like_total, collect_total, comment_total
</sql>
<select id="selectCommentTotalByNoteId" resultType="long">
select comment_total
from t_note_count
where note_id = #{noteId}
</select>
<update id="updateCommentTotalByNoteId" parameterType="map">
update t_note_count
set comment_total = comment_total + #{count}
where note_id = #{noteId}
</update>
</mapper>

View File

@@ -57,6 +57,16 @@ public interface MQConstants {
*/ */
String TOPIC_NOTE_OPERATE = "NoteOperateTopic"; String TOPIC_NOTE_OPERATE = "NoteOperateTopic";
/**
* Topic: 评论点赞数更新
*/
String TOPIC_COMMENT_LIKE_OR_UNLIKE = "CommentLikeUnlikeTopic";
/**
* Topic: 计数 - 评论点赞数落库
*/
String TOPIC_COUNT_COMMENT_LIKE_2_DB = "CountCommentLike2DBTTopic";
/** /**
* Tag 标签:笔记发布 * Tag 标签:笔记发布
*/ */

View File

@@ -31,11 +31,35 @@ public class RedisKeyConstants {
*/ */
private static final String COUNT_NOTE_KEY_PREFIX = "count:note:"; private static final String COUNT_NOTE_KEY_PREFIX = "count:note:";
/**
* Hash Field: 笔记评论总数
*/
public static final String FIELD_COMMENT_TOTAL = "commentTotal";
/** /**
* Hash Field: 笔记收藏总数 * Hash Field: 笔记收藏总数
*/ */
public static final String FIELD_COLLECT_TOTAL = "collectTotal"; public static final String FIELD_COLLECT_TOTAL = "collectTotal";
/**
* Hash Field: 子评论总数
*/
public static final String FIELD_CHILD_COMMENT_TOTAL = "childCommentTotal";
/**
* 评论维度计数 Key 前缀
*/
private static final String COUNT_COMMENT_KEY_PREFIX = "count:comment:";
/**
* 构建评论维度计数 Key
*
* @param commentId 评论ID
* @return 评论维度计数 Key
*/
public static String buildCountCommentKey(Long commentId) {
return COUNT_COMMENT_KEY_PREFIX + commentId;
}
/** /**
* 构建用户维度计数 Key * 构建用户维度计数 Key
* *

View File

@@ -0,0 +1,55 @@
package com.hanserwei.hannote.count.biz.consumer;
import cn.hutool.core.collection.CollUtil;
import com.google.common.util.concurrent.RateLimiter;
import com.hanserwei.framework.common.utils.JsonUtils;
import com.hanserwei.hannote.count.biz.constant.MQConstants;
import com.hanserwei.hannote.count.biz.domain.mapper.CommentDOMapper;
import com.hanserwei.hannote.count.biz.model.dto.AggregationCountLikeUnlikeCommentMqDTO;
import jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import org.apache.rocketmq.spring.annotation.RocketMQMessageListener;
import org.apache.rocketmq.spring.core.RocketMQListener;
import org.springframework.stereotype.Component;
import java.util.List;
@SuppressWarnings("ALL")
@Component
@RocketMQMessageListener(consumerGroup = "han_note_group_" + MQConstants.TOPIC_COUNT_COMMENT_LIKE_2_DB, // Group 组
topic = MQConstants.TOPIC_COUNT_COMMENT_LIKE_2_DB // 主题 Topic
)
@Slf4j
public class CountCommentLike2DBConsumer implements RocketMQListener<String> {
// 每秒创建 5000 个令牌
private final RateLimiter rateLimiter = RateLimiter.create(5000);
@Resource
private CommentDOMapper commentDOMapper;
@Override
public void onMessage(String body) {
// 流量削峰:通过获取令牌,如果没有令牌可用,将阻塞,直到获得
rateLimiter.acquire();
log.info("## 消费到了 MQ 【计数: 评论点赞数入库】, {}...", body);
List<AggregationCountLikeUnlikeCommentMqDTO> countList = null;
try {
countList = JsonUtils.parseList(body, AggregationCountLikeUnlikeCommentMqDTO.class);
} catch (Exception e) {
log.error("## 解析 JSON 字符串异常", e);
}
if (CollUtil.isNotEmpty(countList)) {
// 更新评论点赞数
countList.forEach(item -> {
Long commentId = item.getCommentId();
Integer count = item.getCount();
commentDOMapper.updateLikeTotalByCommentId(count, commentId);
});
}
}
}

View File

@@ -0,0 +1,138 @@
package com.hanserwei.hannote.count.biz.consumer;
import com.github.phantomthief.collection.BufferTrigger;
import com.google.common.collect.Lists;
import com.hanserwei.framework.common.utils.JsonUtils;
import com.hanserwei.hannote.count.biz.constant.MQConstants;
import com.hanserwei.hannote.count.biz.constant.RedisKeyConstants;
import com.hanserwei.hannote.count.biz.enums.LikeUnlikeCommentTypeEnum;
import com.hanserwei.hannote.count.biz.model.dto.AggregationCountLikeUnlikeCommentMqDTO;
import com.hanserwei.hannote.count.biz.model.dto.CountLikeUnlikeCommentMqDTO;
import jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import org.apache.rocketmq.client.producer.SendCallback;
import org.apache.rocketmq.client.producer.SendResult;
import org.apache.rocketmq.spring.annotation.RocketMQMessageListener;
import org.apache.rocketmq.spring.core.RocketMQListener;
import org.apache.rocketmq.spring.core.RocketMQTemplate;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.messaging.Message;
import org.springframework.messaging.support.MessageBuilder;
import org.springframework.stereotype.Component;
import java.time.Duration;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.stream.Collectors;
@Component
@RocketMQMessageListener(consumerGroup = "han_note_group_count_" + MQConstants.TOPIC_COMMENT_LIKE_OR_UNLIKE, // Group 组
topic = MQConstants.TOPIC_COMMENT_LIKE_OR_UNLIKE // 主题 Topic
)
@Slf4j
public class CountCommentLikeConsumer implements RocketMQListener<String> {
@Resource
private RedisTemplate<String, Object> redisTemplate;
@Resource
private RocketMQTemplate rocketMQTemplate;
private final BufferTrigger<String> bufferTrigger = BufferTrigger.<String>batchBlocking()
.bufferSize(50000) // 缓存队列的最大容量
.batchSize(1000) // 一批次最多聚合 1000 条
.linger(Duration.ofSeconds(1)) // 多久聚合一次
.setConsumerEx(this::consumeMessage) // 设置消费者方法
.build();
@Override
public void onMessage(String body) {
// 往 bufferTrigger 中添加元素
bufferTrigger.enqueue(body);
}
private void consumeMessage(List<String> bodys) {
log.info("==> 【评论点赞数】聚合消息, size: {}", bodys.size());
log.info("==> 【评论点赞数】聚合消息, {}", JsonUtils.toJsonString(bodys));
// List<String> 转 List<CountLikeUnlikeCommentMqDTO>
List<CountLikeUnlikeCommentMqDTO> countLikeUnlikeCommentMqDTOS = bodys.stream()
.map(body -> JsonUtils.parseObject(body, CountLikeUnlikeCommentMqDTO.class)).toList();
// 按评论 ID 进行分组
Map<Long, List<CountLikeUnlikeCommentMqDTO>> groupMap = countLikeUnlikeCommentMqDTOS.stream()
.collect(Collectors.groupingBy(CountLikeUnlikeCommentMqDTO::getCommentId));
// 按组汇总数据,统计出最终的计数
// 最终操作的计数对象
List<AggregationCountLikeUnlikeCommentMqDTO> countList = Lists.newArrayList();
for (Map.Entry<Long, List<CountLikeUnlikeCommentMqDTO>> entry : groupMap.entrySet()) {
// 评论 ID
Long commentId = entry.getKey();
List<CountLikeUnlikeCommentMqDTO> list = entry.getValue();
// 最终的计数值,默认为 0
int finalCount = 0;
for (CountLikeUnlikeCommentMqDTO countLikeUnlikeCommentMqDTO : list) {
// 获取操作类型
Integer type = countLikeUnlikeCommentMqDTO.getType();
// 根据操作类型,获取对应枚举
LikeUnlikeCommentTypeEnum likeUnlikeCommentTypeEnum = LikeUnlikeCommentTypeEnum.valueOf(type);
// 若枚举为空,跳到下一次循环
if (Objects.isNull(likeUnlikeCommentTypeEnum)) continue;
switch (likeUnlikeCommentTypeEnum) {
case LIKE -> finalCount += 1; // 如果为点赞操作,点赞数 +1
case UNLIKE -> finalCount -= 1; // 如果为取消点赞操作,点赞数 -1
}
}
// 将分组后统计出的最终计数,存入 countList 中
countList.add(AggregationCountLikeUnlikeCommentMqDTO.builder()
.commentId(commentId)
.count(finalCount)
.build());
}
log.info("## 【评论点赞数】聚合后的计数数据: {}", JsonUtils.toJsonString(countList));
// 更新 Redis
countList.forEach(item -> {
// 评论 ID
Long commentId = item.getCommentId();
// 聚合后的计数
Integer count = item.getCount();
// Redis 中评论计数 Hash Key
String countCommentRedisKey = RedisKeyConstants.buildCountCommentKey(commentId);
// 判断 Redis 中 Hash 是否存在
boolean isCountCommentExisted = redisTemplate.hasKey(countCommentRedisKey);
// 若存在才会更新
// (因为缓存设有过期时间,考虑到过期后,缓存会被删除,这里需要判断一下,存在才会去更新,而初始化工作放在查询计数来做)
if (isCountCommentExisted) {
// 对目标用户 Hash 中的点赞数字段进行计数操作
redisTemplate.opsForHash().increment(countCommentRedisKey, RedisKeyConstants.FIELD_LIKE_TOTAL, count);
}
});
// 发送 MQ, 评论点赞数据落库
Message<String> message = MessageBuilder.withPayload(JsonUtils.toJsonString(countList))
.build();
// 异步发送 MQ 消息
rocketMQTemplate.asyncSend(MQConstants.TOPIC_COUNT_COMMENT_LIKE_2_DB, message, new SendCallback() {
@Override
public void onSuccess(SendResult sendResult) {
log.info("==> 【计数服务评论点赞数写库】MQ 发送成功SendResult: {}", sendResult);
}
@Override
public void onException(Throwable throwable) {
log.error("==> 【计数服务评论点赞数写库】MQ 发送异常: ", throwable);
}
});
}
}

View File

@@ -5,6 +5,7 @@ import com.github.phantomthief.collection.BufferTrigger;
import com.google.common.collect.Lists; import com.google.common.collect.Lists;
import com.hanserwei.framework.common.utils.JsonUtils; import com.hanserwei.framework.common.utils.JsonUtils;
import com.hanserwei.hannote.count.biz.constant.MQConstants; import com.hanserwei.hannote.count.biz.constant.MQConstants;
import com.hanserwei.hannote.count.biz.constant.RedisKeyConstants;
import com.hanserwei.hannote.count.biz.domain.mapper.CommentDOMapper; import com.hanserwei.hannote.count.biz.domain.mapper.CommentDOMapper;
import com.hanserwei.hannote.count.biz.enums.CommentLevelEnum; import com.hanserwei.hannote.count.biz.enums.CommentLevelEnum;
import com.hanserwei.hannote.count.biz.model.dto.CountPublishCommentMqDTO; import com.hanserwei.hannote.count.biz.model.dto.CountPublishCommentMqDTO;
@@ -15,6 +16,7 @@ import org.apache.rocketmq.client.producer.SendResult;
import org.apache.rocketmq.spring.annotation.RocketMQMessageListener; import org.apache.rocketmq.spring.annotation.RocketMQMessageListener;
import org.apache.rocketmq.spring.core.RocketMQListener; import org.apache.rocketmq.spring.core.RocketMQListener;
import org.apache.rocketmq.spring.core.RocketMQTemplate; import org.apache.rocketmq.spring.core.RocketMQTemplate;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.messaging.Message; import org.springframework.messaging.Message;
import org.springframework.messaging.support.MessageBuilder; import org.springframework.messaging.support.MessageBuilder;
import org.springframework.stereotype.Component; import org.springframework.stereotype.Component;
@@ -39,6 +41,9 @@ public class CountNoteChildCommentConsumer implements RocketMQListener<String> {
@Resource @Resource
private CommentDOMapper commentDOMapper; private CommentDOMapper commentDOMapper;
@Resource
private RedisTemplate<String, Object> redisTemplate;
private final BufferTrigger<String> bufferTrigger = BufferTrigger.<String>batchBlocking() private final BufferTrigger<String> bufferTrigger = BufferTrigger.<String>batchBlocking()
.bufferSize(50000) // 缓存队列的最大容量 .bufferSize(50000) // 缓存队列的最大容量
.batchSize(1000) // 一批次最多聚合 1000 条 .batchSize(1000) // 一批次最多聚合 1000 条
@@ -82,6 +87,19 @@ public class CountNoteChildCommentConsumer implements RocketMQListener<String> {
// 评论数 // 评论数
int count = CollUtil.size(entry.getValue()); int count = CollUtil.size(entry.getValue());
// 更新 Redis 缓存中的评论计数数据
// 构建 Key
String commentCountHashKey = RedisKeyConstants.buildCountCommentKey(parentId);
// 判断 Hash 是否存在
boolean hasKey = redisTemplate.hasKey(commentCountHashKey);
// 若 Hash 存在,则更新子评论总数
if (hasKey) {
// 累加
redisTemplate.opsForHash()
.increment(commentCountHashKey, RedisKeyConstants.FIELD_CHILD_COMMENT_TOTAL, count);
}
// 更新一级评论的下级评论总数,进行累加操作 // 更新一级评论的下级评论总数,进行累加操作
commentDOMapper.updateChildCommentTotal(parentId, count); commentDOMapper.updateChildCommentTotal(parentId, count);
} }

View File

@@ -5,12 +5,14 @@ import com.github.phantomthief.collection.BufferTrigger;
import com.google.common.collect.Lists; import com.google.common.collect.Lists;
import com.hanserwei.framework.common.utils.JsonUtils; import com.hanserwei.framework.common.utils.JsonUtils;
import com.hanserwei.hannote.count.biz.constant.MQConstants; import com.hanserwei.hannote.count.biz.constant.MQConstants;
import com.hanserwei.hannote.count.biz.constant.RedisKeyConstants;
import com.hanserwei.hannote.count.biz.domain.mapper.NoteCountDOMapper; import com.hanserwei.hannote.count.biz.domain.mapper.NoteCountDOMapper;
import com.hanserwei.hannote.count.biz.model.dto.CountPublishCommentMqDTO; import com.hanserwei.hannote.count.biz.model.dto.CountPublishCommentMqDTO;
import jakarta.annotation.Resource; import jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.apache.rocketmq.spring.annotation.RocketMQMessageListener; import org.apache.rocketmq.spring.annotation.RocketMQMessageListener;
import org.apache.rocketmq.spring.core.RocketMQListener; import org.apache.rocketmq.spring.core.RocketMQListener;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component; import org.springframework.stereotype.Component;
import java.time.Duration; import java.time.Duration;
@@ -27,6 +29,8 @@ public class CountNoteCommentConsumer implements RocketMQListener<String> {
@Resource @Resource
private NoteCountDOMapper noteCountDOMapper; private NoteCountDOMapper noteCountDOMapper;
@Resource
private RedisTemplate<String, Object> redisTemplate;
private final BufferTrigger<String> bufferTrigger = BufferTrigger.<String>batchBlocking() private final BufferTrigger<String> bufferTrigger = BufferTrigger.<String>batchBlocking()
.bufferSize(50000) // 缓存队列的最大容量 .bufferSize(50000) // 缓存队列的最大容量
@@ -67,6 +71,19 @@ public class CountNoteCommentConsumer implements RocketMQListener<String> {
// 评论数 // 评论数
int count = CollUtil.size(entry.getValue()); int count = CollUtil.size(entry.getValue());
// 更新 Redis 缓存中的笔记评论总数
// 构建 Key
String noteCountHashKey = RedisKeyConstants.buildCountNoteKey(noteId);
// 判断 Hash 是否存在
boolean hasKey = redisTemplate.hasKey(noteCountHashKey);
// 若 Hash 存在
if (hasKey) {
// 累加更新
redisTemplate.opsForHash()
.increment(noteCountHashKey, RedisKeyConstants.FIELD_COMMENT_TOTAL, count);
}
// 若评论数大于零,则执行更新操作:累加评论总数 // 若评论数大于零,则执行更新操作:累加评论总数
if (count > 0) { if (count > 0) {
noteCountDOMapper.insertOrUpdateCommentTotalByNoteId(count, noteId); noteCountDOMapper.insertOrUpdateCommentTotalByNoteId(count, noteId);

View File

@@ -28,7 +28,7 @@ import java.util.stream.Collectors;
@Component @Component
@Slf4j @Slf4j
@RocketMQMessageListener( @RocketMQMessageListener(
consumerGroup = "han_note_group_" + MQConstants.TOPIC_LIKE_OR_UNLIKE, consumerGroup = "han_note_count_group_" + MQConstants.TOPIC_LIKE_OR_UNLIKE,
topic = MQConstants.TOPIC_LIKE_OR_UNLIKE topic = MQConstants.TOPIC_LIKE_OR_UNLIKE
) )
public class CountNoteLikeConsumer implements RocketMQListener<String> { public class CountNoteLikeConsumer implements RocketMQListener<String> {

View File

@@ -16,4 +16,14 @@ public interface CommentDOMapper extends BaseMapper<CommentDO> {
* @return 更新结果 * @return 更新结果
*/ */
int updateChildCommentTotal(@Param("parentId") Long parentId, @Param("count") int count); int updateChildCommentTotal(@Param("parentId") Long parentId, @Param("count") int count);
/**
* 更新评论点赞数
*
* @param count 计数
* @param commentId 评论 ID
* @return 更新结果
*/
int updateLikeTotalByCommentId(@Param("count") Integer count,
@Param("commentId") Long commentId);
} }

View File

@@ -7,7 +7,6 @@ import org.apache.ibatis.annotations.Param;
@Mapper @Mapper
public interface NoteCountDOMapper extends BaseMapper<NoteCountDO> { public interface NoteCountDOMapper extends BaseMapper<NoteCountDO> {
/** /**
* 添加笔记计数记录或更新笔记点赞数 * 添加笔记计数记录或更新笔记点赞数
* *

View File

@@ -0,0 +1,27 @@
package com.hanserwei.hannote.count.biz.enums;
import lombok.AllArgsConstructor;
import lombok.Getter;
import java.util.Objects;
@Getter
@AllArgsConstructor
public enum LikeUnlikeCommentTypeEnum {
// 点赞
LIKE(1),
// 取消点赞
UNLIKE(0),
;
private final Integer code;
public static LikeUnlikeCommentTypeEnum valueOf(Integer code) {
for (LikeUnlikeCommentTypeEnum likeUnlikeCommentTypeEnum : LikeUnlikeCommentTypeEnum.values()) {
if (Objects.equals(code, likeUnlikeCommentTypeEnum.getCode())) {
return likeUnlikeCommentTypeEnum;
}
}
return null;
}
}

View File

@@ -0,0 +1,24 @@
package com.hanserwei.hannote.count.biz.model.dto;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;
@Data
@AllArgsConstructor
@NoArgsConstructor
@Builder
public class AggregationCountLikeUnlikeCommentMqDTO {
/**
* 评论 ID
*/
private Long commentId;
/**
* 聚合后的计数
*/
private Integer count;
}

View File

@@ -0,0 +1,22 @@
package com.hanserwei.hannote.count.biz.model.dto;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;
@Data
@AllArgsConstructor
@NoArgsConstructor
@Builder
public class CountLikeUnlikeCommentMqDTO {
private Long userId;
private Long commentId;
/**
* 0: 取消点赞, 1点赞
*/
private Integer type;
}

View File

@@ -48,4 +48,11 @@
where id = #{parentId} where id = #{parentId}
and level = 1 and level = 1
</update> </update>
<update id="updateLikeTotalByCommentId" parameterType="map">
update t_comment
set like_total = like_total + #{count},
update_time = now()
where id = #{commentId}
</update>
</mapper> </mapper>

View File

@@ -0,0 +1,78 @@
package com.hanserwei.hannote.count.biz;
import com.hanserwei.framework.common.utils.JsonUtils;
import lombok.Builder;
import lombok.Getter;
import lombok.Setter;
import org.apache.rocketmq.spring.core.RocketMQTemplate;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.messaging.Message;
import org.springframework.messaging.support.MessageBuilder;
import java.time.LocalDateTime;
@SpringBootTest
public class TestCommentLikeUnLikeConsumer {
@Autowired
private RocketMQTemplate rocketMQTemplate;
/**
* 测试:模拟发送评论点赞、取消点赞消息
*/
@Test
void testBatchSendLikeUnlikeCommentMQ() {
Long userId = 2001L;
Long commentId = 4001L;
for (long i = 0; i < 32; i++) {
// 构建消息体 DTO
LikeUnlikeCommentMqDTO likeUnlikeCommentMqDTO = LikeUnlikeCommentMqDTO.builder()
.userId(userId)
.commentId(commentId)
.createTime(LocalDateTime.now())
.build();
// 通过冒号连接, 可让 MQ 发送给主题 Topic 时,携带上标签 Tag
String destination = "CommentLikeUnlikeTopic:";
if (i % 2 == 0) { // 偶数
likeUnlikeCommentMqDTO.setType(0); // 取消点赞
destination = destination + "Unlike";
} else { // 奇数
likeUnlikeCommentMqDTO.setType(1); // 点赞
destination = destination + "Like";
}
// MQ 分区键
String hashKey = String.valueOf(userId);
// 构建消息对象,并将 DTO 转成 Json 字符串设置到消息体中
Message<String> message = MessageBuilder.withPayload(JsonUtils.toJsonString(likeUnlikeCommentMqDTO))
.build();
// 同步发送 MQ 消息
rocketMQTemplate.syncSendOrderly(destination, message, hashKey);
}
}
@Getter
@Setter
@Builder
static class LikeUnlikeCommentMqDTO {
private Long userId;
private Long commentId;
/**
* 0: 取消点赞, 1点赞
*/
private Integer type;
private LocalDateTime createTime;
}
}

View File

@@ -2,15 +2,15 @@ package com.hanserwei.hannote.kv.api;
import com.hanserwei.framework.common.response.Response; import com.hanserwei.framework.common.response.Response;
import com.hanserwei.hannote.kv.constant.ApiConstants; import com.hanserwei.hannote.kv.constant.ApiConstants;
import com.hanserwei.hannote.kv.dto.req.AddNoteContentReqDTO; import com.hanserwei.hannote.kv.dto.req.*;
import com.hanserwei.hannote.kv.dto.req.BatchAddCommentContentReqDTO; import com.hanserwei.hannote.kv.dto.resp.FindCommentContentRspDTO;
import com.hanserwei.hannote.kv.dto.req.DeleteNoteContentReqDTO;
import com.hanserwei.hannote.kv.dto.req.FindNoteContentReqDTO;
import com.hanserwei.hannote.kv.dto.resp.FindNoteContentRspDTO; import com.hanserwei.hannote.kv.dto.resp.FindNoteContentRspDTO;
import org.springframework.cloud.openfeign.FeignClient; import org.springframework.cloud.openfeign.FeignClient;
import org.springframework.web.bind.annotation.PostMapping; import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody; import org.springframework.web.bind.annotation.RequestBody;
import java.util.List;
@FeignClient(name = ApiConstants.SERVICE_NAME) @FeignClient(name = ApiConstants.SERVICE_NAME)
public interface KeyValueFeignApi { public interface KeyValueFeignApi {
@@ -28,4 +28,9 @@ public interface KeyValueFeignApi {
@PostMapping(value = PREFIX + "/comment/content/batchAdd") @PostMapping(value = PREFIX + "/comment/content/batchAdd")
Response<?> batchAddCommentContent(@RequestBody BatchAddCommentContentReqDTO batchAddCommentContentReqDTO); Response<?> batchAddCommentContent(@RequestBody BatchAddCommentContentReqDTO batchAddCommentContentReqDTO);
@PostMapping(value = PREFIX + "/comment/content/batchFind")
Response<List<FindCommentContentRspDTO>> batchFindCommentContent(@RequestBody BatchFindCommentContentReqDTO batchFindCommentContentReqDTO);
@PostMapping(value = PREFIX + "/comment/content/delete")
Response<?> deleteCommentContent(@RequestBody DeleteCommentContentReqDTO deleteCommentContentReqDTO);
} }

View File

@@ -0,0 +1,29 @@
package com.hanserwei.hannote.kv.dto.req;
import jakarta.validation.Valid;
import jakarta.validation.constraints.NotEmpty;
import jakarta.validation.constraints.NotNull;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;
import java.util.List;
@Data
@AllArgsConstructor
@NoArgsConstructor
@Builder
public class BatchFindCommentContentReqDTO {
/**
* 笔记 ID
*/
@NotNull(message = "评论 ID 不能为空")
private Long noteId;
@NotEmpty(message = "评论内容 Key 集合")
@Valid // 指定集合中的 DTO 也需要进行参数校验
private List<FindCommentContentReqDTO> commentContentKeys;
}

View File

@@ -0,0 +1,25 @@
package com.hanserwei.hannote.kv.dto.req;
import jakarta.validation.constraints.NotBlank;
import jakarta.validation.constraints.NotNull;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;
@Data
@AllArgsConstructor
@NoArgsConstructor
@Builder
public class DeleteCommentContentReqDTO {
@NotNull(message = "笔记 ID 不能为空")
private Long noteId;
@NotBlank(message = "发布年月不能为空")
private String yearMonth;
@NotBlank(message = "评论正文 ID 不能为空")
private String contentId;
}

View File

@@ -0,0 +1,21 @@
package com.hanserwei.hannote.kv.dto.req;
import jakarta.validation.constraints.NotBlank;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;
@Data
@AllArgsConstructor
@NoArgsConstructor
@Builder
public class FindCommentContentReqDTO {
@NotBlank(message = "发布年月不能为空")
private String yearMonth;
@NotBlank(message = "评论正文 ID 不能为空")
private String contentId;
}

View File

@@ -0,0 +1,24 @@
package com.hanserwei.hannote.kv.dto.resp;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;
@Data
@AllArgsConstructor
@NoArgsConstructor
@Builder
public class FindCommentContentRspDTO {
/**
* 评论内容 UUID
*/
private String contentId;
/**
* 评论内容
*/
private String content;
}

View File

@@ -4,6 +4,8 @@ import com.hanserwei.framework.biz.operationlog.aspect.ApiOperationLog;
import com.hanserwei.framework.common.response.Response; import com.hanserwei.framework.common.response.Response;
import com.hanserwei.hannote.kv.biz.service.CommentContentService; import com.hanserwei.hannote.kv.biz.service.CommentContentService;
import com.hanserwei.hannote.kv.dto.req.BatchAddCommentContentReqDTO; import com.hanserwei.hannote.kv.dto.req.BatchAddCommentContentReqDTO;
import com.hanserwei.hannote.kv.dto.req.BatchFindCommentContentReqDTO;
import com.hanserwei.hannote.kv.dto.req.DeleteCommentContentReqDTO;
import jakarta.annotation.Resource; import jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.springframework.validation.annotation.Validated; import org.springframework.validation.annotation.Validated;
@@ -26,4 +28,16 @@ public class CommentContentController {
return commentContentService.batchAddCommentContent(batchAddCommentContentReqDTO); return commentContentService.batchAddCommentContent(batchAddCommentContentReqDTO);
} }
@PostMapping(value = "/comment/content/batchFind")
@ApiOperationLog(description = "批量查询评论内容")
public Response<?> batchFindCommentContent(@Validated @RequestBody BatchFindCommentContentReqDTO batchFindCommentContentReqDTO) {
return commentContentService.batchFindCommentContent(batchFindCommentContentReqDTO);
}
@PostMapping(value = "/comment/content/delete")
@ApiOperationLog(description = "删除评论内容")
public Response<?> deleteCommentContent(@Validated @RequestBody DeleteCommentContentReqDTO deleteCommentContentReqDTO) {
return commentContentService.deleteCommentContent(deleteCommentContentReqDTO);
}
} }

View File

@@ -0,0 +1,32 @@
package com.hanserwei.hannote.kv.biz.domain.repository;
import com.hanserwei.hannote.kv.biz.domain.dataobject.CommentContentDO;
import com.hanserwei.hannote.kv.biz.domain.dataobject.CommentContentPrimaryKey;
import org.springframework.data.cassandra.repository.CassandraRepository;
import java.util.List;
import java.util.UUID;
public interface CommentContentRepository extends CassandraRepository<CommentContentDO, CommentContentPrimaryKey> {
/**
* 批量查询评论内容
*
* @param noteId 笔记ID
* @param yearMonths 年月
* @param contentIds 评论 ID列表
* @return 评论内容
*/
List<CommentContentDO> findByPrimaryKeyNoteIdAndPrimaryKeyYearMonthInAndPrimaryKeyContentIdIn(
Long noteId, List<String> yearMonths, List<UUID> contentIds
);
/**
* 删除评论正文
*
* @param noteId 笔记ID
* @param yearMonth 年月
* @param contentId 评论 ID
*/
void deleteByPrimaryKeyNoteIdAndPrimaryKeyYearMonthAndPrimaryKeyContentId(Long noteId, String yearMonth, UUID contentId);
}

View File

@@ -2,8 +2,33 @@ package com.hanserwei.hannote.kv.biz.service;
import com.hanserwei.framework.common.response.Response; import com.hanserwei.framework.common.response.Response;
import com.hanserwei.hannote.kv.dto.req.BatchAddCommentContentReqDTO; import com.hanserwei.hannote.kv.dto.req.BatchAddCommentContentReqDTO;
import com.hanserwei.hannote.kv.dto.req.BatchFindCommentContentReqDTO;
import com.hanserwei.hannote.kv.dto.req.DeleteCommentContentReqDTO;
public interface CommentContentService { public interface CommentContentService {
/**
* 批量添加评论内容
*
* @param batchAddCommentContentReqDTO 批量添加评论内容请求参数
* @return 批量添加结果
*/
Response<?> batchAddCommentContent(BatchAddCommentContentReqDTO batchAddCommentContentReqDTO); Response<?> batchAddCommentContent(BatchAddCommentContentReqDTO batchAddCommentContentReqDTO);
/**
* 批量查询评论内容
*
* @param batchFindCommentContentReqDTO 批量查询评论内容请求参数
* @return 批量查询结果
*/
Response<?> batchFindCommentContent(BatchFindCommentContentReqDTO batchFindCommentContentReqDTO);
/**
* 删除评论内容
*
* @param deleteCommentContentReqDTO 删除评论内容请求参数
* @return 删除结果
*/
Response<?> deleteCommentContent(DeleteCommentContentReqDTO deleteCommentContentReqDTO);
} }

View File

@@ -1,12 +1,16 @@
package com.hanserwei.hannote.kv.biz.service.impl; package com.hanserwei.hannote.kv.biz.service.impl;
import cn.hutool.core.collection.CollUtil;
import com.google.common.collect.Lists;
import com.hanserwei.framework.common.response.Response; import com.hanserwei.framework.common.response.Response;
import com.hanserwei.hannote.kv.biz.domain.dataobject.CommentContentDO; import com.hanserwei.hannote.kv.biz.domain.dataobject.CommentContentDO;
import com.hanserwei.hannote.kv.biz.domain.dataobject.CommentContentPrimaryKey; import com.hanserwei.hannote.kv.biz.domain.dataobject.CommentContentPrimaryKey;
import com.hanserwei.hannote.kv.biz.domain.repository.CommentContentRepository;
import com.hanserwei.hannote.kv.biz.service.CommentContentService; import com.hanserwei.hannote.kv.biz.service.CommentContentService;
import com.hanserwei.hannote.kv.dto.req.BatchAddCommentContentReqDTO; import com.hanserwei.hannote.kv.dto.req.*;
import com.hanserwei.hannote.kv.dto.req.CommentContentReqDTO; import com.hanserwei.hannote.kv.dto.resp.FindCommentContentRspDTO;
import jakarta.annotation.Resource; import jakarta.annotation.Resource;
import jakarta.validation.constraints.NotBlank;
import org.springframework.data.cassandra.core.CassandraTemplate; import org.springframework.data.cassandra.core.CassandraTemplate;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
@@ -18,6 +22,8 @@ public class CommentContentServiceImpl implements CommentContentService {
@Resource @Resource
private CassandraTemplate cassandraTemplate; private CassandraTemplate cassandraTemplate;
@Resource
private CommentContentRepository commentContentRepository;
@Override @Override
public Response<?> batchAddCommentContent(BatchAddCommentContentReqDTO batchAddCommentContentReqDTO) { public Response<?> batchAddCommentContent(BatchAddCommentContentReqDTO batchAddCommentContentReqDTO) {
@@ -47,4 +53,51 @@ public class CommentContentServiceImpl implements CommentContentService {
.execute(); .execute();
return Response.success(); return Response.success();
} }
@Override
public Response<?> batchFindCommentContent(BatchFindCommentContentReqDTO batchFindCommentContentReqDTO) {
// 归属的笔记ID
Long noteId = batchFindCommentContentReqDTO.getNoteId();
// 查询评论的发布年月、内容 UUID
List<FindCommentContentReqDTO> commentContentKeys = batchFindCommentContentReqDTO.getCommentContentKeys();
// 过滤出年月
List<@NotBlank(message = "发布年月不能为空") String> yearMonths = commentContentKeys.stream()
.map(FindCommentContentReqDTO::getYearMonth)
.distinct()
.toList();
// 过滤出内容 UUID
List<UUID> contentIds = commentContentKeys.stream()
.map(r -> UUID.fromString(r.getContentId()))
.distinct()
.toList();
// 批量查询 Cassandra
List<CommentContentDO> commentContentDOS = commentContentRepository
.findByPrimaryKeyNoteIdAndPrimaryKeyYearMonthInAndPrimaryKeyContentIdIn(noteId, yearMonths, contentIds);
// DO 转 DTO
List<FindCommentContentRspDTO> findCommentContentRspDTOS = Lists.newArrayList();
if (CollUtil.isNotEmpty(commentContentDOS)) {
findCommentContentRspDTOS = commentContentDOS.stream()
.map(commentContentDO -> FindCommentContentRspDTO.builder()
.contentId(String.valueOf(commentContentDO.getPrimaryKey().getContentId()))
.content(commentContentDO.getContent())
.build())
.toList();
}
return Response.success(findCommentContentRspDTOS);
}
@Override
public Response<?> deleteCommentContent(DeleteCommentContentReqDTO deleteCommentContentReqDTO) {
Long noteId = deleteCommentContentReqDTO.getNoteId();
String yearMonth = deleteCommentContentReqDTO.getYearMonth();
String contentId = deleteCommentContentReqDTO.getContentId();
// 删除评论正文
commentContentRepository.deleteByPrimaryKeyNoteIdAndPrimaryKeyYearMonthAndPrimaryKeyContentId(noteId, yearMonth, UUID.fromString(contentId));
return Response.success();
}
} }

View File

@@ -7,6 +7,11 @@ public class RedisKeyConstants {
*/ */
public static final String NOTE_DETAIL_KEY = "note:detail:"; public static final String NOTE_DETAIL_KEY = "note:detail:";
/**
* Roaring Bitmap用户笔记点赞 前缀
*/
public static final String R_BITMAP_USER_NOTE_LIKE_LIST_KEY = "rbitmap:note:likes:";
/** /**
* 布隆过滤器:用户笔记点赞 * 布隆过滤器:用户笔记点赞
*/ */
@@ -76,4 +81,14 @@ public class RedisKeyConstants {
public static String buildUserNoteCollectZSetKey(Long userId) { public static String buildUserNoteCollectZSetKey(Long userId) {
return USER_NOTE_COLLECT_ZSET_KEY + userId; return USER_NOTE_COLLECT_ZSET_KEY + userId;
} }
/**
* 构建完整的 Roaring Bitmap用户笔记点赞 KEY
*
* @param userId 用户ID
* @return Roaring Bitmap用户笔记点赞 KEY
*/
public static String buildRBitmapUserNoteLikeListKey(Long userId) {
return R_BITMAP_USER_NOTE_LIKE_LIST_KEY + userId;
}
} }

View File

@@ -21,6 +21,7 @@ import com.hanserwei.hannote.note.biz.domain.dataobject.NoteDO;
import com.hanserwei.hannote.note.biz.domain.dataobject.NoteLikeDO; import com.hanserwei.hannote.note.biz.domain.dataobject.NoteLikeDO;
import com.hanserwei.hannote.note.biz.domain.dataobject.TopicDO; import com.hanserwei.hannote.note.biz.domain.dataobject.TopicDO;
import com.hanserwei.hannote.note.biz.domain.mapper.NoteDOMapper; import com.hanserwei.hannote.note.biz.domain.mapper.NoteDOMapper;
import com.hanserwei.hannote.note.biz.domain.mapper.NoteLikeDOMapper;
import com.hanserwei.hannote.note.biz.enums.*; import com.hanserwei.hannote.note.biz.enums.*;
import com.hanserwei.hannote.note.biz.model.dto.CollectUnCollectNoteMqDTO; import com.hanserwei.hannote.note.biz.model.dto.CollectUnCollectNoteMqDTO;
import com.hanserwei.hannote.note.biz.model.dto.LikeUnlikeNoteMqDTO; import com.hanserwei.hannote.note.biz.model.dto.LikeUnlikeNoteMqDTO;
@@ -76,6 +77,8 @@ public class NoteServiceImpl extends ServiceImpl<NoteDOMapper, NoteDO> implement
private RedisTemplate<String, String> redisTemplate; private RedisTemplate<String, String> redisTemplate;
@Resource @Resource
private RocketMQTemplate rocketMQTemplate; private RocketMQTemplate rocketMQTemplate;
@Resource
private NoteLikeDOMapper noteLikeDOMapper;
/** /**
* 笔记详情本地缓存 * 笔记详情本地缓存
@@ -630,14 +633,17 @@ public class NoteServiceImpl extends ServiceImpl<NoteDOMapper, NoteDO> implement
// 2. 判断目标笔记,是否已经点赞过 // 2. 判断目标笔记,是否已经点赞过
Long userId = LoginUserContextHolder.getUserId(); Long userId = LoginUserContextHolder.getUserId();
// 布隆过滤器Key // Roaring Bitmap Key
String bloomUserNoteLikeListKey = RedisKeyConstants.buildBloomUserNoteLikeListKey(userId); String rbitmapUserNoteLikeListKey = RedisKeyConstants.buildRBitmapUserNoteLikeListKey(userId);
DefaultRedisScript<Long> script = new DefaultRedisScript<>(); DefaultRedisScript<Long> script = new DefaultRedisScript<>();
// Lua 脚本路径 // Lua 脚本路径
script.setScriptSource(new ResourceScriptSource(new ClassPathResource("/lua/bloom_note_like_check.lua"))); script.setScriptSource(new ResourceScriptSource(new ClassPathResource("/lua/rbitmap_note_like_check.lua")));
// 返回值类型
script.setResultType(Long.class); script.setResultType(Long.class);
// 执行 Lua 脚本,拿到返回结果 // 执行 Lua 脚本,拿到返回结果
Long result = redisTemplate.execute(script, Collections.singletonList(bloomUserNoteLikeListKey), noteId); Long result = redisTemplate.execute(script, Collections.singletonList(rbitmapUserNoteLikeListKey), noteId);
NoteLikeLuaResultEnum noteLikeLuaResultEnum = NoteLikeLuaResultEnum.valueOf(result); NoteLikeLuaResultEnum noteLikeLuaResultEnum = NoteLikeLuaResultEnum.valueOf(result);
@@ -659,38 +665,25 @@ public class NoteServiceImpl extends ServiceImpl<NoteDOMapper, NoteDO> implement
// 目标笔记已经被点赞 // 目标笔记已经被点赞
if (count > 0) { if (count > 0) {
// 异步初始化布隆过滤器 // 异步初始化 Roaring Bitmap
threadPoolTaskExecutor.submit(() -> batchAddNoteLike2BloomAndExpire(userId, expireSeconds, bloomUserNoteLikeListKey)); threadPoolTaskExecutor.submit(() ->
batchAddNoteLike2RBitmapAndExpire(userId, expireSeconds, rbitmapUserNoteLikeListKey));
throw new ApiException(ResponseCodeEnum.NOTE_ALREADY_LIKED); throw new ApiException(ResponseCodeEnum.NOTE_ALREADY_LIKED);
} }
// 若笔记未被点赞,查询当前用户是否点赞其他用户,有则同步初始化布隆过滤器 // 若目标笔记未被点赞,查询当前用户是否点赞其他笔记,有则同步初始化 Roaring Bitmap
batchAddNoteLike2BloomAndExpire(userId, expireSeconds, bloomUserNoteLikeListKey); batchAddNoteLike2RBitmapAndExpire(userId, expireSeconds, rbitmapUserNoteLikeListKey);
// 若数据库中也没有点赞记录,说明该用户还未点赞过任何笔记 // 添加当前点赞笔记 ID 到 Roaring Bitmap 中
// Lua 脚本路径 // Lua 脚本路径
script.setScriptSource(new ResourceScriptSource(new ClassPathResource("/lua/bloom_add_note_like_and_expire.lua"))); script.setScriptSource(new ResourceScriptSource(new ClassPathResource("/lua/rbitmap_add_note_like_and_expire.lua")));
// 返回值类型 // 返回值类型
script.setResultType(Long.class); script.setResultType(Long.class);
redisTemplate.execute(script, Collections.singletonList(bloomUserNoteLikeListKey), noteId, expireSeconds); redisTemplate.execute(script, Collections.singletonList(rbitmapUserNoteLikeListKey), noteId, expireSeconds);
} }
// 目标笔记已经被点赞 // 目标笔记已经被点赞
case NOTE_LIKED -> { case NOTE_LIKED -> {
// 校验 ZSet 列表中是否包含被点赞的笔记ID throw new ApiException(ResponseCodeEnum.NOTE_ALREADY_LIKED);
Double score = redisTemplate.opsForZSet().score(userNoteLikeZSetKey, noteId);
if (Objects.nonNull(score)) {
throw new ApiException(ResponseCodeEnum.NOTE_ALREADY_LIKED);
}
// 若 Score 为空,则表示 ZSet 点赞列表中不存在,查询数据库校验
long count = noteLikeDOService.count(new LambdaQueryWrapper<>(NoteLikeDO.class)
.eq(NoteLikeDO::getNoteId, noteId)
.eq(NoteLikeDO::getStatus, LikeStatusEnum.LIKE.getCode()));
if (count > 0) {
// 数据库里面有点赞记录,而 Redis 中 ZSet 不存在,需要重新异步初始化 ZSet
asynInitUserNoteLikesZSet(userId, userNoteLikeZSetKey);
throw new ApiException(ResponseCodeEnum.NOTE_ALREADY_LIKED);
}
} }
} }
// 3. 更新用户 ZSET 点赞列表 // 3. 更新用户 ZSET 点赞列表
@@ -768,6 +761,37 @@ public class NoteServiceImpl extends ServiceImpl<NoteDOMapper, NoteDO> implement
return Response.success(); return Response.success();
} }
/**
* 初始化笔记点赞 Roaring Bitmap
*
* @param userId 用户 ID
* @param expireSeconds 过期时间
* @param rbitmapUserNoteLikeListKey RBitmap 列表 Key
*/
private void batchAddNoteLike2RBitmapAndExpire(Long userId, long expireSeconds, String rbitmapUserNoteLikeListKey) {
try {
// 异步全量同步一下,并设置过期时间
List<NoteLikeDO> noteLikeDOS = noteLikeDOMapper.selectList(new LambdaQueryWrapper<>(NoteLikeDO.class)
.eq(NoteLikeDO::getUserId, userId));
if (CollUtil.isNotEmpty(noteLikeDOS)) {
DefaultRedisScript<Long> script = new DefaultRedisScript<>();
// Lua 脚本路径
script.setScriptSource(new ResourceScriptSource(new ClassPathResource("/lua/rbitmap_batch_add_note_like_and_expire.lua")));
// 返回值类型
script.setResultType(Long.class);
// 构建 Lua 参数
List<Object> luaArgs = Lists.newArrayList();
noteLikeDOS.forEach(noteLikeDO -> luaArgs.add(noteLikeDO.getNoteId())); // 将每个点赞的笔记 ID 传入
luaArgs.add(expireSeconds); // 最后一个参数是过期时间(秒)
redisTemplate.execute(script, Collections.singletonList(rbitmapUserNoteLikeListKey), luaArgs.toArray());
}
} catch (Exception e) {
log.error("## 异步初始化【笔记点赞】Roaring Bitmap 异常: ", e);
}
}
@Override @Override
public Response<?> unlikeNote(UnlikeNoteReqVO unlikeNoteReqVO) { public Response<?> unlikeNote(UnlikeNoteReqVO unlikeNoteReqVO) {
// 笔记ID // 笔记ID
@@ -780,37 +804,38 @@ public class NoteServiceImpl extends ServiceImpl<NoteDOMapper, NoteDO> implement
// 当前登录用户ID // 当前登录用户ID
Long userId = LoginUserContextHolder.getUserId(); Long userId = LoginUserContextHolder.getUserId();
// 布隆过滤器Key // Roaring Bitmap Key
String bloomUserNoteLikeListKey = RedisKeyConstants.buildBloomUserNoteLikeListKey(userId); String rbitmapUserNoteLikeListKey = RedisKeyConstants.buildRBitmapUserNoteLikeListKey(userId);
DefaultRedisScript<Long> script = new DefaultRedisScript<>(); DefaultRedisScript<Long> script = new DefaultRedisScript<>();
// Lua 脚本路径 // Lua 脚本路径
script.setScriptSource(new ResourceScriptSource(new ClassPathResource("/lua/bloom_note_unlike_check.lua"))); script.setScriptSource(new ResourceScriptSource(new ClassPathResource("/lua/rbitmap_note_unlike_check.lua")));
// 返回值类型
script.setResultType(Long.class); script.setResultType(Long.class);
// 执行 Lua 脚本,拿到返回结果 // 执行 Lua 脚本,拿到返回结果
Long result = redisTemplate.execute(script, Collections.singletonList(bloomUserNoteLikeListKey), noteId); Long result = redisTemplate.execute(script, Collections.singletonList(rbitmapUserNoteLikeListKey), noteId);
NoteUnlikeLuaResultEnum noteUnlikeLuaResultEnum = NoteUnlikeLuaResultEnum.valueOf(result); NoteUnlikeLuaResultEnum noteUnlikeLuaResultEnum = NoteUnlikeLuaResultEnum.valueOf(result);
log.info("==> 【笔记取消点赞】Lua 脚本返回结果: {}", noteUnlikeLuaResultEnum); log.info("==> 【笔记取消点赞】Lua 脚本返回结果: {}", noteUnlikeLuaResultEnum);
switch (Objects.requireNonNull(noteUnlikeLuaResultEnum)) { switch (Objects.requireNonNull(noteUnlikeLuaResultEnum)) {
// 布隆过滤器不存在 // 布隆过滤器不存在
case NOT_EXIST -> {//笔记不存在 case NOT_EXIST -> {//笔记不存在
//异步初始化布隆过滤器 // 异步初始化 Roaring Bitmap
threadPoolTaskExecutor.submit(() -> { threadPoolTaskExecutor.submit(() -> {
// 保底1天+随机秒数 // 保底1天+随机秒数
long expireSeconds = 60 * 60 * 24 + RandomUtil.randomInt(60 * 60 * 24); long expireSeconds = 60 * 60 * 24 + RandomUtil.randomInt(60 * 60 * 24);
batchAddNoteLike2BloomAndExpire(userId, expireSeconds, bloomUserNoteLikeListKey); batchAddNoteLike2RBitmapAndExpire(userId, expireSeconds, rbitmapUserNoteLikeListKey);
}); });
// 从数据库中校验笔记是否被点赞 // 从数据库中校验笔记是否被点赞
long count = noteLikeDOService.count(new LambdaQueryWrapper<>(NoteLikeDO.class) long count = noteLikeDOMapper.selectCount(new LambdaQueryWrapper<>(NoteLikeDO.class)
.eq(NoteLikeDO::getUserId, userId) .eq(NoteLikeDO::getUserId, userId)
.eq(NoteLikeDO::getNoteId, noteId) .eq(NoteLikeDO::getNoteId, noteId));
.eq(NoteLikeDO::getStatus, LikeStatusEnum.LIKE.getCode()));
if (count == 0) { // 未点赞,无法取消点赞操作,抛出业务异常
log.info("==> 【笔记取消点赞】用户未点赞该笔记"); log.info("1111111");
throw new ApiException(ResponseCodeEnum.NOTE_NOT_LIKED); if (count == 0) throw new ApiException(ResponseCodeEnum.NOTE_NOT_LIKED);
}
} }
// 布隆过滤器校验目标笔记未被点赞(判断绝对正确) // 布隆过滤器校验目标笔记未被点赞(判断绝对正确)
case NOTE_NOT_LIKED -> throw new ApiException(ResponseCodeEnum.NOTE_NOT_LIKED); case NOTE_NOT_LIKED -> throw new ApiException(ResponseCodeEnum.NOTE_NOT_LIKED);
@@ -820,14 +845,9 @@ public class NoteServiceImpl extends ServiceImpl<NoteDOMapper, NoteDO> implement
// 用户点赞列表ZsetKey // 用户点赞列表ZsetKey
String userNoteLikeZSetKey = RedisKeyConstants.buildUserNoteLikeZSetKey(userId); String userNoteLikeZSetKey = RedisKeyConstants.buildUserNoteLikeZSetKey(userId);
// TODO: 后续考虑换掉布隆过滤器。
Long removed = redisTemplate.opsForZSet().remove(userNoteLikeZSetKey, noteId); Long removed = redisTemplate.opsForZSet().remove(userNoteLikeZSetKey, noteId);
if (Objects.nonNull(removed) && removed == 0) {
log.info("==> 【笔记取消点赞】用户未点赞该笔记");
throw new ApiException(ResponseCodeEnum.NOTE_NOT_LIKED);
}
//4. 发送 MQ, 数据更新落库 //4. 发送 MQ, 数据更新落库
// 构建MQ消息体 // 构建MQ消息体

View File

@@ -0,0 +1,10 @@
-- 操作的 Key
local key = KEYS[1]
local noteId = ARGV[1] -- 笔记ID
local expireSeconds = ARGV[2] -- 过期时间(秒)
redis.call("R64.SETBIT", key, noteId, 1)
-- 设置过期时间
redis.call("EXPIRE", key, expireSeconds)
return 0

View File

@@ -0,0 +1,12 @@
-- 操作的 Key
local key = KEYS[1]
for i = 1, #ARGV - 1 do
redis.call("R64.SETBIT", key, ARGV[i], 1)
end
-- 最后一个参数为过期时间
local expireTime = ARGV[#ARGV]
-- 设置过期时间
redis.call("EXPIRE", key, expireTime)
return 0

View File

@@ -0,0 +1,20 @@
-- LUA 脚本:点赞 Roaring Bitmap
local key = KEYS[1] -- 操作的 Redis Key
local noteId = ARGV[1] -- 笔记ID
-- 使用 EXISTS 命令检查 Roaring Bitmap 是否存在
local exists = redis.call('EXISTS', key)
if exists == 0 then
return -1
end
-- 校验该篇笔记是否被点赞过(1 表示已经点赞0 表示未点赞)
local isLiked = redis.call('R64.GETBIT', key, noteId)
if isLiked == 1 then
return 1
end
-- 未被点赞,添加点赞数据
redis.call('R64.SETBIT', key, noteId, 1)
return 0

View File

@@ -0,0 +1,17 @@
local key = KEYS[1] -- 操作的 Redis Key
local noteId = ARGV[1] -- 笔记ID
-- 使用 EXISTS 命令检查 Roaring Bitmap 是否存在
local exists = redis.call('EXISTS', key)
if exists == 0 then
return -1
end
-- 校验该篇笔记是否被点赞过(1 表示已经点赞0 表示未点赞)
local isLiked = redis.call('R64.GETBIT', key, noteId)
if isLiked == 0 then
return 0
end
-- 取消点赞,设置 Value 为 0
return redis.call('R64.SETBIT', key, noteId, 0)

View File

@@ -3,7 +3,7 @@ POST http://localhost:8000/auth/verification/code/send
Content-Type: application/json Content-Type: application/json
{ {
"email": "2628273921@qq.com" "email": "ssw010723@gmail.com"
} }
### 登录/注册 ### 登录/注册
@@ -11,8 +11,8 @@ POST http://localhost:8000/auth/login
Content-Type: application/json Content-Type: application/json
{ {
"email": "2628273921@qq.com", "email": "ssw010723@gmail.com",
"code": "825004", "code": "116253",
"type": 1 "type": 1
} }
@@ -202,16 +202,16 @@ Content-Type: application/json
Authorization: Bearer {{thirdToken}} Authorization: Bearer {{thirdToken}}
{ {
"id": 1981698494959714362 "id": 1985254482941837349
} }
### 笔记取消点赞入口 ### 笔记取消点赞入口
POST http://localhost:8000/note/note/unlike POST http://localhost:8000/note/note/unlike
Content-Type: application/json Content-Type: application/json
Authorization: Bearer {{otherToken}} Authorization: Bearer {{thirdToken}}
{ {
"id": 1977249693272375330 "id": 1985254482941837349
} }
### 笔记收藏入口 ### 笔记收藏入口
@@ -298,7 +298,7 @@ Authorization: Bearer {{token}}
{ {
"noteId": 1862481582414102549, "noteId": 1862481582414102549,
"content": "这是一条测试评论计数的二级评论666", "content": "这是一条测试同步Redis更新计数的评论",
"imageUrl": "https://cdn.pixabay.com/photo/2025/10/05/15/06/autumn-9875155_1280.jpg", "imageUrl": "https://cdn.pixabay.com/photo/2025/10/05/15/06/autumn-9875155_1280.jpg",
"replyCommentId": 4002 "replyCommentId": 4002
} }
@@ -328,4 +328,78 @@ Content-Type: application/json
"content": "这是一条评论内容3" "content": "这是一条评论内容3"
} }
] ]
} }
### 批量查询评论内容
POST http://localhost:8084/kv/comment/content/batchFind
Content-Type: application/json
{
"noteId": 1862481582414102549,
"commentContentKeys": [
{
"yearMonth": "2025-11",
"contentId": "a2537301-dc80-4fd6-ab2f-a9a908baebba"
},
{
"yearMonth": "2025-11",
"contentId": "a66b06ce-bf45-4a21-a5f2-31e5fb5fb5eb"
}
]
}
### 分页查询评论
POST http://localhost:8093/comment/list
Content-Type: application/json
{
"noteId": 1862481582414102549,
"pageNo": 1
}
### 分页查询子评论
POST http://localhost:8000/comment/comment/child/list
Content-Type: application/json
Authorization: Bearer {{token}}
{
"parentCommentId": 4002,
"pageNo": 1
}
### 点赞评论
POST http://localhost:8000/comment/comment/like
Content-Type: application/json
Authorization: Bearer {{token}}
{
"commentId": 8001
}
### 取消点赞评论
POST http://localhost:8000/comment/comment/unlike
Content-Type: application/json
Authorization: Bearer {{token}}
{
"commentId": 8001
}
### 删除评论
POST http://localhost:8084/kv/comment/content/delete
Content-Type: application/json
{
"noteId": 1862481582414102549,
"yearMonth": "2025-11",
"contentId": "0fa4376f-a098-4fee-821b-f5b7e627a72c"
}
### 删除评论,同步删除一切相关缓存
POST http://localhost:8000/comment/comment/delete
Content-Type: application/json
Authorization: Bearer {{token}}
{
"commentId": 8001
}