ZooKeeper3.4.6学习笔记（四）权限控制ACL

苍天有泪

鲁春利的工作笔记，谁说程序员不能有文艺范？
　　

　　为了避免存储在ZooKeeper服务器上的数据被其他进程干扰或人为操作修改，需要对ZooKeeper上的数据访问进行权限控制（Access Control）。ZooKeeper提供了ACL的权限控制机制，来控制客户端对数据节点的访问控制。
    ZooKeeper提供了多种权限控制模式（Scheme)，分别是world、auth、digest、ip和super。
开发人员如果要使用ZooKeeper的权限控制功能，需要在ZooKeeper回话创建后，给该回话添加相关的权限信息（AuthInfo）。ZooKeeper客户端提供了相应的API接口来进行权限信息的设置：

    addAuthInfo(String scheme, byte [] auth)
    scheme ：权限控制模式，world、auth、digest、ip和super
    auth ：具体的权限信息　　说明：该接口主要用于为当前ZooKeeper会话添加权限信息，之后凡是通过该会话对ZooKeeper服务端进行的任何操作，都会带上该权限信息。
　　使用包含权限信息的ZooKeeper会话操作数据节点
package com.invic.zk;
import java.io.IOException;
import java.util.concurrent.CountDownLatch;
import org.apache.zookeeper.CreateMode;
import org.apache.zookeeper.KeeperException;
import org.apache.zookeeper.WatchedEvent;
import org.apache.zookeeper.Watcher;
import org.apache.zookeeper.ZooDefs.Ids;
import org.apache.zookeeper.ZooKeeper;
import org.apache.zookeeper.data.Stat;
/**
*
* @author lucl
*
*/
public class CusZookeeperACL {
    /**
     * @param args
     */
    public static void main(String[] args) {
        String path = "/zk-book-auth";
        String childPath = path + "/child";
        String connectString = "nnode:2181,dnode1:2181,dnode2:2181";
        int sessionTimeout = 30 * 1000;
        System.out.println("===============第一次、创建节点================");
        {
            try {
               final CountDownLatch latch = new CountDownLatch(1);
                ZooKeeper zk = new ZooKeeper(connectString, sessionTimeout, new Watcher() {
                    @Override
                    public void process(WatchedEvent event) {
                        System.out.println(event.getState());        // SyncConnected
                        if (KeeperState.SyncConnected == event.getState()) {
                            System.out.println("zk session established : SyncConnected.");
                            if (EventType.None == event.getType() && null == event.getPath()) {
                                latch.countDown();
                            }
                        }
                    }
                });
                latch.await();
                // 权限信息
                //        模式：world、auth、digest、ip和super
                //        auth：具体权限信息
                zk.addAuthInfo("digest", "foo:true".getBytes());
                // 创建节点，此时节点就受到了权限控制
                byte [] initData = "init".getBytes();
                byte [] childData = "child".getBytes();
                zk.create(path, initData , Ids.CREATOR_ALL_ACL, CreateMode.PERSISTENT);
                zk.create(childPath, childData, Ids.CREATOR_ALL_ACL, CreateMode.EPHEMERAL);
                Thread.sleep(sessionTimeout);
            } catch (IOException e) {
                e.printStackTrace();
            } catch (InterruptedException e) {
                e.printStackTrace();
            } catch (KeeperException e) {
                e.printStackTrace();
            }
        }
        System.out.println("==============第二次、未带权限信息的连接===============");
        {
            ZooKeeper zk2 = null;
            try {
                zk2 = new ZooKeeper(connectString, sessionTimeout, new Watcher() {
                    @Override
                    public void process(WatchedEvent event) {
                        System.out.println(event.getState());        // SyncConnected
                    }
                });
            } catch (IOException e) {
                e.printStackTrace();
            }
            try {
                Thread.sleep(sessionTimeout);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            if (null != zk2) {
                Stat stat = new Stat();
                try {
                    // 获取数据报错: KeeperErrorCode = NoAuth for /zk-book-auth
                    byte [] bytes = zk2.getData(path, false, stat);        
                    System.out.println("Get data from zk : " + new String(bytes));
                } catch (KeeperException e) {
                    e.printStackTrace();
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
            try {
                Thread.sleep(sessionTimeout);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
        System.out.println("==============第三次、正确的权限信息的连接===============");
        {
            ZooKeeper zk3 = null;
            try {
                zk3 = new ZooKeeper(connectString, sessionTimeout, new Watcher() {
                    @Override
                    public void process(WatchedEvent event) {
                        System.out.println(event.getState());        // SyncConnected
                    }
                });
                zk3.addAuthInfo("digest", "foo:true".getBytes());
            } catch (IOException e) {
                e.printStackTrace();
            }
            try {
                Thread.sleep(sessionTimeout);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            if (null != zk3) {
                try {
                    byte[] datas = zk3.getData(path, true, null);
                    // Get data from zk : init
                    System.out.println("Get data from zk : " + new String(datas));   
                } catch (KeeperException e) {
                    e.printStackTrace();
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }   
            }
            try {
                Thread.sleep(sessionTimeout);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
        System.out.println("==============第四次、错误的权限信息的连接===============");
        {
            ZooKeeper zk4 = null;
            try {
                zk4 = new ZooKeeper(connectString, sessionTimeout, new Watcher() {
                    @Override
                    public void process(WatchedEvent event) {
                        System.out.println(event.getState());        // SyncConnected
                    }
                });
                zk4.addAuthInfo("digest", "foo:false".getBytes());
            } catch (IOException e) {
                e.printStackTrace();
            }
            try {
                Thread.sleep(sessionTimeout);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            if (null != zk4) {
                try {
                    byte[] datas = zk4.getData(path, true, null);
                    // Get data from zk : init
                    System.out.println("Get data from zk : " + new String(datas));   
                } catch (KeeperException e) {
                    e.printStackTrace();
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }   
            }
            try {
                Thread.sleep(sessionTimeout);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
        System.out.println("==============第五次、未指定权限信息的连接===============");
        {
            ZooKeeper zk5 = null;
            try {
                zk5 = new ZooKeeper(connectString, sessionTimeout, new Watcher() {
                    @Override
                    public void process(WatchedEvent event) {
                        System.out.println(event.getState());        // SyncConnected
                    }
                });
            } catch (IOException e) {
                e.printStackTrace();
            }
            try {
                Thread.sleep(sessionTimeout);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            if (null != zk5) {
                try {
                    // 删除path的子节点KeeperErrorCode = NoAuth for /zk-book-auth/child
                    zk5.delete(path + "/child", -1);            
                } catch (InterruptedException e) {
                    e.printStackTrace();
                } catch (KeeperException e) {
                    e.printStackTrace();
                }
            }
            try {
                Thread.sleep(sessionTimeout);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
        System.out.println("==============第六次、指定权限信息的连接===============");
        {
            ZooKeeper zk6 = null;
            try {
                zk6 = new ZooKeeper(connectString, sessionTimeout, new Watcher() {
                    @Override
                    public void process(WatchedEvent event) {
                        System.out.println(event.getState());        // SyncConnected
                    }
                });
                zk6.addAuthInfo("digest", "foo:true".getBytes());
            } catch (IOException e) {
                e.printStackTrace();
            }
            try {
                Thread.sleep(sessionTimeout);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            if (null != zk6) {
                try {
                    // 成功删除
                    zk6.delete(childPath, -1);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                } catch (KeeperException e) {
                    e.printStackTrace();
                }
            }
            try {
                Thread.sleep(sessionTimeout);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
        System.out.println("==============第七次、未指定权限信息的连接===============");
        {
            ZooKeeper zk7 = null;
            try {
                zk7 = new ZooKeeper(connectString, sessionTimeout, new Watcher() {
                    @Override
                    public void process(WatchedEvent event) {
                        System.out.println(event.getState());        // SyncConnected
                    }
                });
            } catch (IOException e) {
                e.printStackTrace();
            }
            try {
                Thread.sleep(sessionTimeout);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            if (null != zk7) {
                try {
                    // 本次为删除path路径，成功删除--权限信息只对子节点起左右
                    zk7.delete(path, -1);            
                } catch (InterruptedException e) {
                    e.printStackTrace();
                } catch (KeeperException e) {
                    e.printStackTrace();
                }
            }
            try {
                Thread.sleep(sessionTimeout);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
    }
}