基于redis的高并发秒杀的java实现!

在Redis的事务中,WATCH命令可用于提供CAS(check-and-set)功能。假设我们通过WATCH命令在事务执行之前监控了多个Keys,倘若在WATCH之后有任何Key的值发生了变化,EXEC命令执行的事务都将被放弃,同时返回Null multi-bulk应答以通知调用者事务执行失败。例如,我们再次假设Redis中并未提供incr命令来完成键值的原子性递增,如果要实现该功能,我们只能自行编写相应的代码。其伪码如下:


val = GET mykey


val = val + 1


SET mykey $val

以上代码只有在单连接的情况下才可以保证执行结果是正确的,因为如果在同一时刻有多个客户端在同时执行该段代码,那么就会出现多线程程序中经常出现的一种错误场景--竞态争用(race condition)。比如,客户端A和B都在同一时刻读取了mykey的原有值,假设该值为10,此后两个客户端又均将该值加一后set回Redis服务器,这样就会导致mykey的结果为11,而不是我们认为的12。为了解决类似的问题,我们需要借助WATCH命令的帮助,见如下代码:


WATCH mykey


val = GET mykey


val = val + 1


MULTI


SET mykey $val


EXEC

和此前代码不同的是,新代码在获取mykey的值之前先通过WATCH命令监控了该键,此后又将set命令包围在事务中,这样就可以有效的保证每个连接在执行EXEC之前,如果当前连接获取的mykey的值被其它连接的客户端修改,那么当前连接的EXEC命令将执行失败。这样调用者在判断返回值后就可以获悉val是否被重新设置成功。



根据这样的思路,我们在JAVA下进行实现:


新建一个项目,首先引入JAVA的redis操作库:Jedis,这里用的是jedis-2.9.0.jar



a) 新建一个类:MyRedistest.class做线程操作


package com.myredistest;
import java.util.Random;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
 
import redis.clients.jedis.Jedis;


public class MyRedistest {
     public static void main(String[] args) {
            final String watchkeys = "watchkeys";
            ExecutorService executor = Executors.newFixedThreadPool(20);  //20个线程池并发数
 
            final Jedis jedis = new Jedis("192.168.56.101", 6379);
            jedis.set(watchkeys, "100");//设置起始的抢购数
           // jedis.del("setsucc", "setfail");
            jedis.close();
             
            for (int i = 0; i < 1000; i++) {//设置1000个人来发起抢购
                executor.execute(new MyRunnable("user"+getRandomString(6)));
            }
            executor.shutdown();
        }
 
      
     public static String getRandomString(int length) { //length是随机字符串长度
            String base = "abcdefghijklmnopqrstuvwxyz0123456789";  
            Random random = new Random();  
            StringBuffer sb = new StringBuffer();  
            for (int i = 0; i < length; i++) {  
                int number = random.nextInt(base.length());  
                sb.append(base.charAt(number));  
            }  
            return sb.toString();  
         } 
}


b) 建一个类:MyRunnable.class 实现Runnable做线程操作:


package com.myredistest;
import java.util.List;
import redis.clients.jedis.Jedis;
import redis.clients.jedis.Transaction;
 
public class MyRunnable implements Runnable {
 
    String watchkeys = "watchkeys";// 监视keys
    Jedis jedis = new Jedis("192.168.56.101", 6379);
    String userinfo;
    public MyRunnable() {
    }
    public MyRunnable(String uinfo) {
        this.userinfo=uinfo;
    }
    @Override
    public void run() {
        try {
            jedis.watch(watchkeys);// watchkeys
 
            String val = jedis.get(watchkeys);
            int valint = Integer.valueOf(val);
            
            if (valint <= 100 && valint>=1) {
            
                 Transaction tx = jedis.multi();// 开启事务
               // tx.incr("watchkeys");
                tx.incrBy("watchkeys", -1);
 
                List<Object> list = tx.exec();// 提交事务,如果此时watchkeys被改动了,则返回null
                 
                if (list == null ||list.size()==0) {
 
                    String failuserifo = "fail"+userinfo;
                    String failinfo="用户:" + failuserifo + "商品争抢失败,抢购失败";
                    System.out.println(failinfo);
                    /* 抢购失败业务逻辑 */
                    jedis.setnx(failuserifo, failinfo);
                } else {
                    for(Object succ : list){
                         String succuserifo ="succ"+succ.toString() +userinfo ;
                         String succinfo="用户:" + succuserifo + "抢购成功,当前抢购成功人数:"
                                 + (1-(valint-100));
                         System.out.println(succinfo);
                         /* 抢购成功业务逻辑 */
                         jedis.setnx(succuserifo, succinfo);
                    }
                }
            } else {
                String failuserifo ="kcfail" +  userinfo;
                String failinfo1="用户:" + failuserifo + "商品被抢购完毕,抢购失败";
                System.out.println(failinfo1);
                jedis.setnx(failuserifo, failinfo1);
                // Thread.sleep(500);
                return;
            }
 
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            jedis.close();
        }
    }
}

执行MyRedistest ,查看redis中插入的key值