分布式锁实现(一)

逃离我推掉我的手 2022-05-30 07:17 145阅读 0赞

使用Redis的 SETNX 命令实现分布式锁。

# SETNX命令简介 #

## 命令格式 ##

> SETNX key value

将 key 的值设为 value，当且仅当 key 不存在。   
若给定的 key 已经存在，则 SETNX 不做任何动作。   
SETNX 是SET if Not eXists的简写。

## 返回值 ##

返回整数，具体为   
\- 1，当 key 的值被设置   
\- 0，当 key 的值没被设置

## 例子 ##

> redis> SETNX mykey “hello”   
> (integer) 1   
> redis> SETNX mykey “hello”   
> (integer) 0   
> redis> GET mykey   
> “hello”

Java代码实现

private static final String LOCK_SUCCESS = "OK";
    	private static final String SET_IF_NOT_EXIST = "NX";
    	private static final String SET_WITH_EXPIRE_TIME = "PX";
    	private static final Long RELEASE_SUCCESS = 1L;
    
    public boolean getRedisLock(String LockValue,int expireTime) {
        	Jedis jedis=null;
        	try {
        		jedis=pool_0.getResource();
    	        long now = System.currentTimeMillis();        
    	        while(true){
    	            if ((System.currentTimeMillis() - now) > expireTime)
    	            {
    	            	logger.error("获取锁超时:"+LockValue);
    	                now = System.currentTimeMillis();
    	                jedis.del(LockValue);
    	                return false;
    	            }            
    
    				String result = jedis.set(LockValue, LockValue, SET_IF_NOT_EXIST, SET_WITH_EXPIRE_TIME, expireTime);
    				if (LOCK_SUCCESS.equals(result)) {
    					break;
    				}
    	            try {
    					Thread.currentThread().sleep(1);
    				} catch (InterruptedException e) {
    					e.printStackTrace();
    				}
    	        }
        	} catch (Exception e) {
        		logger.error("getRedisLock is error:"+LockValue);
    		}finally{
    			if(jedis!=null){
    				jedis.close();
    			}
    		}
            return true;
        }
        
    
    public boolean releaseRedisLock(String LockValue) {
    		Jedis jedis = null;
    		try {
    			jedis = pool_0.getResource();
    			String script = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end";
    			Object result = jedis.eval(script, Collections.singletonList(LockValue), Collections.singletonList(LockValue));
    			if (RELEASE_SUCCESS.equals(result)) {
    				return true;
    			}
    		} catch (Exception e) {
    			logger.error("getRedisLock is error",e);
    		} finally {
    			if (jedis != null) {
    				jedis.close();
    			}
    		}
    		return false;
        }
    
    
    
    public static void main(String[] args) {
    		try {
    			//获取分布式锁
    			if(redisUtil.getRedisLock("a_lock",10000)){
    				//业务
    			}
    		} catch (Exception e) {
    			e.printStackTrace();
    		}finally{
    			redisUtil.releaseRedisLock("a_lock");
    		}
    	}

二、基于Redis实现的另一种分布式锁实现（redssion）地址：[http://blog.csdn.net/wwd0501/article/details/50717111][http_blog.csdn.net_wwd0501_article_details_50717111]

[http_blog.csdn.net_wwd0501_article_details_50717111]: http://blog.csdn.net/wwd0501/article/details/50717111