Redis實現點贊功能模組
阿新 • • 發佈:2019-12-31
之前看了一篇文章,講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,likedPostId);
synchronized (this) {
//只有未點讚的使用者才可以進行點贊
likeArticleLogicValidate(articleId,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:{}",likedPostId);
}
}
複製程式碼
- 取消點贊
public void unlikeArticle(Long articleId,likedPostId); //引數校驗
logger.info("取消點贊資料存入redis開始,articleId:{},likedUserId:{},likedPostId:{}",likedPostId);
//1.使用者總點贊數-1
synchronized (this) {
//只有點讚的使用者才可以取消點贊
unlikeArticleLogicValidate(articleId,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(--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,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,FastjsonUtil.serialize(likePostIdSet));
}
logger.info("取消點贊資料存入redis結束,articleId:{},likedUserId:{},likedPostId:{}",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,如有紕漏歡迎指正