zookeeper 实现分布式锁

https://github.com/hd19940104/java/tree/master/project/micro/spring-cloud/spring-cloud-zookeeper

1、首先搭载zookeeper集群

--------------------------------------

直接在这压缩包解压，

修改环境变量

vi /etc/profile   需要安装jdk（同理直接解压配置环境变量）

 

修改zoo_sample.cfg文件

cd /usr/local/zookeeper/conf
mv zoo_sample.cfg zoo.cfg

创建服务标识server.服务标识=hadoop*

.

--------------------------------------------

zkServer.sh start/stop/status

--------------------------------上述完成zookeeper集群搭建---------------------------------------------------

【分布式锁】：在分布式架构中，存在多个线程共享资源，=，就会引发线程安全问题；

如何解决线程安全，对共享资源进行加锁。在分布式中，是有数据库分布式锁（影响性能）、redis分布式锁（不好判断失效时间）

但是zookeeper有一个特点：临时节点；并且有监听节点状态机制；恰恰刚好能解决资源共享问题；

一个线程获取资源的时候获取锁除了完了再释放锁，就可以利用zookeeper写

-------------------列子【生成订单号、多线程下】-----------------------------

===============================OrderNumGenerator .java========================================

package com.zixue.lock;

import java.text.SimpleDateFormat;
import java.util.Date;

//生成订单号规则
public class OrderNumGenerator {
    private static int count = 0;

    //生成订单号规则方法
    
    
    public String orderNumber() {
        SimpleDateFormat simpt = new SimpleDateFormat("yyyy-MM-dd-HH-mm-ss");
        return simpt.format(new Date()) + "-" + ++count;
    }

}

===============================OrderService .java========================================

package com.zixue.lock;

//订单服务
public class OrderService implements Runnable {
    private OrderNumGenerator orderNumGenerator = new OrderNumGenerator();
    private static Object oj = new Object();
    private Lock lock=new ZookeeperDistrbuteLock();

    public void run() {
        getNumber();
    }

    public void getNumber() {
        // synchronized (oj) {
        lock.getLock();
        String number = orderNumGenerator.orderNumber();
        System.out.println(Thread.currentThread().getName() + ",生成订单号:" + number);
        lock.unLock();

    }

    public static void main(String[] args) {
        for (int i = 0; i < 100; i++) {
            new Thread(new OrderService()).start();
        }
    }

}

===============================Lock .java========================================

package com.zixue.lock;

public interface Lock {
    //获取到锁的资源
    public void getLock();
    // 释放锁
    public void unLock();
}

===============================ZookeeperAbstractLock .java========================================

package com.zixue.lock;

import org.I0Itec.zkclient.ZkClient;

//将重复的代码,具体业务逻辑有子类去实现.
public abstract class ZookeeperAbstractLock implements Lock {
    private static final String CONNECT_ADDRES = "192.168.1.103:2181,192.168.1.113:2181,192.168.1.114:2181";
    protected ZkClient zkClient = new ZkClient(CONNECT_ADDRES);
    protected static final String PATH = "/lock";

    public void getLock() {
        // tryLock() 创建zk临时节点 如果创建成功返回true 否则返回false
        if (tryLock()) {
            System.out.println("获取到锁的资源 get lock");
        } else {
            // 等待
            waitLock();
            // 重写获取锁的资源
            getLock();
        }

    }

    protected abstract boolean tryLock();

    protected abstract void waitLock();

    // 释放锁
    public void unLock() {
        if (zkClient != null) {
            zkClient.close();
        }
        System.out.println("释放锁的资源..");

    }

}

===============================ZookeeperDistrbuteLock .java========================================

package com.zixue.lock;

import java.util.concurrent.CountDownLatch;

import org.I0Itec.zkclient.IZkDataListener;

public class ZookeeperDistrbuteLock extends ZookeeperAbstractLock {
    private CountDownLatch countDownLatch = null;

    @Override
    protected boolean tryLock() {
        try {
            zkClient.createEphemeral(PATH);
            return true;
        } catch (Exception e) {
        //e.printStackTrace();
            return false;
        }
    }

    @Override
    protected void waitLock() {
        IZkDataListener zkDataListener = new IZkDataListener() {
            // 节点被删除的时候 事件通知
            public void handleDataDeleted(String path) throws Exception {
                // 唤醒被等待的线程
                if (countDownLatch != null) {
                    countDownLatch.countDown();
                    System.out.println("删除节点.....");
                }
            }

            public void handleDataChange(String path, Object data) throws Exception {

            }
        };
        // 注册到zkclient进行监听
        zkClient.subscribeDataChanges(PATH, zkDataListener);
        if (zkClient.exists(PATH)) {
            countDownLatch = new CountDownLatch(1);
            try {
                countDownLatch.await();
            } catch (Exception e) {
                // TODO: handle exception
            }
        }
        // 删除监听
        zkClient.unsubscribeDataChanges(PATH, zkDataListener);
    }

}

==================pom.xml===============

 

<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
  xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
  <modelVersion>4.0.0</modelVersion>

  <groupId>zixue.com</groupId>
  <artifactId>spring-cloud-zookeeper</artifactId>
  <version>0.0.1-SNAPSHOT</version>
  <packaging>jar</packaging>

  <name>spring-cloud-zookeeper</name>
  <url>http://maven.apache.org</url>
  <parent>
        <groupId>com.zixue</groupId>
        <artifactId>shopmall-parent</artifactId>
        <version>0.0.1-SNAPSHOT</version>
    </parent>
    <dependencies>
        <dependency>
            <groupId>org.apache.zookeeper</groupId>
            <artifactId>zookeeper</artifactId>
            <version>3.4.6</version>
        </dependency>
        <dependency>
            <groupId>com.101tec</groupId>
            <artifactId>zkclient</artifactId>
            <version>0.10</version>
        </dependency>
    </dependencies>

</project>