实现分布式锁

实现分布式锁的两个核心:
一、获取锁
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("获取锁失败");
        }
    }
}

  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值