使用redis实现分布式锁

基于Redis的分布式锁

使用命令

SETNX

SETNX key val当且仅当key不存在时，set一个key为val的字符串，返回1；若key存在，则什么都不做，返回0。

Expire

expire key timeout

为key设置一个超时时间，单位为second，超过这个时间锁会自动释放，避免死锁。

Delete

delete key 删除key

实现思路

1.获取锁的时候，使用setnx加锁，并使用expire命令为锁添加一个超时时间，超过该时间则自动释放锁，锁的value值为一个随机生成的UUID，通过此在释放锁的时候进行判断。

2.获取锁的时候还设置一个获取的超时时间，若超过这个时间则放弃获取锁。

3.释放锁的时候，通过UUID判断是不是该锁，若是该锁，则执行delete进行锁释放。

核心代码

public class LockRedis {

	private JedisPool jedisPool;

	public LockRedis(JedisPool jedisPool) {
		this.jedisPool = jedisPool;
	}

	/**
	 * redis 上锁方法 
	 * @param lockKey
	 *            锁的key<br>
	 * @param acquireTimeout
	 *            在没有上锁之前,获取锁的超时时间<br>
	 * @param timeOut
	 *            上锁成功后,锁的超时时间<br>
	 * @return
	 */
	public String lockWithTimeout(String lockKey, Long acquireTimeout, Long timeOut) {
		Jedis conn = null;
		String retIdentifierValue = null;
		try {
			// 1.建立redis连接
			conn = jedisPool.getResource();
			// 2.随机生成一个value
			String identifierValue = UUID.randomUUID().toString();
			// 3.定义锁的名称
			String lockName = "redis_lock" + lockKey;
			// 4.定义上锁成功之后,锁的超时时间
			int expireLock = (int) (timeOut / 1000);
			// 5.定义在没有获取锁之前,锁的超时时间
			Long endTime = System.currentTimeMillis() + acquireTimeout;
			while (System.currentTimeMillis() < endTime) {
				// 6.使用setnx方法设置锁值
				if (conn.setnx(lockName, identifierValue) == 1) {
					// 7.判断返回结果如果为1,则可以成功获取锁,并且设置锁的超时时间
					conn.expire(lockName, expireLock);
					retIdentifierValue = identifierValue;
					return retIdentifierValue;
				}
				// 8.否则情况下继续循环等待
			}

		} catch (Exception e) {
			e.printStackTrace();
		} finally {
			if (conn != null) {
				conn.close();
			}
		}
		return retIdentifierValue;
	}

	/**
	 * 释放锁
	 * 
	 * @return
	 */
	public boolean releaseLock(String lockKey, String identifier) {

		Jedis conn = null;
		boolean flag = false;
		try {
			// 1.建立redis连接
			conn = jedisPool.getResource();
			// 2.定义锁的名称
			String lockName = "redis_lock" + lockKey;
			// 3.如果value与redis中一致直接删除，否则等待超时
			if (identifier.equals(conn.get(lockName))) {
				conn.del(lockName);
				System.out.println(identifier + "解锁成功......");
			}
		} catch (Exception e) {
			e.printStackTrace();
		} finally {
			if (conn != null) {
				conn.close();
			}
		}
		return flag;
	}
}



public class LockService {
	private static JedisPool pool = null;
	static {
		JedisPoolConfig config = new JedisPoolConfig();
		// 设置最大连接数
		config.setMaxTotal(200);
		// 设置最大空闲数
		config.setMaxIdle(8);
		// 设置最大等待时间
		config.setMaxWaitMillis(1000 * 100);
		// 在borrow一个jedis实例时，是否需要验证，若为true，则所有jedis实例均是可用的
		config.setTestOnBorrow(true);
		pool = new JedisPool(config, "39.107.69.43", 6379, 3000);
	}

	LockRedis lockRedis = new LockRedis(pool);

	public void seckill() {
		String identifier = lockRedis.lockWithTimeout("wangcj", 5000l, 5000l);
		if (StringUtils.isEmpty(identifier)) {
			// 获取锁失败
			System.out.println(Thread.currentThread().getName() + ",获取锁失败，原因时间超时!!!");
			return;
		}
		System.out.println(Thread.currentThread().getName() + "获取锁成功,锁id identifier:" + identifier + "，执行业务逻辑");
		try {
			Thread.sleep(30);
		} catch (Exception e) {

		}
		// 释放锁
		boolean releaseLock = lockRedis.releaseLock("wangcj", identifier);
		if (releaseLock) {
			System.out.println(Thread.currentThread().getName() + "释放锁成功,锁id identifier:" + identifier);
		}
	}
}
class ThreadRedis extends Thread {
	private LockService lockService;

	public ThreadRedis(LockService lockService) {
		this.lockService = lockService;
	}

	@Override
	public void run() {
		lockService.seckill();

	}

}

public class Test001 {

	public static void main(String[] args) {
		LockService lockService = new LockService();
		for (int i = 0; i < 50; i++) {
			ThreadRedis threadRedis = new ThreadRedis(lockService);
			threadRedis.start();
		}
	}

}

Redis实现分布式锁与Zookeeper实现分布式锁区别

使用redis实现分布式锁

redis中的set  nx 命令，当key不存在时，才能在redis中将key添加成功，利用该属性可以实现分布式锁，并且redis对于key有失效时间，可以控制当某个客户端加锁成功之后挂掉，导致阻塞的问题。

使用Zookeeper实现分布式锁

多个客户端在Zookeeper上创建一个相同的临时节点，因为临时节点只能允许一个客户端创建成功，那么只要任意一个客户端创建节点成功，谁就成功的获取到锁，当释放锁后，其他客户端同样道理在Zookeeper节点。zookeeper关闭会话，临时节点就自动删除，所以比redis更加可靠