Redis分布式锁

一、Redis分布式加锁与解锁

import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Component;

/**
 * 分布式锁
 */
@Component
@Slf4j
public class RedisLock {

    @Autowired
    private StringRedisTemplate redisTemplate;

    /**
     * 加锁
     *
     * @param key   商品ID
     * @param value 当前时间 + 超时时间
     * @return
     */
    public boolean lock(String key, String value) {
        /** 如果key不存在，返回true--》则该客户端获得锁，把lock.foo的键值设置为时间值表示该键已被锁定，该客户端最后可以通过DEL lock.foo来释放该锁
如果key存在，返回false-->表明该锁已被其他客户端取得，这时我们可以先返回或进行重试等对方完成或等待锁超时
*/
        if (redisTemplate.opsForValue().setIfAbsent(key, value)) {
            return true;
        }
        String currentValue = redisTemplate.opsForValue().get(key);
        // 如果锁过期
        if (!StringUtils.isEmpty(currentValue)
                && Long.parseLong(currentValue) < System.currentTimeMillis()) {
            // 获取上一个锁时间
            // 设置最新的结果并返回设置之前的结果
            String oldValue = redisTemplate.opsForValue().getAndSet(key, value);
            if (!StringUtils.isEmpty(oldValue)
                    && oldValue.equals(currentValue)) {
                return true;
            }
        }
        return false;
    }

    /**
     * 解锁
     *
     * @param key
     * @param value
     */
    public void unLock(String key, String value) {
        try {
            String currentValue = redisTemplate.opsForValue().get(key);
            if (!StringUtils.isEmpty(currentValue)
                    && currentValue.equals(value)) {
                redisTemplate.opsForValue().getOperations().delete(key);
            }
        } catch (Exception e) {
            log.error("【redis分布式锁】解锁异常, {}", e);
        }
    }
}

二、如何使用

1、设置超时时间、并注入分布式锁

 private static final int TIMEOUT = 10 * 1000; // 超时时间

 @Autowired
 private RedisLock redisLock;

2、在需要使用的地方使用

    @Override
    public SellerInfo findSellerInfoByOpenid(String openid) {
        // 加锁
        long time = System.currentTimeMillis() + TIMEOUT;
        if (!redisLock.lock(productId, String.valueOf(time))) {
            // 加锁是吧抛出异常
            throw new SellException(101, "人太多了，换个姿势试试~~");
        }
        // 执行业务代码
        // ......
        // ......
        // ......
        // 解锁
        redisLock.unLock(productId, String.valueOf(time));
    }