【分布式锁】基于Redisson的分布式锁实现

本文主要是介绍【分布式锁】基于Redisson的分布式锁实现，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

以下代码是基于Redisson的分布式锁实现

yml文件中配置redis

# Spring配置
spring:# redis 配置redis:host: 127.0.0.1database: 1password:port: 6379address: redis://127.0.0.1:6379

编写配置类，注册redisClinet


/*** @description Redisson配置*/
@Configuration
public class RedissonConfig {@Value("${spring.redis.address}")private String redisAddress;@Value("${spring.redis.password}")private String password;@Value("${spring.redis.database}")private int database;@Bean(name = "redissonClient")public RedissonClient singleRedissonClient() {Config config = new Config();SingleServerConfig singleServerConfig = config.useSingleServer();singleServerConfig.setAddress(redisAddress).setDatabase(database);if (!StringUtils.isEmpty(password)) {singleServerConfig.setPassword(password);}return Redisson.create(config);}
}

编写分布式锁-接口方便扩展

import com.ruoyi.common.lock.DistributedLock;/*** @description 分布式锁工程接口*/
public interface DistributedLockFactory {/*** 根据key获取分布式锁*/DistributedLock getDistributedLock(String key);
}

编写实现类

import com.ruoyi.common.lock.DistributedLock;
import com.ruoyi.common.lock.factory.DistributedLockFactory;
import org.redisson.api.RLock;
import org.redisson.api.RedissonClient;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;import javax.annotation.Resource;
import java.util.concurrent.TimeUnit;/*** @description 基于Redisson的分布式锁实现*/
@Component
public class RedissonLockFactory implements DistributedLockFactory {private final Logger logger = LoggerFactory.getLogger(RedissonLockFactory.class);@Autowiredprivate RedissonClient redissonClient;@Overridepublic DistributedLock getDistributedLock(String key) {RLock rLock = redissonClient.getLock(key);return new DistributedLock() {@Overridepublic boolean tryLock(long waitTime, long leaseTime, TimeUnit unit) throws InterruptedException {boolean isLockSuccess = rLock.tryLock(waitTime, leaseTime, unit);logger.info("waitTime-leaseTime | {} get lock result:{}", key, isLockSuccess);return isLockSuccess;}@Overridepublic boolean tryLock(long waitTime, TimeUnit unit) throws InterruptedException {boolean b = rLock.tryLock(waitTime, unit);logger.info("waitTime | {} get lock result:{}", key, b);return b;}@Overridepublic boolean tryLock() throws InterruptedException {return rLock.tryLock();}@Overridepublic void lock(long leaseTime, TimeUnit unit) {rLock.lock(leaseTime, unit);}@Overridepublic void unlock() {if (isLocked() && isHeldByCurrentThread()) {rLock.unlock();}}@Overridepublic boolean isLocked() {return rLock.isLocked();}@Overridepublic boolean isHeldByThread(long threadId) {return rLock.isHeldByThread(threadId);}@Overridepublic boolean isHeldByCurrentThread() {return rLock.isHeldByCurrentThread();}};}
}

用法

DistributedLock distributedLock = distributedLockFactory.getDistributedLock(String.format("invoice_methods_lock_key_" + file.getName()));boolean b = distributedLock.tryLock(5, TimeUnit.SECONDS);if(b){//执行具体逻辑}
//释放锁distributedLock.unlock();