Redis实现点赞功能模块

以前看了一篇文章，讲redis的应用场景，其中一个应用场景就是实现点赞功能，纸上得来恐觉浅，必须实战一波

功能点设计

好比我喜欢发文章的掘金网站就有点赞的功能，统计文章点赞的总数，用户全部文章的点赞数，所以设计的点赞功能模块具备如下功能点：

• 某篇文章的点赞数
• 用户全部文章的点赞数
• 用户点赞的文章
• 持久化到MySQL数据库

数据库设计

• Redis数据库设计 Redis是K-V数据库，没有统一的数据结构，针对不一样的功能点设计了不一样的K-V存储结构

  • 用户某篇文章的点赞数 使用HashMap数据结构，HashMap中的key为articleId，value为Set，Set中的值为用户ID，即HashMap<String, Set<String>>
  • 用户总的点赞数 使用HashMap数据结构，HashMap中的key为userId，value为String记录总的点赞数
  • 用户点赞的文章 使用HashMap数据结构，HashMap中的key为userId，value为Set，Set中的值为文章ID，即HashMap<String, Set<String>>

• MySQL数据库设计 最主要的两张表，article表和user_like_article

  • article表结构

  字段值	字段类型	说明
  article_name	varchar	文章名字
  content	blob	文章内容
  total_like_count	bigint	文章总点赞数

  文章总的点赞数须要和Redis中的点赞数进行同步

  • user_like_article表结构

  字段值	字段类型	说明
  user_id	bigint	用户ID
  article_id	bigint	文章ID

  记录用户点赞文章的信息，是一张中间表

说明：表结构设计省略了id、deleted、gmt_create、gmt_modified字段

流程图

流程图比较简单，点赞和取消点赞基本实现步骤相同

• 参数校验 对传入的参数进行null值判断
• 逻辑校验 对于用户点赞，用户不能重复点赞相同的文章 对于取消点赞，用户不能取消未点赞的文章
• 存入Redis 存入的数据主要有全部文章的点赞数、某篇文章的点赞数、用户点赞的文章
• 定时任务 经过定时【1小时执行一次】，从Redis读取数据持久化到MySQL中

代码功能实现

• 点赞

public void likeArticle(Long articleId, Long likedUserId, Long likedPostId) {
    validateParam(articleId, likedUserId, likedPostId);  //参数验证

    logger.info("点赞数据存入redis开始，articleId:{}，likedUserId:{}，likedPostId:{}", articleId, likedUserId, likedPostId);
    synchronized (this) {
        //只有未点赞的用户才能够进行点赞
        likeArticleLogicValidate(articleId, likedUserId, likedPostId);
        //1.用户总点赞数+1
        redisTemplate.opsForHash().increment(TOTAL_LIKE_COUNT_KEY, String.valueOf(likedUserId), 1);

        //2.用户喜欢的文章+1
        String userLikeResult = (String) redisTemplate.opsForHash().get(USER_LIKE_ARTICLE_KEY, String.valueOf(likedPostId));
        Set<Long> articleIdSet = userLikeResult == null ? new HashSet<>() : FastjsonUtil.deserializeToSet(userLikeResult, Long.class);
            articleIdSet.add(articleId);
        redisTemplate.opsForHash().put(USER_LIKE_ARTICLE_KEY, String.valueOf(likedPostId), FastjsonUtil.serialize(articleIdSet));

        //3.文章点赞数+1
        String articleLikedResult = (String) redisTemplate.opsForHash().get(ARTICLE_LIKED_USER_KEY, String.valueOf(articleId));
        Set<Long> likePostIdSet = articleLikedResult == null ? new HashSet<>() : FastjsonUtil.deserializeToSet(articleLikedResult, Long.class);
        likePostIdSet.add(likedPostId);
        redisTemplate.opsForHash().put(ARTICLE_LIKED_USER_KEY, String.valueOf(articleId), FastjsonUtil.serialize(likePostIdSet));
        logger.info("取消点赞数据存入redis结束，articleId:{}，likedUserId:{}，likedPostId:{}", articleId, likedUserId, likedPostId);
    }
}
复制代码

