springboot整合lua脚本在Redis实现商品库存扣减

1、目的

使用lua脚本,可以保证多条命令的操作原子性;同时可以减少操作IO(比如说判断redis对应数据是否小于0,小于0就重置为100,这个场景一般是取出来再判断,再存放进行,就至少存在2次IO,用lua脚本一条命令1次IO就解决了,在批量扣减情况存在多次IO,lua脚本1次也可以解决),提高速度,降低IO.

2、使用案列

根据传入的产品标识及数量扣减该产品数量;此处为单个产品扣减,可优化为批量产品传入,lua内部用table处理。

2.1 初始化redis参数

添加产品及库存(hash结构)。添加两种水果的库存。

java 复制代码
// 向Redis中添加数据
redisTemplate.opsForHash().put("productMap", "pro1", "{\"name\":\"苹果\",\"stock\":100}");
redisTemplate.opsForHash().put("productMap", "pro2", "{\"name\":\"西瓜\",\"stock\":1200}");

查看结果:

2.2 业务代码

传入lua脚本,实现对应产品库存数量扣减。(可优化为多产品批量扣减)

通过setnx加锁,防止死锁设置锁超时时间,同时业务执行完手动释放锁。设置锁等待时间、及锁等待轮询获取锁。(eg:自旋释放cpu资源重新抢占资源)

java 复制代码
@Test
public void tete(){
    // 向Redis中添加数据
    redisTemplate.opsForHash().put("productMap", "pro1", "{\"name\":\"苹果\",\"stock\":100}");
    redisTemplate.opsForHash().put("productMap", "pro2", "{\"name\":\"西瓜\",\"stock\":1200}");

    // Lua 脚本字符串    
   String luaScript = "local productKey = KEYS[1]; " +
       "local pro = KEYS[2]; " +
       "local lockKey = KEYS[3]; " +
       "local lockTimeout = tonumber(ARGV[1]); " +
       "local deductAmount = tonumber(ARGV[2]); " +
       "local spinIntervalMs = tonumber(ARGV[3]); " +
       "local maxSpinCount = tonumber(ARGV[4]); " +
       "local lockAcquired = redis.call('setnx', lockKey, 1); " +
       "if lockAcquired == 1 then " +
       "    redis.call('pexpire', lockKey, lockTimeout); " +
       "    local currentValue = redis.call('hget', productKey, pro); " +
       "    if currentValue then " +
       "        local dbObj = cjson.decode(currentValue);" +
       "        local currentStock = tonumber(dbObj.stock); " +
       "        if currentStock >= deductAmount then " +
       "            dbObj.stock = currentStock - deductAmount; " +
       "            local updatedValue = cjson.encode(dbObj); " +
       "            redis.call('hset', productKey, pro, updatedValue); " +
       "            redis.call('del', lockKey); " +  // 释放锁
       "            return true; " +
       "        else " +
       "            return false; " +
       "        end " +
       "    else " +
       "        return false; " +
       "    end " +
       "else " +
       "    local spinCount = 0; " +
       "    while spinCount < maxSpinCount do " +
       "        local lockValue = redis.call('get', lockKey); " +
       "        if not lockValue then " +
       "            lockAcquired = redis.call('setnx', lockKey, 1); " +
       "            if lockAcquired == 1 then " +
       "                redis.call('pexpire', lockKey, lockTimeout); " +
       "                local currentValue = redis.call('hget', productKey, pro); " +
       "                if currentValue then " +
       "                   local dbObj = cjson.decode(currentValue);" +
       "                   local currentStock = tonumber(dbObj.stock); " +
       "                    if currentStock >= deductAmount then " +
       "                        dbObj.stock = currentStock - deductAmount; " +
       "                        local updatedValue = cjson.encode(dbObj); " +
       "                        redis.call('hset', productKey, pro, updatedValue); " +
       "                        redis.call('del', lockKey); " +  
       "                        return true; " +
       "                    else " +
       "                        return false; " +
       "                    end " +
       "                else " +
       "                    return false; " +
       "                end " +
       "            end " +
       "            break; " +
       "        end " +
       "        spinCount = spinCount + 1; " +
       "    end " +
       "    return false; " +
       "end";
    
    // 创建DefaultRedisScript对象
    DefaultRedisScript<Boolean> script = new DefaultRedisScript<>();
    script.setScriptText(luaScript);
    script.setResultType(Boolean.class); // 设置返回类型为Boolean
    // 执行脚本
    Boolean result = (Boolean) redisTemplate.execute(script,
            Collections.unmodifiableList(List.of("productMap","pro1","lock:pro1")), // KEYS参数
            "50000", // ARGV参数第一个:锁过期时间(毫秒)
            "10", // ARGV参数第二个:扣减数量
            "1000",// ARGV参数第3个:等待时间
            "5");// ARGV参数第4个:轮询次数
    System.out.println("result1:"+result);
    Boolean result2 = (Boolean) redisTemplate.execute(script,
            Collections.unmodifiableList(List.of("productMap","pro1","lock:pro1")), // KEYS参数
            "50000", // ARGV参数第一个:锁过期时间(毫秒)
            "10", // ARGV参数第二个:扣减数量
            "1000",
            "5");
    System.out.println("result2:"+result2);
    Boolean result3 = (Boolean) redisTemplate.execute(script,
            Collections.unmodifiableList(List.of("productMap","pro1","lock:pro1")), // KEYS参数
            "50000", // ARGV参数第一个:锁过期时间(毫秒)
            "10", // ARGV参数第二个:扣减数量
            "1000",
            "5");
    System.out.println("result3:"+result3);
    if (result2) {
        System.out.println("Stock deduction successful.");
    } else {
        System.out.println("Insufficient stock or lock already acquired.");
    }
    // 验证库存是否正确扣减
    Object updatedValue = redisTemplate.opsForHash().get("productMap", "pro1");
    System.out.println(updatedValue);
    Boolean result5 = (Boolean) redisTemplate.execute(script,
            Collections.unmodifiableList(List.of("productMap","pro2","lock:pro2")), // KEYS参数
            "5000", // ARGV参数第一个:锁过期时间(毫秒)
            "500", // ARGV参数第二个:扣减数量
            "1000",
            "5");
}
2.3 正常执行结果
2.4 若获取锁超时,则会出现扣减失败

