✨Feat: 使用Redis优化点赞查询性能

pom.xml

@@ -67,6 +67,27 @@
             <artifactId>mysql-connector-j</artifactId>
             <scope>runtime</scope>
         </dependency>
+
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter-data-redis</artifactId>
+            <exclusions>
+                <exclusion>
+                    <groupId>lettuce-core</groupId>
+                    <artifactId>io.lettuce</artifactId>
+                </exclusion>
+            </exclusions>
+        </dependency>
+
+        <dependency>
+            <groupId>redis.clients</groupId>
+            <artifactId>jedis</artifactId>
+        </dependency>
+
+        <dependency>
+            <groupId>org.springframework.session</groupId>
+            <artifactId>spring-session-data-redis</artifactId>
+        </dependency>
     </dependencies>
 
 

src/main/java/cn/hezhaohui/thumb/config/RedisConfig.java

@@ -0,0 +1,42 @@
+package cn.hezhaohui.thumb.config;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.jsontype.impl.LaissezFaireSubTypeValidator;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.data.redis.connection.RedisConnectionFactory;
+import org.springframework.data.redis.core.RedisTemplate;
+import org.springframework.data.redis.serializer.GenericJackson2JsonRedisSerializer;
+import org.springframework.data.redis.serializer.Jackson2JsonRedisSerializer;
+import org.springframework.data.redis.serializer.RedisSerializer;
+import org.springframework.data.redis.serializer.StringRedisSerializer;
+
+@Configuration
+public class RedisConfig {
+
+    @Bean
+    public RedisTemplate<String, Object> redisTemplate(RedisConnectionFactory connectionFactory) {
+        RedisTemplate<String, Object> template = new RedisTemplate<>();
+        template.setConnectionFactory(connectionFactory);
+
+        // 对象 序列化
+        ObjectMapper objectMapper = new ObjectMapper();
+        objectMapper.activateDefaultTyping(LaissezFaireSubTypeValidator.instance, ObjectMapper.DefaultTyping.NON_FINAL);
+        Jackson2JsonRedisSerializer<Object> serializer = new Jackson2JsonRedisSerializer<>(objectMapper, Object.class);
+
+        // Key
+        template.setKeySerializer(new StringRedisSerializer());
+        template.setValueSerializer(serializer);
+        template.setHashKeySerializer(new StringRedisSerializer());
+        template.setHashValueSerializer(serializer);
+
+        template.afterPropertiesSet();
+        return template;
+
+    }
+
+    @Bean
+    public RedisSerializer<Object> springSessionDefaultRedisSerializer() {
+        return new GenericJackson2JsonRedisSerializer();
+    }
+}

src/main/java/cn/hezhaohui/thumb/constent/ThumbConstant.java

@@ -0,0 +1,15 @@
+package cn.hezhaohui.thumb.constent;
+
+
+/**
+ * 点赞常量接口
+ * 用于定义与点赞功能相关的常量
+ */
+public interface ThumbConstant {
+
+    /**
+     * 用户点赞前缀常量
+     * 用于作为Redis中存储用户点赞信息的key前缀
+     */
+    String USER_THUMB_KEY_PREFIX = "thumb:";
+}

src/main/java/cn/hezhaohui/thumb/service/ThumbService.java

@@ -16,4 +16,6 @@ public interface ThumbService extends IService<Thumb> {
     Boolean doThumb(DoThumbRequest doThumbRequest, HttpServletRequest request);
 
     Boolean undoThumb(DoThumbRequest doThumbRequest, HttpServletRequest request);
+
+    Boolean hasThumb(Long blogId, Long userId);
 }

src/main/java/cn/hezhaohui/thumb/service/impl/BlogServiceImpl.java

@@ -1,6 +1,6 @@
 package cn.hezhaohui.thumb.service.impl;
 
