分布式锁-zookeeper-SharedLock基于InterProcessSemaphoreMutex（二）分布式共享锁

最新推荐文章于 2024-08-14 15:07:43 发布

最后的miku殿下

最新推荐文章于 2024-08-14 15:07:43 发布

阅读量3.1k

点赞数

分类专栏： zookeeper apache curator 分布式锁文章标签：分布式 zookeeper 继承

本文链接：https://blog.csdn.net/Hatsune_Miku_/article/details/79234079

版权

apache 同时被 3 个专栏收录

4 篇文章 0 订阅

订阅专栏

curator

4 篇文章 0 订阅

订阅专栏

分布式锁

4 篇文章 0 订阅

订阅专栏

分布式共享锁：InterProcessSemaphoreMutex完全分布的全局同步锁，意味着在任何快照时间，没有两个客户端认为他们拥有相同的锁，继承上一篇分布式锁-zookeeper的AbstractZookeeperLock。细节与注意事项请看分布式锁-基于zookeeper客户端curator实现锁机制（一）锁基础

package com.miku.common.util.lock.zookeeper;

import com.miku.common.util.lock.support.AbstractZookeeperLock;
import org.apache.curator.framework.recipes.locks.InterProcessSemaphoreMutex;

/**
 * 分布式共享锁
 *
 * @author:panw.
 */
public class SharedLock extends AbstractZookeeperLock<InterProcessSemaphoreMutex> {

    private final String lockKey = "shared-lock";

    /**
     * Shared Lock
     *
     * 获取排他锁，阻塞直到可用，单实例{@code InterProcessSemaphoreMutex} <br>
     * interProcessMutex {@link InterProcessSemaphoreMutex#InterProcessSemaphoreMutex} : <ul>
     * <li>参数1: zk连接上下文{@link AbstractZookeeperLock#curatorFramework} </li>
     * <li>参数2: zk锁路径，路径相同锁相同 </li>
     * </ul>
     *
     * @param var
     * @throws Exception 抛出异常时，锁不在持有，获取锁失败，谨慎处理业务
     */
    public void acquire(String var){
        try {
            super.lock(getSharedLock(var));
        } catch (Exception e) {
            throw new IllegalStateException("获取共享锁失败=>" + e);
        }
    }

    /**
     * Shared Lock
     *
     * @see com.miku.common.util.lock.zookeeper.SharedLock#acquire(String)
     *
     * @param var
     * @param time  获取互斥锁：阻止直到可用或给定的时间到期
     * @return  如果获得了互斥锁，则返回true，否则返回false
     * @throws Exception    抛出异常时，锁不在持有，获取锁失败，谨慎处理业务
     */
    public boolean acquire(String var,long time) throws Exception {
        try {
            return super.lock(getSharedLock(var),time);
        } catch (Exception e) {
            throw new IllegalStateException("获取共享锁失败=>" + e);
        }
    }

    private InterProcessSemaphoreMutex getSharedLock(String var){
        String key = lockKey + var;
        InterProcessSemaphoreMutex interProcessSemaphoreMutex = null;
        if (locks.containsKey(key)){
            interProcessSemaphoreMutex = locks.get(key);
        }else {
            interProcessSemaphoreMutex = new InterProcessSemaphoreMutex(curatorFramework,var);
            locks.put(key,interProcessSemaphoreMutex);
        }
        return interProcessSemaphoreMutex;
    }

    public void release(String var) throws Exception {
        super.unlock(locks.get(lockKey + var));
    }
}