1 package cn.hjf.redis.rediDemo.lock;
 2 
 3 import java.util.UUID;
 4 
 5 import redis.clients.jedis.Jedis;
 6 import redis.clients.jedis.Transaction;
 7 /**
 8  * redis setnx 实现分布式锁
 9  * 真实环境下应该是在分布式多进程的情况下
10  * @author hjf
11  */
12 public class SimpleLock
13 {
14     Jedis conn = new Jedis("127.0.0.1", 6379);
15     
16     private final static String LOCK_NAME = "lock";
17     
18     // 获得锁  重入锁和非重入锁 
19     // 设置超时时间 
20     public String accequireLock(int timeOut){
21         // 随机生成一个uuid
22         String uuid = UUID.randomUUID().toString();
23         // 结束时间 
24         long end = System.currentTimeMillis() + timeOut;
25         // 设置成功返回1  失败则返回0  
26         // 当且仅当key不存在时将key设为value
27         // 若给定的key已经存在 那么setnx不会做任何操作
28         while(System.currentTimeMillis() < end){
29             // setnx()和expire()加起来不是一个原子操作
30             if(conn.setnx(LOCK_NAME, uuid).intValue() == 1){
31                 // 增加redis的超时机制 一旦出现异常等情况可以自动去释放锁
32                 conn.expire(LOCK_NAME, timeOut);   
33                 return uuid;
34             }
35             
36             // 检查是否设置超时机制(保证原子性)
37             if(conn.ttl(LOCK_NAME) == -1){  
38                 conn.expire(LOCK_NAME, timeOut);
39             }
40             
41             try
42             {
43                 // 未获得锁时 休眠一段时间
44                 Thread.sleep(1000);
45             } catch (InterruptedException e)
46             {
47                 e.printStackTrace();
48             }
49         }
50         
51         return null;
52     }
53     
54     // 释放锁
55     public boolean releaseLock(String uuid){
56         while(true){
57             // 添加监听 一旦LOCK_NAME发生变化
58             // 那么下面的事务有效
59             conn.watch(LOCK_NAME);
60             if(uuid.equals(conn.get(LOCK_NAME))){
61                 // 通过事务来操作
62                 Transaction transaction = conn.multi();
63                 transaction.del(LOCK_NAME);
64                 // 执行失败的时候会返回null
65                 if(transaction.exec() == null){
66                     continue;
67                 }
68                 // 执行成功
69                 return true;
70             }
71             // 取消监听
72             conn.unwatch();
73             break;
74         }
75         
76         return false;
77     }
78     
79     public static void main(String[] args)
80     {
81         // 单机测试环境下 可以采用手动去删除redis库中对应的LOCK_NAME
82         // 以便accequireLock可以获取到锁
83         SimpleLock simpleLock = new SimpleLock();
84         String uuid = simpleLock.accequireLock(10000);
85         if(null != uuid){
86             System.out.println("获取锁成功");
87         }else{
88             System.out.println("获取锁失败");
89         }
90     }
91 }