快捷導(dǎo)航

非常全面的Java?SpringBoot點贊功能實現(xiàn)

更新時間：2022年01月23日 11:02:28 作者：JoeyHua

但是這些功能再項目中是高頻出現(xiàn)的,如果直接操作數(shù)據(jù)庫的話，對數(shù)據(jù)庫壓力太大。那遇到這個問題怎么解決?這篇文章主要給大家介紹了關(guān)于Java?SpringBoot點贊功能實現(xiàn)?的相關(guān)資料,需要的朋友可以參考下

前言

最近公司在做一個NFT商城的項目，大致就是一個只買賣數(shù)字產(chǎn)品的平臺，項目中有個需求是用戶可以給商品點贊，還需要獲取商品的點贊總數(shù)，類似下圖

起初感覺這功能很好實現(xiàn)，無非就是加個點贊表嘛，后來發(fā)現(xiàn)事情并沒有這么簡單。

一開始的設(shè)計是這樣的，一共有三張表：商品表、用戶表、點贊表，用戶點贊的時候把用戶id和商品id加到點贊表中，并給對應(yīng)的商品點贊數(shù)+1?？雌饋頉]什么問題，邏輯也比較簡單，但是測試的時候缺發(fā)現(xiàn)了奇怪的bug，點贊數(shù)量有時候會不正確，結(jié)果會比預(yù)期的大。

下面貼下關(guān)鍵代碼（項目使用了Mybatis-Plus）：

public boolean like(Integer userId, Integer productId) {
        // 查詢是否有記錄，如果有記錄直接返回
        Like like = getOne(new QueryWrapper<Like>().lambda()
                .eq(Like::getUserId, userId)
                .eq(Like::getProductId, productId));
        if(like != null) {
            return true;
        }

        // 保存并商品點贊數(shù)加1
        save(Like.builder()
                .userId(userId)
                .productId(productId)
                .build());
        return productService.update(new UpdateWrapper<Product>().lambda()
                .setSql("like_count = like_count + 1")
                .eq(Product::getId, productId));
}

看上去沒什么問題，但是測試后數(shù)據(jù)卻不正確，為什么呢？

實際上這是一個并發(fā)問題，只要在并發(fā)的情況下就會出現(xiàn)問題，我們知道Spring Mvc是基于servlet的，servlet在接收到用戶請求后會從線程池中拿一個線程分配給它，每個請求都是一個單獨的線程。試想一下，如果A線程在執(zhí)行完查詢操作后，發(fā)現(xiàn)沒有記錄，隨后由于CPU調(diào)度，把控制權(quán)讓了出去，然后B線程執(zhí)行查詢，也發(fā)現(xiàn)沒有記錄，這時候A和B線程都會執(zhí)行保存并商品點贊數(shù)加1這個操作，導(dǎo)致數(shù)據(jù)不正確。

CPU操作順序：A線程查詢 -> B線程查詢 -> A線程保存 -> B線程保存

下面使用JMeter模擬一下并發(fā)的情況，模擬用戶在1秒內(nèi)對商品執(zhí)行100次點贊請求，結(jié)果應(yīng)該是1，但得到的結(jié)果卻是28（實際結(jié)果不一定是28，可能是任何數(shù)字）。

解決方案

青銅版

使用synchronized關(guān)鍵字鎖住讀寫操作，操作完成后釋放鎖

public boolean like(Integer userId, Integer productId) {
        String lock = buildLock(userId, productId);
        synchronized (lock) {
            // 查詢是否有記錄，如果有記錄直接返回
            Like like = getOne(new QueryWrapper<Like>().lambda()
                    .eq(Like::getUserId, userId)
                    .eq(Like::getProductId, productId), false);
            if(like != null) {
                return true;
            }

            // 保存并商品點贊數(shù)加1
            save(Like.builder()
                    .userId(userId)
                    .productId(productId)
                    .build());
            return productService.update(new UpdateWrapper<Product>().lambda()
                    .setSql("like_count = like_count + 1")
                    .eq(Product::getId, productId));
        }
}

private String buildLock(Integer userId, Integer productId) {
        StringBuilder sb = new StringBuilder();
        sb.append(userId);
        sb.append("::");
        sb.append(productId);
        String lock = sb.toString().intern();

        return lock;
}

這里要注意一點，使用String作為鎖時一定要調(diào)用intern()方法，intern()會先從常量池中查找有沒有相同的String，如果有就直接返回，沒有的話會把當(dāng)前String加入常量池，然后再返回。如果不調(diào)用這個方法鎖會失效。

JMeter性能數(shù)據(jù)

優(yōu)點：

保證了正確性

缺點：

性能太差，并發(fā)低的情況下還可以應(yīng)付，并發(fā)高時用戶體驗極差

白銀版

點贊表user_id和product_id加上聯(lián)合索引，并使用try catch捕獲異常，防止報錯。由于使用了聯(lián)合索引，所以不需要在新增前查詢了，mysql會幫我們做這件事。

public boolean like(Integer userId, Integer productId) {
        try {
            // 保存并商品點贊數(shù)加1
            save(Like.builder()
                    .userId(userId)
                    .productId(productId)
                    .build());
            return productService.update(new UpdateWrapper<Product>().lambda()
                    .setSql("like_count = like_count + 1")
                    .eq(Product::getId, productId));
        }catch (DuplicateKeyException exception) {

        }

        return true;
}

JMeter性能數(shù)據(jù)

優(yōu)點：

性能比上一個方案好

缺點：

中規(guī)中矩，沒什么大的缺點

黃金版

使用Redis緩存點贊數(shù)據(jù)（點贊操作使用lua腳本實現(xiàn)，保證操作的原子性），然后定時同步到mysql。

注意：Redis需要開啟持久化，最好aof和rdb都開啟，不然重啟數(shù)據(jù)就丟失了

public boolean like(Integer userId, Integer productId) {
        List<String> keys = new ArrayList<>();
        keys.add(buildUserRedisKey(userId));
        keys.add(buildProductRedisKey(productId));

        int value1 = 1;

        redisUtil.execute("lua-script/like.lua", keys, value1);

        return true;
}

private String buildUserRedisKey(Integer userId) {
        return "userId_" + userId;
}

private String buildProductRedisKey(Integer productId) {
        return "productId_" + productId;
}

lua腳本

local userId = KEYS[1]
local productId = KEYS[2]
local flag = ARGV[1] -- 1：點贊 0：取消點贊


if flag == '1' then
  -- 用戶set添加商品并商品點贊數(shù)加1
  if redis.call('SISMEMBER', userId, productId) == 0 then
    redis.call('SADD', userId, productId)
    redis.call('INCR', productId)
  end
else
  -- 用戶set刪除商品并商品點贊數(shù)減1
  redis.call('SREM', userId, productId)
  local oldValue = tonumber(redis.call('GET', productId))
  if oldValue and oldValue > 0 then
    redis.call('DECR', productId)
  end
end

return 1

JMeter性能數(shù)據(jù)

優(yōu)點：

性能非常好

缺點：

數(shù)據(jù)量多了內(nèi)存占用較高總結(jié)

如果對性能沒有要求，可以使用白銀版的實現(xiàn)方式，如果有要求，就使用黃金版的方式，內(nèi)存占用大的問題也可以通過一些手段來解決，比如可以根據(jù)業(yè)務(wù)需求定期刪除一些不常用的緩存數(shù)據(jù)，但是相對應(yīng)的，查詢的時候就需要在查詢失敗時再去查數(shù)據(jù)庫。