lua脚本操作Redis

一次扣减一个商品库存

@Component

public class LuaDemo {

复制代码
@Autowired
private StringRedisTemplate redisTemplate;


public void test() {

  List<String> keys = new ArrayList<>();
    keys.add("yanjie007");

    String lua = "return redis.call('set',KEYS[1],ARGV[1])";

    RedisScript<String> redisScript=  RedisScript.of(lua,String.class);

    String shuai = stringRedisTemplate.execute(redisScript, keys, "今天真帅");

    System.out.println(shuai);

}

}

//扣减库存

@Test

void test2() throws IOException {

valueOperations.set(key,15L);

StringBuilder sb = new StringBuilder();

sb.append(" local key = KEYS[1] ");

sb.append(" local qty = ARGV[1] ");

sb.append(" local redis_qty = redis.call('get',key) ");

sb.append(" if tonumber(redis_qty) >= tonumber(qty) then ");

sb.append(" redis.call('decrby',key,qty) ");

sb.append(" return -1 ");

sb.append(" else ");

sb.append(" return tonumber(redis_qty) "); // 0, 1,2,3 ...

sb.append(" end ");

sb.append(" ");

RedisScript script = RedisScript.of(sb.toString(),Long.class);

ExecutorService executorService = Executors.newCachedThreadPool();

复制代码
    for (int i = 1; i <= 10; i++) {
        int finalI = i;
        executorService.execute(()->{
            int needQty = RandomUtil.randomInt(1, 5);
            Long qty =  stringRedisTemplate.execute(script, CollUtil.newArrayList(key), needQty+"");
         if(qty.intValue() == -1 ){
             System.out.println("线程"+ finalI +"扣减成功,需求量是:"+needQty);
          } else {

             System.out.println("线程"+ finalI +"扣减失败,当前库存变量是:"+qty);
         }
        });
    }


    System.in.read();
}

一次扣减多个商品的库存

void test4() {

复制代码
    StringBuilder sb = new StringBuilder();
    sb.append(" local table = {}  "); // 你要扣减的key :product.1
    sb.append(" local values =  redis.call('mget',  unpack(KEYS) )"); // [product.1,product.2]   =>  product.1 product.2
    sb.append(" for i = 1, #KEYS   do  ");
    sb.append("   if  tonumber(ARGV[i]) > tonumber(values[i])   then ");
    sb.append("     table[#table + 1] =  KEYS[i] .. '=' .. values[i] "); // product.1=23
    sb.append("   end ");
    sb.append(" end ");
    sb.append(" if #table > 0 then ");
    sb.append("   return table  ");
    sb.append(" end ");
    sb.append(" for i = 1 , #KEYS do ");
    sb.append("   redis.call('decrby',KEYS[i],ARGV[i])  ");
    sb.append(" end ");
    sb.append(" return {} ");
    RedisScript<List> luaScript = RedisScript.of(sb.toString(), List.class);

      List<StockProduct> stockProducts =  new ArrayList<>();
      stockProducts.add(new StockProduct(5,1));
      stockProducts.add(new StockProduct(4,2));

    List<String> keys = stockProducts.stream().map(it -> "product." + it.getId()).collect(Collectors.toList());
    Object[] qtys = stockProducts.stream().map(it -> it.getQty() + "").toArray();
    List<String> list = stringRedisTemplate.execute(luaScript,
            keys,
            qtys);
    if(list.isEmpty()){
        System.out.println("库存冻结成功");
    } else {

        for (String key_qty : list) {
            String[] split = key_qty.split("=");
            System.out.println(split[0] + "库存不足,剩余库存量:" + split[1]);
        }

    }


    ThreadUtil.safeSleep(3000);


}
相关推荐
Hello.Reader2 小时前
Redis 延迟排查与优化全攻略
数据库·redis·缓存
东窗西篱梦9 小时前
Redis集群部署指南:高可用与分布式实践
数据库·redis·分布式
半新半旧10 小时前
Redis集群和 zookeeper 实现分布式锁的优势和劣势
redis·分布式·zookeeper
@ chen12 小时前
Redis事务机制
数据库·redis
静若繁花_jingjing14 小时前
Redis线程模型
java·数据库·redis
在肯德基吃麻辣烫14 小时前
《Redis》缓存与分布式锁
redis·分布式·缓存
先睡20 小时前
Redis的缓存击穿和缓存雪崩
redis·spring·缓存
weixin_446122461 天前
JAVA内存区域划分
java·开发语言·redis
TT哇1 天前
JavaEE==网站开发
java·redis·java-ee
qq_392397121 天前
Redis常用操作
数据库·redis·wpf