引言
功能點設計
比如我喜歡發文章的掘金網站就有點讚的功能，統計文章點讚的總數，用戶所有文章的點贊數，因此設計的點贊功能模塊具有以下功能點：
- 某篇文章的點贊數
- 用戶所有文章的點贊數
- 用戶點讚的文章
- 持久化到MySQL數據庫
數據庫設計
- Redis數據庫設計 Redis是K-V數據庫，沒有統一的數據結構，針對不同的功能點設計了不同的K-V存儲結構
- 用戶某篇文章的點贊數 使用HashMap數據結構，HashMap中的key為articleId，value為Set，Set中的值為用戶ID，即HashMap<string>>/<string>
- 用戶總的點贊數 使用HashMap數據結構，HashMap中的key為userId，value為String記錄總的點贊數
- 用戶點讚的文章 使用HashMap數據結構，HashMap中的key為userId，value為Set，Set中的值為文章ID，即HashMap<string>>/<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);
}
}
複製代碼
/<long>/<long>- 取消點贊
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);
}
複製代碼
/<long>/<long> - 異步落庫
@Scheduled(cron = "0 0 0/1 * * ? ")
public void redisDataToMySQL() {
logger.info("time:{}，開始執行Redis數據持久化到MySQL任務", LocalDateTime.now().format(formatter));
//1.更新文章總的點贊數
Map<string> articleCountMap = redisTemplate.opsForHash().entries(ARTICLE_LIKED_USER_KEY);
for (Map.Entry<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));
}
複製代碼
/<long>/<string>/<string>
說明：
- 針對存在併發的問題，通過添加synchronize關鍵字實現
- 另外還有獲取某篇文章的點贊數、用戶所有文章的點贊數、用戶點讚的文章方法實現，方法實現比較簡單不說明，可以在完整代碼中找到
目前存在的不足
- 用戶點贊\\取消點贊方法中，Redis事務沒有保證
- 該應用只適用於單機環境，分佈式環境下存在併發問題，分佈式鎖待完成
小結
私信小編髮送“架構”（免費獲取JAVA相關的面試架構資料喲）
最後，每一位讀到這裡的Java程序猿朋友們，感謝你們能耐心地看完。希望在成為一名更優秀的Java程序猿的道路上，我們可以一起學習、一起進步！都能贏取白富美，走向架構師的人生巔峰!
分享到:
閱讀更多 JAVA大飛哥 的文章
