RedisLockUtil.java 1.32 KB
Newer Older
hkl's avatar
hkl committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58
package org.jeecg.modules.utils;

import org.springframework.data.redis.core.RedisTemplate;

import java.util.concurrent.TimeUnit;

/**
 * redis 锁
 */
public class RedisLockUtil {

    private RedisLockUtil() {
    }

    private static RedisTemplate<String, Object> redisTemplate;

    public static void setService(RedisTemplate<String, Object> redisTemplate) {
        RedisLockUtil.redisTemplate = redisTemplate;
    }

    /**
     * @param key 键
     * @return
     */
    public static boolean lock(String key) {
        /**
         * 默认期望过期时间
         */
        Long expireTime = 6000L;
        return lock(key, expireTime);
    }

    /**
     * @param key        键
     * @param expireTime 过期时间
     * @return
     */
    public static boolean lock(String key, Long expireTime) {
        if (key == null) {
            return false;
        }
        Boolean flag = redisTemplate.hasKey(key);
        if (Boolean.FALSE.equals(flag)) {
            redisTemplate.opsForValue().set(key, key, expireTime, TimeUnit.MILLISECONDS);
            return true;
        } else {
            return false;
        }
    }

    public static boolean unlock(String key) {
        if (key == null) {
            return true;
        } else {
            return Boolean.TRUE.equals(redisTemplate.delete(key));
        }
    }
}