/**
 * @description 商品類
 * @author yiridancan
 * @date 2024/3/23 9:06
 */
public class Goods {

    private int id;

    /**
     * 商品名稱
     */
    private String name;

    /**
     * 庫存數(shù)量
     */
    private int inventoryCount;

    public int getId() {
        return id;
    }

    public void setId(int id) {
        this.id = id;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public int getInventoryCount() {
        return inventoryCount;
    }

    public void setInventoryCount(int inventoryCount) {
        this.inventoryCount = inventoryCount;
    }
}

import com.yiridancan.reduceInventory.entity.Goods;
import com.yiridancan.reduceInventory.mapper.GoodsMapper;
import com.yiridancan.reduceInventory.service.IGoodsService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.Objects;

/**
 *  商品實(shí)現(xiàn)類
 * @author yiridancan
 * @date 2024/3/23 18:35
 */
@Slf4j
@Service
public class GoodsServiceImpl implements IGoodsService {

    @Autowired
    private GoodsMapper goodsMapper;
    /**
     * 扣減庫存
     * @param goodsId 商品id
     * @author yiridancan
     * @date 2024/3/23 18:33
     */
    @Override
    public void reduceInventory(int goodsId) {
        //1.根據(jù)商品id獲取商品庫存數(shù)量
        Goods goods = goodsMapper.findGoodsInventory(goodsId);
        if(Objects.isNull(goods)){
            log.error("未獲取到商品信息");
            return;
        }
        //2.如果庫存數(shù)量大于0則扣減庫存，如果等于0代表沒有貨物打印錯誤信息
        if(goods.getInventoryCount() > 0 ){
            //默認(rèn)扣減庫存1
            goods.setInventoryCount(goods.getInventoryCount()-1);
            goodsMapper.updateGoodsInventory(goods);
            log.info("{}扣減庫存成功，扣減后庫存為：{}",goods.getName(),goods.getInventoryCount());
        }else {
            log.error("{}庫存為0",goods.getName());
        }
    }
}

    @Test
	void contextLoads() {
		//商品id
		int goodsId = 1;
		//創(chuàng)建固定數(shù)量的線程池
		int num = 20;
		ExecutorService executorService = Executors.newFixedThreadPool(num);
		//模擬20個并發(fā)同時請求接口
		for (int i = 0; i < num; i++) {
			executorService.submit(() -> {
					goodsService.reduceInventory(goodsId);
			});
		}
		executorService.shutdown();
		try {
			executorService.awaitTermination(1, TimeUnit.MINUTES);
		} catch (InterruptedException e) {
			throw new RuntimeException(e);
		}
		//獲取商品最終庫存數(shù)量
		Goods goodsInventory = goodsMapper.findGoodsInventory(goodsId);
		if(Objects.isNull(goodsInventory)){
			return;
		}
		log.info("{}商品最終庫存為：{}",goodsInventory.getName(),goodsInventory.getInventoryCount());
	}

    /**
     * 通過樂觀鎖實(shí)現(xiàn)扣減庫存
     * @author yiridancan
     * @date 2024/3/25 22:33
     * @param goodsId 商品id
     */
    void casReduceInventory(int goodsId);

    /**
     * 通過樂觀鎖實(shí)現(xiàn)扣減庫存
     * @param goodsId 商品id
     * @author yiridancan
     * @date 2024/3/25 22:33
     */
    @Override
    public void casReduceInventory(int goodsId) {
        int retryCount = 0;
        //重試次數(shù)設(shè)置為3，避免無休止的重試占用紫鳶
        while (retryCount <=3){
            //1.根據(jù)商品id獲取商品信息
            Goods goods = goodsMapper.findGoodsInventory(goodsId);
            if(Objects.isNull(goods) || goods.getInventoryCount() == 0){
                log.error("未獲取到商品信息或庫存數(shù)量不足");
                return;
            }
            //默認(rèn)扣減庫存1
            goods.setInventoryCount(goods.getInventoryCount()-1);
            int updateRow = goodsMapper.updateGoodsInventoryByCAS(goods);
            //如果修改條數(shù)大于0代表扣減庫存成功
            if(updateRow > 0 ){
                log.info("{}扣減庫存成功，扣減后庫存為：{}",goods.getName(),goods.getInventoryCount());
                return;
            }
            retryCount++;
            log.error("{}商品被修改過，進(jìn)行重試！！版本號：{}",goods.getName(),goods.getDataVersion());
        }
    }

<update id="updateGoodsInventoryByCAS">
        update goods set inventory_count=#{inventoryCount},data_version=data_version+1 where id=#{id} and data_version=#{dataVersion}
</update>

local key = KEYS[1] -- 商品的鍵名
local amount = tonumber(ARGV[1]) -- 扣減的數(shù)量

-- 獲取商品當(dāng)前的庫存量
local stock = tonumber(redis.call('get', key))

-- 如果庫存足夠，則減少庫存并返回新的庫存量
if stock >= amount then
    redis.call('decrby', key, amount)
    return redis.call('get', key)
else
    return "INSUFFICIENT STOCK"
end

/**
     * 通過Redis扣減庫存
     *
     * @param goodsId 商品id
     * @author yiridancan
     * @date 2024/3/27 15:48
     */
    @Override
    public void redisReduceInventory(int goodsId) {
        String prefix = "goodsInventory:";
        //將商品數(shù)據(jù)緩存到Redis中，key是商品id，value是商品庫存數(shù)量
        goodsMapper.findGoodsAll().forEach(goods -> {
            stringRedisTemplate.opsForValue().set(prefix+goods.getId(),String.valueOf(goods.getInventoryCount()));
        });

        //lua腳本，一般放在文件中
        String script = "local key = KEYS[1] -- 商品的鍵名\n" +
                "local amount = tonumber(ARGV[1]) -- 扣減的數(shù)量\n" +
                "\n" +
                "-- 獲取商品當(dāng)前的庫存量\n" +
                "local stock = tonumber(redis.call('get', key))\n" +
                "\n" +
                "-- 如果庫存足夠，則減少庫存并返回新的庫存量\n" +
                "if stock >= amount then\n" +
                "    redis.call('decrby', key, amount)\n" +
                "    return redis.call('get', key)\n" +
                "else\n" +
                "    return \"INSUFFICIENT STOCK\"\n" +
                "end\n";

        DefaultRedisScript<String> redisScript = new DefaultRedisScript<>(script, String.class);

        // 創(chuàng)建一個包含庫存key的列表
        List<String> keys = Collections.singletonList(prefix + goodsId);
        // 創(chuàng)建一個包含扣減數(shù)量的參數(shù)列表
        List<String> args = Collections.singletonList(Integer.toString(1));

        // 執(zhí)行Lua腳本，傳入鍵列表和參數(shù)列表
        String result = stringRedisTemplate.execute(redisScript, keys, args.toArray(new String[0]));
        //如果不是庫存不足代表扣減成功
        if(!result.equals("INSUFFICIENT STOCK")){
            log.info("扣減庫存成功，庫存數(shù)量：{}",result);
        }else {
            log.error("庫存數(shù)量不足");
        }
    }