ChatGPT 用redis实现分布式锁

背景

打算基于springboot、RedisTemplate，由ChatGPT来实现分布式锁，记录全过程。

基础介绍

基于RedisTemplate实现分布式锁

基于springboot及RedisTemplate实现分布式锁
在Spring Boot中，我们可以通过RedisTemplate来使用Redis的分布式锁机制。下面是一个基于RedisTemplate的简单实现：

@Service
public class DistributedLockService {
    
    @Autowired
    private RedisTemplate<String, String> redisTemplate;
    
    public boolean acquire(String lockKey, String requestId, long expireTime) {
        // 使用setnx命令尝试获取锁
        Boolean result = redisTemplate.opsForValue().setIfAbsent(lockKey, requestId, expireTime, TimeUnit.SECONDS);
        return result != null && result;
    }
    
    public void release(String lockKey, String requestId) {
        String currentValue = redisTemplate.opsForValue().get(lockKey);
        // 如果当前锁的值与请求锁的值相等，则删除锁
        if (currentValue != null && currentValue.equals(requestId)) {
            redisTemplate.delete(lockKey);
        }
    }
    
}

上述代码中，我们通过RedisTemplate的opsForValue方法来获取一个ValueOperations对象，然后使用setIfAbsent方法来尝试获取锁。如果返回值为true，说明当前进程或线程成功获得了锁；否则说明其他进程或线程已经获得了锁。在释放锁时，我们通过get方法获取锁的值，如果该值与请求锁的值相等，则说明当前进程或线程持有锁，可以删除锁。由于获取锁和释放锁的操作都是原子的，因此可以避免死锁和竞争等问题。

需要注意的是，上述实现仅仅是一个简单的示例，实际应用中还需要考虑到锁的可重入性、锁的超时处理、锁的多进程竞争等问题，否则可能会导致锁无法正常释放或者产生其他问题。因此，在实际开发中，需要仔细考虑锁的实现细节，并进行充分测试。

基于springboottest写单元测试

好的，这里是一个基于SpringBootTest框架的单元测试示例，它测试了DistributedLockService的基本功能：

import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.data.redis.core.RedisTemplate;

import java.util.concurrent.TimeUnit;

import static org.junit.jupiter.api.Assertions.*;

@SpringBootTest
public class DistributedLockServiceTest {

    @Autowired
    private DistributedLockService lockService;

    @Autowired
    private RedisTemplate<String, String> redisTemplate;

    @Test
    public void testAcquireLock() {
        String lockKey = "test-lock";
        String requestId = "request-id";

        // 尝试获取锁
        boolean result = lockService.acquire(lockKey, requestId);

        // 验证结果
        assertTrue(result);
        assertNotNull(redisTemplate.opsForValue().get(lockKey));
    }

    @Test
    public void testReleaseLock() {
        String lockKey = "test-lock";
        String requestId = "request-id";

        // 获取锁
        redisTemplate.opsForValue().set(lockKey, requestId, 30, TimeUnit.SECONDS);

        // 尝试释放锁
        lockService.release(lockKey, requestId);

        // 验证结果
        assertNull(redisTemplate.opsForValue().get(lockKey));
    }

}

在上述测试中，我们使用了@SpringBootTest注解来启动Spring Boot上下文，并自动注入了DistributedLockService和RedisTemplate。然后，我们分别测试了acquire和release方法。在testAcquireLock方法中，我们尝试获取锁并验证了锁的值不为null。在testReleaseLock方法中，我们先获取了锁，然后尝试释放锁并验证了锁的值为null。注意，在testReleaseLock方法中，我们使用了redisTemplate.opsForValue().set(lockKey, requestId, 30, TimeUnit.SECONDS)来手动设置锁的过期时间，因为在DistributedLockService中并没有实现自动续期的功能。

说明

实现的分布式锁，没有去处理可重入、没有考虑锁的续期。

总结

个人体验总结，ChatGPT整体结果方向不差，细节容易出错，需要我们加入各种条件和引导，通过引导可以改正给出的结果，我觉得ChatGPT做得能够更好，或者从另一个方面讲，我们能够掌握方法更高效地去引导。