Redis實現點贊功能模塊

以前看了一篇文章,講redis的應用場景,其中一個應用場景就是實現點贊功能,紙上得來恐覺淺,必須實戰一波git

功能點設計

好比我喜歡發文章的掘金網站就有點讚的功能,統計文章點讚的總數,用戶全部文章的點贊數,所以設計的點贊功能模塊具備如下功能點:github

  • 某篇文章的點贊數
  • 用戶全部文章的點贊數
  • 用戶點讚的文章
  • 持久化到MySQL數據庫

數據庫設計

  • Redis數據庫設計 RedisK-V數據庫,沒有統一的數據結構,針對不一樣的功能點設計了不一樣的K-V存儲結構redis

    • 用戶某篇文章的點贊數 使用HashMap數據結構,HashMap中的keyarticleIdvalueSetSet中的值爲用戶ID,即HashMap<String, Set<String>>
    • 用戶總的點贊數 使用HashMap數據結構,HashMap中的keyuserIdvalueString記錄總的點贊數
    • 用戶點讚的文章 使用HashMap數據結構,HashMap中的keyuserIdvalueSetSet中的值爲文章ID,即HashMap<String, Set<String>>
  • MySQL數據庫設計 最主要的兩張表,article表和user_like_article數據庫

    • article表結構
    字段值 字段類型 說明
    article_name varchar 文章名字
    content blob 文章內容
    total_like_count bigint 文章總點贊數

    文章總的點贊數須要和Redis中的點贊數進行同步json

    • user_like_article表結構
    字段值 字段類型 說明
    user_id bigint 用戶ID
    article_id bigint 文章ID

    記錄用戶點贊文章的信息,是一張中間表bash

說明:表結構設計省略了iddeletedgmt_creategmt_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事務沒有保證
  • 該應用只適用於單機環境,分佈式環境下存在併發問題,分佈式鎖待完成

十一事後對假期意猶未盡數據庫設計

最後附:完整代碼地址
歡迎forkstar,若有紕漏歡迎指正

相關文章
相關標籤/搜索