订单抢购防超卖Lua脚本

置顶向优秀致敬

已于 2023-07-12 14:25:32 修改

阅读量804

点赞数

文章标签： java redis lua

于 2020-06-19 10:17:06 首次发布

本文链接：https://blog.csdn.net/lq_1993/article/details/106850727

版权

该方法主要用来防止超卖,将Compare And Set使用Lua脚本封装,传入Redis执行.

因为Redis是单线程执行,所以Lua脚本的执行具有原子性.

/**

 * 该方法主要用来防止超卖,将Compare And Set使用Lua脚本封装,传入Redis执行.

 * 因为Redis是单线程执行,所以Lua脚本的执行具有原子性.

 * @param productId 产品ID

 * @param increment 购买金额

 * @return "NOT_ENOUGH":可售金额不足或者加入购买金额后,可售金额不足

 * "ENOUGH":加入购买金额后,可售金额足够

 * "JUST_SOLD_OUT":加入购买金额后,产品刚好售罄

 */

public StockStatus compareAndSetProductCachedStock(Integer productId, BigDecimal increment) {

    String script = "local key, amount = KEYS[1], ARGV[1];\n" +

            "local sold_amount = redis.call('hget', key, 'sold_amount') * 100; \n" +

            "local raised_amount = redis.call('hget', key, 'raised_amount')*100; \n" +

            "local temp_amount = sold_amount + amount*100;  \n" +

            "if tostring(temp_amount) ~= tostring(raised_amount) and temp_amount > raised_amount then\n" +

            "    return 'NOT_ENOUGH';\n" +

            "elseif tostring(temp_amount) ~= tostring(raised_amount) and temp_amount < raised_amount then\n" +

            "    redis.call('hincrbyfloat', key, 'sold_amount', amount);\n" +

            "    return 'ENOUGH';\n" +

            "else\n" +

            "    redis.call('hincrbyfloat', key, 'sold_amount', amount);\n" +

            "    redis.call('hset', key, 'trade_status_id', 3);\n" +

            "    return 'JUST_SOLD_OUT';\n" +

            "end;";

    String key = getCachedProductKey(productId);

    List<String> keys = new ArrayList<String>() {{

        add(key);

    }};

    String result = String.valueOf(RedisUtil.execute(script, keys, String.valueOf(increment.setScale(2, BigDecimal.ROUND_HALF_UP))));

    return StockStatus.valueOf(result);

}