Redis實現點贊功能模組

何甜甜在嗎發表於2019-10-15
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欄位

流程圖

Redis實現點贊功能模組

流程圖比較簡單，點贊和取消點贊基本實現步驟相同

  • 引數校驗 對傳入的引數進行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事務沒有保證
  • 該應用只適用於單機環境，分散式環境下存在併發問題，分散式鎖待完成

十一過後對假期意猶未盡

Redis實現點贊功能模組

最後附：完整程式碼地址
歡迎fork與 star，如有紕漏歡迎指正

相關文章