原理:
执行时获得指定名称key是否存在,如果存在不能获得key,如果不存在为该key赋值,然后设置失效时间;
解锁的逻辑即删除该key。
1.自定义类
package com.xxx.cloud.assets.redis.lock;
import java.util.Collections;
import java.util.UUID;
import java.util.concurrent.TimeUnit;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.script.DefaultRedisScript;
/**
* 每次使用时,要new一个新的lock使用
*/
public class RedisLock {
// 锁名称
private String lockName;
// 锁超时时间
private long time;
private static final String KEY_PREFIX = "BusinessName:lock:";
private long randomValue;
private DefaultRedisScript redisScript;
private RedisTemplate redisTemplate;
private boolean locked;
private RedisLock(String lockName, long time, RedisTemplate redisTemplate, DefaultRedisScript redisScript) {
this.lockName = KEY_PREFIX + lockName;
this.time = time;
this.redisTemplate = redisTemplate;
this.redisScript = redisScript;
this.randomValue = UUID.randomUUID().getLeastSignificantBits();
}
public static RedisLock NewRedisLock(String lockName, long time, RedisTemplate redisTemplate,
DefaultRedisScript redisScript) {
return new RedisLock(lockName, time, redisTemplate, redisScript);
}
public boolean tryLock() {
boolean success = redisTemplate.opsForValue().setIfAbsent(lockName, randomValue);
// setIfAbsent对应redis的setNX方法
locked = success;
if (success) {
redisTemplate.expire(lockName, time, TimeUnit.SECONDS);
}
return success;
}
public void unLock() {
if(locked){
redisTemplate.execute(redisScript, Collections.singletonList(lockName), randomValue);
}
}
}
2.测试代码
package com.xxx.cloud.assets.service;
import com.xxx.cloud.assets.redis.lock.RedisLock;
import com.xxx.cloud.assets.test.TestBase;
import org.junit.Test;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.script.DefaultRedisScript;
import javax.annotation.Resource;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.TimeUnit;
public class TestRedisLock extends TestBase {
@Resource(name = "redisTemplate")
private RedisTemplate redisTemplate;
@Resource(name = "dellockScript")
private DefaultRedisScript dellockScript;
@Test
public void testRedisLock() throws InterruptedException {
RedisLock lock = RedisLock.NewRedisLock("test",1000,redisTemplate,dellockScript);
try {
boolean success = lock.tryLock();
System.out.println(success);
}finally {
lock.unLock();
}
}
@Test
public void testMulRedisLock() throws InterruptedException {
final RedisLock lock1 = RedisLock.NewRedisLock("test",1000,redisTemplate,dellockScript);
final RedisLock lock2 = RedisLock.NewRedisLock("test",1000,redisTemplate,dellockScript);
Runnable task1 = new Runnable() {
@Override public void run() {
try {
boolean success = lock1.tryLock();
System.out.println("task1" + success);
TimeUnit.SECONDS.sleep(10);
}catch (Exception e){
e.printStackTrace();
}
finally {
lock1.unLock();
}
}
};
Runnable task2 = new Runnable() {
@Override public void run() {
try {
boolean success = lock2.tryLock();
System.out.println("task2"+success);
TimeUnit.SECONDS.sleep(10);
}catch (Exception e){
e.printStackTrace();
}
finally {
lock2.unLock();
}
}
};
ExecutorService executorService = Executors.newCachedThreadPool();
executorService.submit(task1);
executorService.submit(task2);
TimeUnit.SECONDS.sleep(15);
}
}
3.spring中部分配置
<bean id="redisTemplate" class="org.springframework.data.redis.core.RedisTemplate">
<property name="connectionFactory" ref="jedisConnectionFactory" />
<property name="defaultSerializer">
<bean class="org.springframework.data.redis.serializer.StringRedisSerializer" />
</property>
</bean>
<bean id="dellockScript" class="org.springframework.data.redis.core.script.DefaultRedisScript">
<property name="location" value="classpath:database/dellock.lua" />
<property name="resultType" value="java.lang.Boolean" />
</bean>
4.dellock.lua脚本(也可用代码逻辑替代)
if redis.call("get",KEYS[1]) == ARGV[1] then
return redis.call("del",KEYS[1])
else
return 0
end