org.noear.water.utils.LockUtils Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of water.client Show documentation
The newest version!
package org.noear.water.utils;

import org.noear.redisx.RedisClient;
import org.noear.water.WaterSetting;

/**
 * 分布式锁工具
 *
 * @author noear
 * @since 2.0
 * */
public class LockUtils {
    private static RedisClient _redis_lock = WaterSetting.redis_cfg().getRd(2);

    public static RedisClient getClient(){
        return _redis_lock;
    }

    /**
     * 尝试把 group_key 锁定给 inMaster
     *
     * @param group     分组
     * @param key       关键字
     * @param inSeconds 锁定时间
     * @param inMaster  锁持有人
     */
    public static boolean tryLock(String group, String key, int inSeconds, String inMaster) {
        String key2 = group + ".lk." + key;

        return _redis_lock.openAndGet((ru) -> ru.key(key2).expire(inSeconds).lock(inMaster));
    }

    /**
     * 尝试把 group_key 锁定
     *
     * @param inSeconds 锁定时间
     */
    public static boolean tryLock(String group, String key, int inSeconds) {
        String key2 = group + ".lk." + key;

        return _redis_lock.openAndGet((ru) -> ru.key(key2).expire(inSeconds).lock("_"));
    }

    /**
     * 尝试把 group_key 锁定 （3秒）
     */
    public static boolean tryLock(String group, String key) {
        return tryLock(group, key, 3);
    }

    /**
     * 检查 group_key 是否已被锁定
     */
    public static boolean isLocked(String group, String key) {
        String key2 = group + ".lk." + key;

        return _redis_lock.openAndGet((ru) -> ru.key(key2).exists());
    }

    public static String getLockValue(String group, String key) {
        String key2 = group + ".lk." + key;
        return _redis_lock.openAndGet((ru) -> ru.key(key2).get());
    }


    /**
     * 解锁 group_key
     */
    public static void unLock(String group, String key, String inMaster) {
        String key2 = group + ".lk." + key;

        _redis_lock.open((ru) -> {
            if (inMaster == null || inMaster.equals(ru.key(key2).getAsLong())) {
                ru.key(key2).delete();
            }
        });
    }

    /**
     * 解锁 group_key
     */
    public static void unLock(String group, String key) {
        unLock(group, key, null);
    }
}