脚本执行时间过长会导致。(此处可通过删除手动释放锁实现:模拟业务耗时过长没办法手动释放锁需等待锁国企时间)

第二个扣减等待超时。可通过设置调整添加自旋时间重试或业务代码判断重试机制

相关推荐
neoooo3 分钟前
《锁得住,才能活得久》——一篇讲透 Redisson 分布式锁的技术实录
java·spring boot·redis
花落人散处11 分钟前
SpringAI——整合MCP案例
java·后端
胡斌附体1 小时前
mybatis-plus逻辑删除配置
java·mybatis·mybatis-plus·逻辑删除
Lenyiin1 小时前
《LeetCode 热题 100》整整 100 题量大管饱题解套餐 中
java·c++·python·leetcode·面试·刷题·lenyiin
码字的字节1 小时前
深入理解Java内存与运行时机制:从对象内存布局到指针压缩
java·jvm·内存布局·指针压缩
蒟蒻小袁1 小时前
力扣面试150题--颠倒二进制位
java·算法·leetcode
WAZYY06191 小时前
处理jdk21版本及No such algorithm: SM4/ECB/PKCS5Padding jar包冲突问题
java·jar·jdk21·sm4
Mr Aokey1 小时前
告别配置混乱!Spring Boot 中 Properties 与 YAML 的深度解析与最佳实践
java·spring·rpc
java叶新东老师2 小时前
maven 打包报错 process terminated
java·maven·intellij-idea
皮卡蛋炒饭.2 小时前
C++中既重要又困难的部分—类和对象
java·开发语言