Redis实现高性能分布式锁
分布式系统中,锁的使用是非常常见的。但是传统单机锁在分布式环境下往往不能满足需求,因此需要实现分布式锁。Redis作为一种高性能、可扩展的Key-Value存储服务,被广泛用于实现分布式锁。
实现思路
Redis提供了一个原子性操作SETNX,支持在Key不存在时设置Key的值,返回1表示设置成功,0表示Key已存在。可以利用SETNX实现分布式锁,即将某个Key设置为一个唯一标识Value,如果SETNX返回1,说明获取锁成功,其他节点不能再对该Key加锁;如果返回0,说明有其他节点已经获取了锁,当前节点需要等待一段时间后重新尝试加锁。
在释放锁时,需要用到另一个Redis原子性操作DEL,支持删除指定Key。
在加锁和释放锁时,需要确保线程安全。可以使用Redis的Lua脚本功能,将多个Redis原子性操作打包成一个Lua脚本,保证在执行过程中不会被其它线程打断。
代码实现
下面是Java语言中使用Redis实现分布式锁的示例代码:
import redis.clients.jedis.Jedis;import redis.clients.jedis.JedisPool;import redis.clients.jedis.JedisPoolConfig;public class RedisLock { private JedisPool jedisPool; public RedisLock(String host, int port, String password) { JedisPoolConfig config = new JedisPoolConfig(); config.setMaxTotal(100); config.setMaxIdle(20); config.setMinIdle(10); config.setMaxWtMillis(10000); config.setTestOnBorrow(true); jedisPool = new JedisPool(config, host, port, 10000, password); } /** * 获取锁 * * @param key 锁关键字 * @param expireTime 锁失效时间,单位:毫秒 * @param retryInterval 尝试获取锁的间隔时间,单位:毫秒 * @param retryTimes 尝试获取锁的最大次数 * @return 如果获取锁成功,返回true;如果获取锁失败,返回false */ public boolean acquireLock(String key, long expireTime, long retryInterval, int retryTimes) { try (Jedis jedis = jedisPool.getResource()) { String value = Thread.currentThread().getId() + ""; int retries = 0; while (retries long result = jedis.setnx(key, value); if (result == 1) { // 获取锁成功 jedis.expire(key, (int)(expireTime / 1000)); return true; } // 获取锁失败,等待一段时间后重新尝试 Thread.sleep(retryInterval); retries++; } return false; } catch (Exception e) { e.printStackTrace(); return false; } } /** * 释放锁 * * @param key 锁关键字 * @return 如果释放锁成功,返回true;如果释放锁失败,返回false */ public boolean releaseLock(String key) { try (Jedis jedis = jedisPool.getResource()) { jedis.del(key); return true; } catch (Exception e) { e.printStackTrace(); return false; } }}
在上面的代码中,我们利用了Java 7之后的try-with-resource语法糖,确保Jedis对象在使用完毕后及时关闭,释放连接资源。
使用示例
下面是使用RedisLock类实现分布式锁的示例代码:
public class TestRedisLock { private static final String REDIS_HOST = "localhost"; private static final int REDIS_PORT = 6379; private static final String REDIS_PASSWORD = null; public static void mn(String[] args) throws InterruptedException { RedisLock lock = new RedisLock(REDIS_HOST, REDIS_PORT, REDIS_PASSWORD); String lockKey = "testLock"; long expireTime = 5000; // 锁失效时间:5秒 long retryInterval = 100; // 尝试获取锁的间隔时间:100毫秒 int retryTimes = 10; // 尝试获取锁的最大次数:10次 // 尝试获取锁 boolean acquired = lock.acquireLock(lockKey, expireTime, retryInterval, retryTimes); if (acquired) { // 获取锁成功 System.out.println(Thread.currentThread().getName() + " acquired lock, do something..."); Thread.sleep(expireTime / 2); // 模拟执行业务逻辑 // 释放锁 boolean released = lock.releaseLock(lockKey); if (released) { // 释放锁成功 System.out.println(Thread.currentThread().getName() + " released lock"); } else { // 释放锁失败 System.out.println(Thread.currentThread().getName() + " fled to release lock"); } } else { // 获取锁失败 System.out.println(Thread.currentThread().getName() + " fled to acquire lock"); } }}
在上面的代码中,我们创建了一个RedisLock对象,调用它的acquireLock方法获取锁,在获取锁成功后执行一些业务逻辑,最后调用releaseLock方法释放锁。
注意,由于实现分布式锁涉及到Redis的原子性操作,因此Redis服务器的性能和网络带宽会成为瓶颈。在高并发场景下建议使用Redis集群,甚至使用多个Redis集群实现读写分离、容错等功能。同时,需要注意设置合适的锁失效时间、尝试获取锁的最大次数等参数,以便实现高可用的分布式锁。
香港服务器首选,2H2G首月10元开通。()提供简单好用,价格厚道的香港/美国云服务器和独立服务器。IDC+ISP+ICP资质。ARIN和APNIC会员。成熟技术团队15年行业经验。