• 取消点赞

public void unlikeArticle(Long articleId, Long likedUserId, Long likedPostId) {
    validateParam(articleId, likedUserId, likedPostId);  //参数校验

    logger.info("取消点赞数据存入redis开始，articleId:{}，likedUserId:{}，likedPostId:{}", articleId, likedUserId, likedPostId);
    //1.用户总点赞数-1
    synchronized (this) {
        //只有点赞的用户才能够取消点赞
        unlikeArticleLogicValidate(articleId, likedUserId, likedPostId);
        Long totalLikeCount = Long.parseLong((String)redisTemplate.opsForHash().get(TOTAL_LIKE_COUNT_KEY, String.valueOf(likedUserId)));
         redisTemplate.opsForHash().put(TOTAL_LIKE_COUNT_KEY, String.valueOf(likedUserId), String.valueOf(--totalLikeCount));

        //2.用户喜欢的文章-1
        String userLikeResult = (String) redisTemplate.opsForHash().get(USER_LIKE_ARTICLE_KEY, String.valueOf(likedPostId));
        Set<Long> articleIdSet = FastjsonUtil.deserializeToSet(userLikeResult, Long.class);
        articleIdSet.remove(articleId);
        redisTemplate.opsForHash().put(USER_LIKE_ARTICLE_KEY, String.valueOf(likedPostId), FastjsonUtil.serialize(articleIdSet));

        //3.取消用户某篇文章的点赞数
        String articleLikedResult = (String) redisTemplate.opsForHash().get(ARTICLE_LIKED_USER_KEY, String.valueOf(articleId));
        Set<Long> likePostIdSet = FastjsonUtil.deserializeToSet(articleLikedResult, Long.class);
        likePostIdSet.remove(likedPostId);
        redisTemplate.opsForHash().put(ARTICLE_LIKED_USER_KEY, String.valueOf(articleId), FastjsonUtil.serialize(likePostIdSet));
    }

    logger.info("取消点赞数据存入redis结束，articleId:{}，likedUserId:{}，likedPostId:{}", articleId, likedUserId, likedPostId);
}
复制代码

• 异步落库

@Scheduled(cron = "0 0 0/1 * * ? ")
public void redisDataToMySQL() {
    logger.info("time:{}，开始执行Redis数据持久化到MySQL任务", LocalDateTime.now().format(formatter));
    //1.更新文章总的点赞数
    Map<String, String> articleCountMap = redisTemplate.opsForHash().entries(ARTICLE_LIKED_USER_KEY);
    for (Map.Entry<String, String> entry : articleCountMap.entrySet()) {
        String articleId = entry.getKey();
        Set<Long> userIdSet = FastjsonUtil.deserializeToSet(entry.getValue(), Long.class);
        //1.同步某篇文章总的点赞数到MySQL
        synchronizeTotalLikeCount(articleId, userIdSet);
        //2.同步用户喜欢的文章
        synchronizeUserLikeArticle(articleId, userIdSet);
    }
    logger.info("time:{}，结束执行Redis数据持久化到MySQL任务", LocalDateTime.now().format(formatter));
}
复制代码

说明：

• 针对存在并发的问题，经过添加synchronize关键字实现
• 另外还有获取某篇文章的点赞数、用户全部文章的点赞数、用户点赞的文章方法实现，方法实现比较简单不说明，能够在完整代码中找到

目前存在的不足

• 用户点赞\取消点赞方法中，Redis事务没有保证
• 该应用只适用于单机环境，分布式环境下存在并发问题，分布式锁待完成

十一事后对假期意犹未尽

最后附：完整代码地址
欢迎fork与 star，若有纰漏欢迎指正