【spring boot】【redis】spring boot基于redis的LUA脚本 实现分布式锁

spring boot基于redis的LUA脚本 实现分布式锁【都是基于redis单点下】

一.spring boot 1.5.X 基于redis 的 lua脚本实现分布式锁

1.pom.xml

org.springframework.boot

spring-boot-starter-data-redis

2.RedisLock 工具类 （注入spring）

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.data.redis.core.RedisTemplate;

import org.springframework.data.redis.core.script.DefaultRedisScript;

import org.springframework.data.redis.core.script.RedisScript;

import org.springframework.data.redis.serializer.StringRedisSerializer;

import org.springframework.stereotype.Component;

import java.util.Collections;

/**

* spring boot 1.5.X

* 使用redis 的 lua脚本 基于单点实现分布式锁

*

* lua脚本作为原子性操作，保证加锁和设置超时时间 为原子性操作

* @author sxd

* @date 2019/5/27 10:52

*/

@Component

public class RedisLock {

@Autowired

RedisTemplate redisTemplate;

private static final Long SUCCESS = 1L;

/**

* 获取锁

*

* @param lockKey redis的key

* @param value redis的value要求是随机串，防止释放其他请求的锁

* @param expireTime redis的key 的过期时间 防止死锁，导致其他请求无法正常执行业务

* @return

*/

public boolean lock(String lockKey, String value, int expireTime) {

String script = "if redis.call('setNx',KEYS[1],ARGV[1]) then " +

" if redis.call('get',KEYS[1])==ARGV[1] then " +

" return redis.call('expire',KEYS[1],ARGV[2]) " +

" else " +

" return 0 " +

" end " +

"end";

RedisScript redisScript = new DefaultRedisScript<>(script, String.class);

//对非string类型的序列化

redisTemplate.setKeySerializer(new StringRedisSerializer());

redisTemplate.setValueSerializer(new StringRedisSerializer());

Object result = redisTemplate.execute(redisScript, Collections.singletonList(lockKey), value, String.valueOf(expireTime));

return SUCCESS.equals(result);

}

/**

* 释放锁

*

* @param lockKey redis的key

* @param value redis的value 只有value比对一致，才能确定是本请求 加的锁 才能正常释放

* @return

*/

public boolean unlock(String lockKey, String value) {

String script = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end";

RedisScript redisScript = new DefaultRedisScript<>(script, String.class);

try {

Object result = redisTemplate.execute(redisScript, Collections.singletonList(lockKey), value);

if (SUCCESS.equals(result)) {

return true;

}

} catch (Exception e) {

e.printStackTrace();

}

return false;

}

}

View Code

3.controller使用

/**

* 使用分布式锁 逻辑

* 1.准备好 key value expireTime

* value要求是随机字符串

* expireTime 是根据业务 衡量决定的 锁过期时间

*

* 2.获取锁

* 成功获取，则执行业务，执行完成，释放锁

* 失败获取，则重试获取，注意获取锁的时间间隔，直到获取成功，执行业务，最后释放锁

*

* 注意：

* 对于redis加锁的业务，尽量用在耗时短的业务上。

*

*/

@RequestMapping("/test")

public void test(){

boolean flag = false; //标识 是否正常获取锁

String uuid = UUID.randomUUID().toString(); //redis的value 是一串随机数

flag = lock.lock("mykey1",uuid,5);

if (flag){

business(uuid);

}else {

//如果未正常获取锁 可以通过重试 直到获取锁成功

while (!flag){

try {

//重试 时间间隔 减少与redis交互次数

Thread.sleep(3000);

System.out.println("重试");

flag = lock.lock("mykey1",uuid,5);

if (flag){

business(uuid);

}else {

continue;

}

} catch (InterruptedException e) {

e.printStackTrace();

}

}

}

}

public void business(String uuid){

try {

System.out.println("加锁成功，执行业务");

Thread.sleep(10000);

} catch (InterruptedException e) {

e.printStackTrace();

}finally {

//业务执行完成 正常释放锁

lock.unlock("mykey1",uuid);

}

}

View Code

二.spring boot 2.x 基于redis 的LUA脚本 实现分布式锁

1.pom.xml

org.springframework.boot

spring-boot-starter-data-redis

org.apache.commons

commons-pool2

2.4.2

com.fasterxml.jackson.core

jackson-core

com.fasterxml.jackson.core

jackson-annotations

com.fasterxml.jackson.core

jackson-databind

2.替代SpringBoot自动配置的RedisTemplate的RedisConfig类

package com.sxd.swapping.config;

import com.fasterxml.jackson.annotation.JsonAutoDetect;

import com.fasterxml.jackson.annotation.PropertyAccessor;

import com.fasterxml.jackson.databind.ObjectMapper;

import org.springframework.boot.autoconfigure.AutoConfigureAfter;

import org.springframework.boot.autoconfigure.data.redis.RedisAutoConfiguration;

import org.springframework.context.annotation.Bean;

import org.springframework.context.annotation.Configuration;

import org.springframework.data.redis.connection.RedisConnectionFactory;

import org.springframework.data.redis.core.RedisTemplate;

import org.springframework.data.redis.serializer.Jackson2JsonRedisSerializer;

import org.springframework.data.redis.serializer.StringRedisSerializer;

import java.io.Serializable;

/**

* @author sxd

* @date 2019/5/27 16:13

*/

/**

* @Description Redis配置类，替代SpringBoot自动配置的RedisTemplate，参加RedisAutoConfiguration

*/

@Configuration

@AutoConfigureAfter(RedisAutoConfiguration.class)

public class RedisConfig {

@Bean

public RedisTemplate redisTemplate(RedisConnectionFactory redisConnectionFactory) {

RedisTemplate template = new RedisTemplate<>();

template.setConnectionFactory(redisConnectionFactory);

//Jackson序列化器

Jackson2JsonRedisSerializer