import redis.clients.jedis.Jedis;

public class LuaLimitByCount extends Thread {

    @Override
    public void run() {
        Jedis jedis=new Jedis("192.168.159.33",6379);
        //在本线程内，模拟在单位时间内发5个请求
        for(int visitNum=0;visitNum<5;visitNum++){
            boolean visitFlag=LimitByCount.canVisit(jedis,Thread.currentThread().getName(),"10","3");
            if(visitFlag){
                System.out.println(Thread.currentThread().getName()+" can visit.");
            }else{
                System.out.println(Thread.currentThread().getName()+" can not visit.");
            }
        }
    }

    public static void main(String[] args) {
        //开启3个线程
        for(int cnt=0;cnt<3;cnt++){
            new LuaLimitByCount().start();
        }
    }
}
//封装是否需要限流的方法
class LimitByCount{
    //判断是否需要限流
    public static boolean canVisit(Jedis jedis,String modelName,String limitTime,
                                   String limitNum){
        String script="local obj=KEYS[1] \n" +
                "local limitNum=tonumber(ARGV[1]) \n" +
                "local curVisitNum=tonumber(redis.call('get',obj) or \"0\")\n" +
                "if curVisitNum+1>limitNum then \n" +
                "   return 0\n" +
                "else  \n" +
                "   redis.call(\"incrby\",obj,\"1\")\n" +
                "   redis.call(\"expire\",obj,\"10\")\n" +
                "   return curVisitNum+1\n" +
                "end";
        String retVal=jedis.eval(script,1,modelName,limitNum,limitTime).toString();
        if("0".equals(retVal)){
            return false;   //不能继续访问
        }else{
            return true;
        }
    }
}

  在main函数里，通过for循环启动了3个线程，并通过它们的run方法在短时间里调用5次LimitByCount类的canVisit方法。结果如下：

  可以看到，在main函数里创建的3个线程均只有3次请求得到允许，其他请求超过了限流最大访问量，所以被“限流”了。

2、用lua脚本防止超卖

  超卖是指在秒杀活动里多卖出了商品，比如某秒杀系统里最多只能卖出100件，但是并发控制没有做好，最终有100多个请求下单成功，这样就会给商家造成损失。
  lua脚本天然具有原子性，而且执行lua脚本的Redis服务器是以单线程模式处理命令，所以用lua脚本能有效地防止超卖。在如下的lua脚本里实现了防超卖的效果。该lua脚本只有一个KEYS[1]参数，用来传入表示商品的键。

local existedNum=tonumber(redis.call('get',KEYS[1]))
if(existedNum>0) then
	redis.call('incrby',KEYS[1],-1)
	return existedNum
end
return -1

  在运行该脚本前，需要确保Redis服务器已经存在(KEYS[1],商品个数)这个键值对。在第1行里，先通过redis.call方法调用get命令，获得该商品当前的存货数，如果通过第2行的if判断发现大于0，就先通过第3行的incrby命令对该商品的存货书进行减1操作，并通过第4行的语句返回当前的商品存货数，反之则执行第6行的语句，返回-1.也就是说，如果运行该脚本得到-1，就说明该次请求会导致超卖，否则能继续后继的购买动作。

package com.baizhi;

import redis.clients.jedis.Jedis;

public class AvoidSellTooMuch extends Thread {
    @Override
    public void run() {
        Jedis jedis=new Jedis("192.168.159.33",6379);
        //在本线程内，模拟在单位时间内发5个请求
        boolean sellFlag=CheckUtil.canSell(jedis,"Computer");
        if(sellFlag){
            System.out.println(Thread.currentThread().getName()+" can buy.");
        }else{
            System.out.println(Thread.currentThread().getName()+" can not buy.");
        }
    }

    public static void main(String[] args) {
        //创建同Redis的连接
        Jedis jedis=new Jedis("192.168.159.33",6379);
        //预设5个电脑商品，并设置10秒的生存时间
        jedis.set("Computer","5");
        jedis.expire("Computer",10);
        //开启10个线程来抢购
        for(int cnt=0;cnt<10;cnt++){
            new AvoidSellTooMuch().start();
        }
    }
}
class CheckUtil{
    //判断当前请求是否会导致超卖
    public static boolean canSell(Jedis jedis,String modelName){
        //防止超卖的脚本
        String script="local existedNum=tonumber(redis.call('get',KEYS[1]))\n" +
                "if(existedNum>0) then\n" +
                "   redis.call('incrby',KEYS[1],-1)\n" +
                "   return existedNum\n" +
                "end\n" +
                "return -1\n";
        String retVal=jedis.eval(script,1,modelName,modelName).toString();
        if("-1".equals(retVal)){
            return false;   //不能继续访问
        }else{
            return true;
        }
    }
}