java 集群分布式锁

原创

mob64ca12e83232 2024-11-21 03:45:25 ©著作权

文章标签 分布式锁 Redis 数据库 文章分类 Java 后端开发

©著作权归作者所有：来自51CTO博客作者mob64ca12e83232的原创作品，请联系作者获取转载授权，否则将追究法律责任

Java 集群分布式锁的概述与实现

在现代微服务架构中，随着系统的复杂性增加，确保数据的一致性和安全性变得尤为重要。尤其是在处理并发请求时，分布式锁成为解决多线程访问冲突的重要工具。本文将对 Java 集群分布式锁进行详细分析，并提供相应的代码示例。

什么是分布式锁？

分布式锁是一种在多个服务实例之间控制对共享资源访问的机制。它防止了由于多个服务实例同时操作共享资源而导致的数据不一致性。分布式锁通常基于某种持久化存储系统，如 Redis、ZooKeeper 或数据库等实现。

分布式锁的实现原理

分布式锁的基本原理是获取锁的实例在整个集群中是唯一的，获取锁的请求必须在一定时间内完成，否则则会释放锁以防止死锁。常用的实现方式包括：

基于数据库：数据库的行锁或表锁可以用来实现分布式锁。
基于 Redis：利用 Redis 的 SETNX 命令可以实现一个简单的分布式锁。
基于 ZooKeeper：利用 ZooKeeper 的临时顺序节点来实现分布式锁。

Redis 实现分布式锁的原理

Redis 提供的 SETNX 可以在键不存在的情况下设置一个值，成功返回1，失败返回0。这意味着我们可以通过 SETNX 命令来尝试获取锁。

SETNX lock_key unique_value

如果获取锁成功，还需设置一个过期时间以防止因意外情况导致锁永久被持有。

示例代码

使用 Redis 实现分布式锁

为了实现分布式锁，我们可以使用 Jedis 客户端与 Redis 进行交互。下面是一个简单的分布式锁实现示例：

import redis.clients.jedis.Jedis;
import java.util.Collections;

public class RedisDistributedLock {
    private final Jedis jedis;
    private static final String LOCK_SCRIPT = "if redis.call('setnx', KEYS[1], ARGV[1]) == 1 then return true else return false end";
    private static final long LOCK_EXPIRE_TIME = 5000; // 5 seconds

    public RedisDistributedLock(Jedis jedis) {
        this.jedis = jedis;
    }

    public boolean acquireLock(String lockKey, String uniqueValue) {
        return (Boolean) jedis.eval（LOCK_SCRIPT, Collections.singletonList(lockKey), Collections.singletonList(uniqueValue));
    }

    public void releaseLock(String lockKey, String uniqueValue) {
        String script = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end";
        jedis.eval（script, Collections.singletonList(lockKey), Collections.singletonList(uniqueValue));
    }
}

使用示例

下面是如何使用上述 RedisDistributedLock 类的示例代码：

public class Main {
    public static void main(String[] args) {
        Jedis jedis = new Jedis("localhost");
        RedisDistributedLock lock = new RedisDistributedLock(jedis);
        String lockKey = "my_lock";
        String uniqueValue = "unique_identifier";

        // 尝试获取锁
        if (lock.acquireLock(lockKey, uniqueValue)) {
            try {
                // 处理共享资源
                System.out.println("Lock acquired. Processing...");
                // 模拟处理耗时操作
                Thread.sleep(3000);
            } catch (InterruptedException e) {
                e.printStackTrace();
            } finally {
                lock.releaseLock(lockKey, uniqueValue);
                System.out.println("Lock released.");
            }
        } else {
            System.out.println("Failed to acquire lock. Resource is busy.");
        }
    }
}

状态图

为了更直观地描述分布式锁的工作流程，我们可以使用状态图来表示。在下图中，LockState 描述不同的状态及其转换。

stateDiagram
    [*] --> Unlocking
    Unlocking --> Locked: acquireLock
    Locked --> Unlocking: releaseLock
    Locked --> Locked: operation

表格展示

下表展示了不同分布式锁实现方式的对比：

实现方式	优点	缺点
数据库锁	简单易实现，可靠性高	性能较差，容易造成数据库瓶颈
Redis 锁	性能优秀，支持高并发	依赖单点 Redis 可能导致锁失效
ZooKeeper锁	支持强一致性，适合分布式系统	使用复杂，性能相对较低