实现分布式锁的两个核心:
一、获取锁
1、获取锁线程互斥性
为了实现只有一个线程能继续执行业务代码,必须保证获取锁具有互斥性,即只有一个线程能获取到锁。
Redis中操作数据是单线程的,可以使用Redis提供的set nx ex命令获取锁。
spring底层封装set nx ex命令实现获取锁的互斥性。stringRedisTemplate.opsForValue().setIfAbsent(K key, V value, long timeout, TimeUnit unit)
二、释放锁
1、释放锁误删问题
必须释放当前线程加的锁。
场景:线程1阻塞,导致锁超时自动释放,线程2获取锁执行业务,这时线程1被唤醒执行释放锁操作,但线程1实际释放线程2的锁,导致线程3能获取到锁进而造成线程2和线程3的并发问题。
解决:获取锁时存入线程标识(uuid-threadid),基于线程标识释放锁。线程标识不能是线程id,因为不同服务的线程id可能相同。
2、释放锁的原子性问题
场景:线程1释放锁时先判断线程标识,还未释放锁,此时线程阻塞(网络或GC,GC时线程会阻塞),锁自动过期,线程1被唤醒后删除了其他线程的锁。
解决:lua脚本保证分布式锁原子性。判断和删除为同一原子操作,成功返回true失败返回false。
if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end
#参考文档
https://blog.csdn.net/qq_36602071/article/details/126002886
写死脚本方式
package hh.club.hisape.server.controller;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.core.script.DefaultRedisScript;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import java.util.Collections;
import java.util.UUID;
import java.util.concurrent.TimeUnit;
@Slf4j
@RequestMapping("/test")
@RestController
public class TestController {
@Autowired
private StringRedisTemplate redisTemplate;
@GetMapping("/testRedisson")
public void testRedisson() {
String uuid = UUID.randomUUID().toString().replaceAll("-", "");
Boolean bool = redisTemplate.opsForValue().setIfAbsent("lock:test", uuid + "-" + Thread.currentThread().getId(), 10, TimeUnit.SECONDS);
if (bool) {
System.out.println("获取锁成功");
String script = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end";
Long execute = redisTemplate.execute(new DefaultRedisScript<>(script, Long.class), Collections.singletonList("lock:test"), uuid + "-" + Thread.currentThread().getId());
if (1L == execute) {
System.out.println("释放锁成功");
} else {
System.out.println("释放锁失败");
}
} else {
System.out.println("获取锁失败");
}
}
}
读取配置文件lua脚本方式
package hh.club.hisape.server.controller;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.io.ClassPathResource;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.core.script.DefaultRedisScript;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import java.util.Collections;
import java.util.UUID;
import java.util.concurrent.TimeUnit;
@Slf4j
@RequestMapping("/test")
@RestController
public class TestController {
@Autowired
private StringRedisTemplate redisTemplate;
@GetMapping("/testRedisson")
public void testRedisson() {
String uuid = UUID.randomUUID().toString().replaceAll("-", "");
Boolean bool = redisTemplate.opsForValue().setIfAbsent("lock:test", uuid + "-" + Thread.currentThread().getId(), 10, TimeUnit.SECONDS);
if (bool) {
System.out.println("获取锁成功");
DefaultRedisScript<Long> defaultRedisScript = new DefaultRedisScript();
defaultRedisScript.setLocation(new ClassPathResource("unlock.lua"));
defaultRedisScript.setResultType(Long.class);
Long execute = redisTemplate.execute(defaultRedisScript, Collections.singletonList("lock:test"), uuid + "-" + Thread.currentThread().getId());
if (1L == execute) {
System.out.println("释放锁成功");
} else {
System.out.println("释放锁失败");
}
} else {
System.out.println("获取锁失败");
}
}
}