-import cn.hezhaohui.thumb.model.entity.Thumb;
+import cn.hezhaohui.thumb.constent.ThumbConstant;
 import cn.hezhaohui.thumb.model.entity.User;
 import cn.hezhaohui.thumb.model.vo.BlogVO;
 import cn.hezhaohui.thumb.service.ThumbService;
@@ -14,12 +14,12 @@ import cn.hezhaohui.thumb.mapper.BlogMapper;
 import jakarta.annotation.Resource;
 import jakarta.servlet.http.HttpServletRequest;
 import org.springframework.context.annotation.Lazy;
+import org.springframework.data.redis.core.RedisTemplate;
 import org.springframework.stereotype.Service;
 
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
-import java.util.Set;
 import java.util.stream.Collectors;
 
 /**
@@ -38,9 +38,8 @@ public class BlogServiceImpl extends ServiceImpl<BlogMapper, Blog>
     @Lazy
     private ThumbService thumbService;
 
-    public BlogServiceImpl(UserService userService) {
-        this.userService = userService;
-    }
+    @Resource
+    private RedisTemplate<String, Object> redisTemplate;
 
     @Override
     public BlogVO getBlogVOById(long blogId, HttpServletRequest request) {
@@ -55,13 +54,15 @@ public class BlogServiceImpl extends ServiceImpl<BlogMapper, Blog>
         Map<Long, Boolean> blogIdHasThumbMap = new HashMap<>();
         if (ObjUtil.isNotEmpty(loginUser)) {
             // Blog Id
-            Set<Long> blogIdSet = blogList.stream().map(Blog::getId).collect(Collectors.toSet());
+            List<Object> blogIdList = blogList.stream().map(blog -> blog.getId().toString()).collect(Collectors.toList());
             // Get Thumb-up Data
-            List<Thumb> thumbList = thumbService.lambdaQuery()
-                    .eq(Thumb::getUserid, loginUser.getId())
-                    .in(Thumb::getBlogId, blogIdSet)
-                    .list();
-            thumbList.forEach(blogThumb -> blogIdHasThumbMap.put(blogThumb.getBlogId(), true));
+            List<Object> thumbList = redisTemplate.opsForHash().multiGet(ThumbConstant.USER_THUMB_KEY_PREFIX + loginUser.getId(), blogIdList);
+            for (int i = 0; i < thumbList.size(); i++) {
+                if (thumbList.get(i) == null) {
+                    continue;
+                }
+                blogIdHasThumbMap.put(Long.valueOf(blogIdList.get(i).toString()), true);
+            }
         }
         return blogList.stream()
                 .map(blog -> {
@@ -80,11 +81,8 @@ public class BlogServiceImpl extends ServiceImpl<BlogMapper, Blog>
             return blogVO;
         }
 
-        Thumb thumb = thumbService.lambdaQuery()
-                .eq(Thumb::getUserid, loginUser.getId())
-                .eq(Thumb::getBlogId, blog.getId())
-                .one();
-        blogVO.setHasThumb(thumb != null);
+        Boolean exist = thumbService.hasThumb(blog.getId(), loginUser.getId());
+        blogVO.setHasThumb(exist);
 
         return blogVO;
     }

src/main/java/cn/hezhaohui/thumb/service/impl/ThumbServiceImpl.java

@@ -1,5 +1,6 @@
 package cn.hezhaohui.thumb.service.impl;
 
+import cn.hezhaohui.thumb.constent.ThumbConstant;
 import cn.hezhaohui.thumb.exception.BusinessException;
 import cn.hezhaohui.thumb.exception.ErrorCode;
 import cn.hezhaohui.thumb.model.dto.DoThumbRequest;
@@ -13,6 +14,7 @@ import cn.hezhaohui.thumb.service.ThumbService;
 import cn.hezhaohui.thumb.mapper.ThumbMapper;
 import jakarta.annotation.Resource;
 import jakarta.servlet.http.HttpServletRequest;
+import org.springframework.data.redis.core.RedisTemplate;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.support.TransactionTemplate;
 
@@ -34,6 +36,9 @@ public class ThumbServiceImpl extends ServiceImpl<ThumbMapper, Thumb>
     @Resource
     private TransactionTemplate transactionTemplate;
 
+    @Resource
+    private RedisTemplate<String, Object> redisTemplate;
+
     @Override
     public Boolean doThumb(DoThumbRequest doThumbRequest, HttpServletRequest request) {
         if (doThumbRequest == null || doThumbRequest.getBlogId() == null) {
@@ -46,10 +51,7 @@ public class ThumbServiceImpl extends ServiceImpl<ThumbMapper, Thumb>
             return transactionTemplate.execute(status -> {
                 Long blogId = doThumbRequest.getBlogId();
                 // 判断是否已点过赞
-                boolean exists = this.lambdaQuery()
-                        .eq(Thumb::getUserid, loginUser.getId())
-                        .eq(Thumb::getBlogId, blogId)
-                        .exists();
+                Boolean exists = this.hasThumb(blogId, loginUser.getId());
                 if (exists) {
                     throw new BusinessException(ErrorCode.OPERATION_ERROR, "用户已点赞");
                 }
@@ -63,7 +65,12 @@ public class ThumbServiceImpl extends ServiceImpl<ThumbMapper, Thumb>
                 thumb.setUserid(loginUser.getId());
                 thumb.setBlogId(blogId);
                 // 两者一起执行
-                return update && this.save(thumb);
+                boolean success = update && this.save(thumb);
+                // 点赞记录存入 Redis
+                if (success) {
+                    redisTemplate.opsForHash().put(ThumbConstant.USER_THUMB_KEY_PREFIX + loginUser.getId().toString(), blogId.toString(), thumb.getId());
+                }
+                return success;
             });
         }
     }
@@ -80,11 +87,9 @@ public class ThumbServiceImpl extends ServiceImpl<ThumbMapper, Thumb>
             return transactionTemplate.execute(status -> {
                 Long blogId = doThumbRequest.getBlogId();
                 // 判断是否已点过赞
-                Thumb thumb = this.lambdaQuery()
-                        .eq(Thumb::getUserid, loginUser.getId())
-                        .eq(Thumb::getBlogId, blogId)
-                        .one();
-                if (thumb == null) {
+                Object thumbIdObj = redisTemplate.opsForHash().get(ThumbConstant.USER_THUMB_KEY_PREFIX + loginUser.getId().toString(), blogId.toString());
+                Long thumbId = thumbIdObj == null ? null : ((Number) thumbIdObj).longValue();
+                if (thumbId == null) {
                     throw new BusinessException(ErrorCode.OPERATION_ERROR, "用户未点赞");
                 }
                 // 更新帖子点赞计数器
@@ -94,10 +99,19 @@ public class ThumbServiceImpl extends ServiceImpl<ThumbMapper, Thumb>
                         .update();
                 // 更新点赞表数据
                 // 两者一起执行
-                return update && this.removeById(thumb.getId());
+                boolean success = update && this.removeById(thumbId);
+                if (success) {
+                    redisTemplate.opsForHash().delete(ThumbConstant.USER_THUMB_KEY_PREFIX + loginUser.getId(), blogId.toString());
+                }
+                return success;
             });
         }
     }
+
+    @Override
+    public Boolean hasThumb(Long blogId, Long userId) {
+        return redisTemplate.opsForHash().hasKey(ThumbConstant.USER_THUMB_KEY_PREFIX + userId, blogId.toString());
+    }
 }
 
 

src/main/resources/application.yml

@@ -9,6 +9,13 @@ spring:
     url: jdbc:mysql://47.121.181.112:3306/thumb_up
     username: root
     password: dn293830
+  data:
+    redis:
+      database: 0
+      host: 47.121.181.112
+      port: 6379
+      password: dn293830
+      connect-timeout: 5000
 
 
 mybatis-plus: