Redis 分布式锁的使用

ゝ一纸荒年。 2022-12-20 03:18 124阅读 0赞

最近有个项目，需要一个生成唯一id的功能，也不是很高的并发，所以使用redis分布式锁来完成。  
具体过程：  
获取id时先从redis获取锁，成功，分配id；失败，继续等待，直到分配完成。  
下面主要讲一下redis分布式锁的使用的关键代码

**springboot2.x 以上使用redis时，默认使用了lettuce封装，比起jedis线程安全**

### 配置 ###

spring.redis.timeout=30000
    spring.redis.host=10.150.150.23
    ##这里修改了默认端口
    spring.redis.port= 6330
    ##这里使用了jasypt插件对密码进行安全处理
    spring.redis.password= ENC(dkN55MpZgNe2zriGEW30hoNpKmX27RZ7)
    spring.redis.database=3
    spring.redis.lettuce.pool.max-idle=8
    spring.redis.lettuce.pool.min-idle=1
    spring.redis.lettuce.pool.max-wait=30000
    spring.redis.lettuce.pool.max-active=8

### 主要redis接口代码 ###

import lombok.extern.slf4j.Slf4j;
    import org.springframework.data.redis.connection.RedisStringCommands;
    import org.springframework.data.redis.connection.ReturnType;
    import org.springframework.data.redis.core.RedisCallback;
    import org.springframework.data.redis.core.RedisTemplate;
    import org.springframework.data.redis.core.types.Expiration;
    import org.springframework.stereotype.Component;
     
    import javax.annotation.Resource;
    import java.nio.charset.Charset;
    import java.util.concurrent.TimeUnit;
     
    /**
     * @Description: 分布式锁工具类
     * @Author:
     * @CreateDate: 
     */
    @Component
    @Slf4j
    public class RedisLock {
        @Resource
        private RedisTemplate redisTemplate;
     
        public static final String UNLOCK_LUA;
     
        /**
         * 释放锁脚本，原子操作
         */
        static {
            StringBuilder sb = new StringBuilder();
            sb.append("if redis.call(\"get\",KEYS[1]) == ARGV[1] ");
            sb.append("then ");
            sb.append("    return redis.call(\"del\",KEYS[1]) ");
            sb.append("else ");
            sb.append("    return 0 ");
            sb.append("end ");
            UNLOCK_LUA = sb.toString();
        }
     
     
        /**
         * 获取分布式锁，原子操作
         * @param lockKey
         * @param requestId 唯一ID, 可以使用UUID.randomUUID().toString();
         * @param expire
         * @param timeUnit
         * @return
         */
        public boolean tryLock(String lockKey, String requestId, long expire, TimeUnit timeUnit) {
            try{
                RedisCallback<Boolean> callback = (connection) -> {
                    return connection.set(lockKey.getBytes(Charset.forName("UTF-8")), requestId.getBytes(Charset.forName("UTF-8")), Expiration.seconds(timeUnit.toSeconds(expire)), RedisStringCommands.SetOption.SET_IF_ABSENT);
                };
                return (Boolean)redisTemplate.execute(callback);
            } catch (Exception e) {
                log.error("redis lock error.", e);
            }
            return false;
        }
     
        /**
         * 释放锁
         * @param lockKey
         * @param requestId 唯一ID
         * @return
         */
        public boolean releaseLock(String lockKey, String requestId) {
            RedisCallback<Boolean> callback = (connection) -> {
                return connection.eval(UNLOCK_LUA.getBytes(), ReturnType.BOOLEAN ,1, lockKey.getBytes(Charset.forName("UTF-8")), requestId.getBytes(Charset.forName("UTF-8")));
            };
            return (Boolean)redisTemplate.execute(callback);
        }
     
        /**
         * 获取Redis锁的value值
         * @param lockKey
         * @return
         */
        public String get(String lockKey) {
            try {
                RedisCallback<String> callback = (connection) -> {
                    return new String(connection.get(lockKey.getBytes()), Charset.forName("UTF-8"));
                };
                return (String)redisTemplate.execute(callback);
            } catch (Exception e) {
                log.error("get redis occurred an exception", e);
            }
            return null;
        }
     
    }

### 业务代码 ###

String testno = "";
    String lockKey = "testLock";
    Date datekey = new Date();
    SimpleDateFormat simpleDateFormat = new SimpleDateFormat("yyyy-MM-dd");
    String testKey = simpleDateFormat.format(datekey);
    String requestID = UUID.randomUUID().toString();
    #等待十秒
    while (!RedisUtil.tryLock(lockKey, requestID,10, TimeUnit.SECONDS)){
        ;
    }
    Integer testval = RedisUtil.getInt(testKey);
    if (testval == null || testval.intValue() == 0){
        RedisUtil.setInt(testval, 100001, 60*60*24);
        testno = "100001";
    }else{
        Integer old_key = RedisUtil.getInt(testval);
        old_key += 1;
        //设置有效期一天
        RedisUtil.setInt(testval, old_key, 60*60*24);
        testno = String.valueOf(old_key);
    }
    
    RedisUtil.releaseLock(testKey, requestID);
    RedisUtil.setInt(token_key, 1, 60*60*24);

## 使用jmeter测试结果 ##

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3UwMTM5MTkxNTM_size_16_color_FFFFFF_t_70_pic_center]  
本文参考文章：  
https://www.cnblogs.com/x-j-p/p/13322022.html  
https://blog.csdn.net/xiehd313/article/details/86495256

在此表示感谢！

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3UwMTM5MTkxNTM_size_16_color_FFFFFF_t_70_pic_center]: /images/20221120/4e1e7663e6254d73b2980192b3d635c0.png