redisTemplate用setNX命令实现分布式锁

项目中需要使用 分布式锁，而redis则是分布式锁的一大优选项；而setNX 命令则更为合适，阅读了不少博客，最后将自己时间的代码copy出来，以做记录。

package org.jeecg.common.util;

import java.util.Objects;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisCallback;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;

/**
 * 
 * @ClassName: RedisLockUril
 * @Description: 使用redis做锁
 * @author WangJing
 * @date 2020年6月8日
 *
 */
@Component
public class RedisLockUtil {

    @Autowired
    RedisTemplate redisTemplate;

    /**
     * 任务锁前缀
     */
    public static final String TASK_LOCK_PREFIX = "t_l_prefix";
    /**
     * 任务锁过期时间 2000毫秒
     */
    public static final Integer TASK_LOCK_EXPIRE = 2000;

    /**
     * 获取锁，true 则得到锁，false 已被锁定
     * 
     * @param lockName   锁名称
     * @param lockExoire 锁时间
     * @return
     */
    public Boolean getLock(String lockName, Integer lockExoire) {
        return (Boolean) redisTemplate.execute((RedisCallback) connection -> {
            // 获取时间毫秒值
            long expireAt = System.currentTimeMillis() + lockExoire + 1;
            // 获取锁
            Boolean acquire = connection.setNX(lockName.getBytes(), String.valueOf(expireAt).getBytes());
            if (acquire) {
                return true;
            } else {
                byte[] bytes = connection.get(lockName.getBytes());
                // 非空判断
                if (Objects.nonNull(bytes) && bytes.length > 0) {
                    long expireTime = Long.parseLong(new String(bytes));
                    // 如果锁已经过期
                    if (expireTime < System.currentTimeMillis()) {
                        // 重新加锁，防止死锁
                        byte[] set = connection.getSet(lockName.getBytes(),
                                String.valueOf(System.currentTimeMillis() + lockExoire + 1).getBytes());
                        return Long.parseLong(new String(set)) < System.currentTimeMillis();
                    }
                }
            }
            return false;
        });
    }

    /**
     * 删除锁
     * 
     * @param lockName 锁名称
     */
    public void delLock(String lockName) {
        redisTemplate.delete(lockName);
    }

    /**
     * 获取锁key
     * 
     * @param prefix 前缀
     * @param name   名称
     * @return
     */
    public static String getFullKey(String prefix, String name) {
        return prefix + "_" + name;
    }
}
 

 注：以上内容仅提供参考和交流，请勿用于商业用途，如有侵权联系本人删除！