1、功能:

实现多进程,或多线程争抢zookeeper锁,例如4个进程,程序名:DLFrame,进程名:DLFrame_10001,去争抢任务锁:10001。只会有一个进程争抢到锁10001,其余经常在等待获取锁,也可以用于单进程去争抢多个任务。

在192.168.1.25 部署4个进程

 去争取10001锁,最终只会有一个进程争抢到。争取到效果如图:

2、使用方法


AppName:是进程名

PrgName:是程序名

TaskId:是任务锁标识

appIns:是进程数(默认配置1,)

以下完整代码:

package com.asap.test;


import com.asap.lock.LostCallBack;
import com.asap.lock.ReconnectCallBack;
import com.asap.lock.SuspendCallBack;
import com.asap.dl.plugin.logmgr.Log;
import com.asap.dl.plugin.logmgr.LogFactory;
import com.asap.sri.ccp.zk.ZkConstant;
import com.asap.sri.ccp.zk.constant.Constants;
import com.asap.sri.ccp.zk.exception.ZkException;
import com.asap.sri.ccp.zk.exception.ZkLockException;
import com.asap.sri.ccp.zk.impl.ZkMgrImpl;
import com.asap.sri.ccp.zk.lock.ProcessLock;

import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Properties;


/**
 * @author wangbh
 * @Description: test
 * @date 2021/8/27 14:05
 */
public class Test {
    private static final Log LOG = LogFactory.getLogger(Test.class);

    static String key = "test123";
    static   int cnt=100;
    static class MyThread implements Runnable {
        @Override
        public void run() {
            // TODO Auto-generated method stub
            try {
                ProcessLock.doGetResource(true);
                System.out.println(Thread.currentThread().getName() +":"+Thread.currentThread().getId() +":release");

            } catch (Exception e) {
                System.out.println(e);
            }
        }
    }


    private static String zk = "192.168.1.25:2184";

    private static void initZK() throws ZkException, ZkLockException {

        /*
         *   任务锁,通常配置成唯一的一串字符,该任务被进程实例抢到会注册在/lock/{PrgName}_task节点下注册临时节点,
         */
        Constants.getAllTaskIdList().add("10001");
        /*
         *  进程名,/lock/PrgName的永久性子节点
         * 每个进程都会在/lock/PrgName下创建lock_0...0xx(xx是往上递增+1)
         *抢到锁的会把自己的lock删掉
         */
        Constants.setPrgName("DLFrame");
        /*
            进程实例名,最终抢到的任务锁的进程会注册在/lock/{PrgName}_run/节点下注册临时节点
         */
        Constants.setAppName("DLFrame_10001");
        /*
            表示进程实例数只有一个,去争取任务锁
         */
        Constants.setAppIns(1);
        /*
            认证方式
        */
        Constants.setZkScheme("digest");
        /*
         * 用户名:密码
         * */
        Constants.setZkAuth("admin1:admin1");
        Constants.setZkMgr(new ZkMgrImpl(zk, 50000, ZkConstant.POLICY_RETRY_MORE));

        // 添加网络中断回调函数,需要自己实现以下三个接口。
        /*
         *需要自己实现ISuspendCallBack接口
         */
        Constants.getZkMgr().suspendNotify(new SuspendCallBack());
        /*
         * 需要自己实现ILostCallBack接口
         */
        Constants.getZkMgr().lostedNotify(new LostCallBack());
        /*
         * 需要自己实现IReconnectCallBack接口
         */
        Constants.getZkMgr().reconnectedNotify(new ReconnectCallBack());

    }

    public static void main(String[] args) throws Exception {
        //初始化zk
        initZK();
        // 争抢锁,没有争抢到在里面一直循环等待
        ProcessLock.doGetResource(true);
        //争抢到锁的进程..
        while (true) {
            LOG.warn("getLock....");
            System.out.println("Now all thread done!");
            Thread.sleep(1000);
        }

    }
}

3、程序架构

 源码:Zookeeper: 基于CuratorFramework实现zookeeper分布式锁,实现任务争抢,程序高可用方案。 - Gitee.com

要源码请联系我:qq:1053714778或私信我

Logo

权威|前沿|技术|干货|国内首个API全生命周期开发者社区

更多推荐