Java Code Examples for org.apache.zookeeper.ZooKeeper#addAuthInfo()

private ZookeeperFactory makeZookeeperFactory(final ZookeeperFactory actualZookeeperFactory)
{
    return new ZookeeperFactory()
    {
        @Override
        public ZooKeeper newZooKeeper(String connectString, int sessionTimeout, Watcher watcher, boolean canBeReadOnly) throws Exception
        {
            ZooKeeper zooKeeper = actualZookeeperFactory.newZooKeeper(connectString, sessionTimeout, watcher, canBeReadOnly);
            for ( AuthInfo auth : authInfos )
            {
                zooKeeper.addAuthInfo(auth.getScheme(), auth.getAuth());
            }

            return zooKeeper;
        }
    };
}
 

private ZookeeperFactory makeZookeeperFactory(final ZookeeperFactory actualZookeeperFactory)
{
    return new ZookeeperFactory()
    {
        @Override
        public ZooKeeper newZooKeeper(String connectString, int sessionTimeout, Watcher watcher, boolean canBeReadOnly) throws Exception
        {
            ZooKeeper zooKeeper = actualZookeeperFactory.newZooKeeper(connectString, sessionTimeout, watcher, canBeReadOnly);
            for ( AuthInfo auth : authInfos )
            {
                zooKeeper.addAuthInfo(auth.getScheme(), auth.getAuth());
            }

            return zooKeeper;
        }
    };
}
 

private void createZookeeper(final CountDownLatch connectionLatch) throws Exception {
    zk = new ZooKeeper(this.properties.getProperty(keys.zkConnectString.toString()),
        Integer.parseInt(this.properties.getProperty(keys.zkSessionTimeout.toString())),
        new Watcher() {
            @Override
            public void process(WatchedEvent event) {
                sessionEvent(connectionLatch, event);
            }
        });
    String authString = this.properties.getProperty(keys.userName.toString()) + ":" + this.properties
        .getProperty(keys.password.toString());
    this.isCheckParentPath = Boolean
        .parseBoolean(this.properties.getProperty(keys.isCheckParentPath.toString(), "true"));
    zk.addAuthInfo("digest", authString.getBytes());
    acl.clear();
    acl.add(new ACL(ZooDefs.Perms.ALL, new Id("digest", DigestAuthenticationProvider.generateDigest(authString))));
    acl.add(new ACL(ZooDefs.Perms.READ, Ids.ANYONE_ID_UNSAFE));
}
 

private void createZookeeper(final CountDownLatch connectionLatch) throws Exception {
  zk = new ZooKeeper(this.properties.getProperty(keys.zkConnectString
      .toString()), Integer.parseInt(this.properties
      .getProperty(keys.zkSessionTimeout.toString())),
      new Watcher() {
        public void process(WatchedEvent event) {
          sessionEvent(connectionLatch, event);
        }
      });
  String authString = this.properties.getProperty(keys.userName.toString())
      + ":" + this.properties.getProperty(keys.password.toString());
  zk.addAuthInfo("digest", authString.getBytes());
  acl.clear();
  acl.add(new ACL(ZooDefs.Perms.ALL, new Id("digest",
      DigestAuthenticationProvider.generateDigest(authString))));
  acl.add(new ACL(ZooDefs.Perms.READ, Ids.ANYONE_ID_UNSAFE));
}
 

/**
 * Get a new zookeeper client instance. protected so that test class can
 * inherit and pass in a mock object for zookeeper
 * 
 * @return new zookeeper client instance
 * @throws IOException
 * @throws KeeperException zookeeper connectionloss exception
 */
protected synchronized ZooKeeper getNewZooKeeper() throws IOException,
    KeeperException {
  
  // Unfortunately, the ZooKeeper constructor connects to ZooKeeper and
  // may trigger the Connected event immediately. So, if we register the
  // watcher after constructing ZooKeeper, we may miss that event. Instead,
  // we construct the watcher first, and have it block any events it receives
  // before we can set its ZooKeeper reference.
  watcher = new WatcherWithClientRef();
  ZooKeeper zk = new ZooKeeper(zkHostPort, zkSessionTimeout, watcher);
  watcher.setZooKeeperRef(zk);

  // Wait for the asynchronous success/failure. This may throw an exception
  // if we don't connect within the session timeout.
  watcher.waitForZKConnectionEvent(zkSessionTimeout);
  
  for (ZKAuthInfo auth : zkAuthInfo) {
    zk.addAuthInfo(auth.getScheme(), auth.getAuth());
  }
  return zk;
}
 

/**
 * Get a new zookeeper client instance. protected so that test class can
 * inherit and pass in a mock object for zookeeper
 * 
 * @return new zookeeper client instance
 * @throws IOException
 * @throws KeeperException zookeeper connectionloss exception
 */
protected synchronized ZooKeeper getNewZooKeeper() throws IOException,
    KeeperException {
  
  // Unfortunately, the ZooKeeper constructor connects to ZooKeeper and
  // may trigger the Connected event immediately. So, if we register the
  // watcher after constructing ZooKeeper, we may miss that event. Instead,
  // we construct the watcher first, and have it block any events it receives
  // before we can set its ZooKeeper reference.
  watcher = new WatcherWithClientRef();
  ZooKeeper zk = new ZooKeeper(zkHostPort, zkSessionTimeout, watcher);
  watcher.setZooKeeperRef(zk);

  // Wait for the asynchronous success/failure. This may throw an exception
  // if we don't connect within the session timeout.
  watcher.waitForZKConnectionEvent(zkSessionTimeout);
  
  for (ZKAuthInfo auth : zkAuthInfo) {
    zk.addAuthInfo(auth.getScheme(), auth.getAuth());
  }
  return zk;
}
 

private void createZookeeper(final CountDownLatch connectionLatch) throws Exception {
    zk = new ZooKeeper(this.properties.getProperty(keys.zkConnectString
            .toString()), Integer.parseInt(this.properties
            .getProperty(keys.zkSessionTimeout.toString())),
            new Watcher() {
                public void process(WatchedEvent event) {
                    sessionEvent(connectionLatch, event);
                }
            });
    String authString = this.properties.getProperty(keys.userName.toString())
            + ":"+ this.properties.getProperty(keys.password.toString());
    zk.addAuthInfo("digest", authString.getBytes());
    acl.clear();
    acl.add(new ACL(ZooDefs.Perms.ALL, new Id("digest",
            DigestAuthenticationProvider.generateDigest(authString))));
    acl.add(new ACL(ZooDefs.Perms.READ, Ids.ANYONE_ID_UNSAFE));
}
 

@Test
public void testCreateTask() throws Exception {
	ZooKeeper zk = new ZooKeeper("localhost:2181", 3000, null);
	List<ACL> acls = new ArrayList<ACL>();
	zk.addAuthInfo("digest", "ScheduleAdmin:password".getBytes());
	acls.add(new ACL(ZooDefs.Perms.ALL, new Id("digest",
			DigestAuthenticationProvider.generateDigest("ScheduleAdmin:password"))));
	acls.add(new ACL(ZooDefs.Perms.READ, Ids.ANYONE_ID_UNSAFE));
	zk.create("/uncode/schedule/task/taskObj#print", new byte[0], acls, CreateMode.PERSISTENT);
	zk.getData("/uncode/schedule/task/taskObj#print", false, null);
}
 

@Test
public void deletePath() throws Exception {
  ZooKeeper zk = new ZooKeeper("localhost:2181", 3000, watcher);
  zk.addAuthInfo("digest", "ScheduleAdmin:123456".getBytes());

  ZKTools.deleteTree(zk, "/schedule/dev");

  Thread.sleep(10 * 1000);
  zk.close();
}
 

/**
 * Creates a new ZooKeeper connection.
 */
private ZooKeeper createZooKeeper() throws IOException {
  ZooKeeper zk = new ZooKeeper(zkStr, sessionTimeout, wrapWatcher(this));
  for (Map.Entry<String, byte[]> authInfo : authInfos.entries()) {
    zk.addAuthInfo(authInfo.getKey(), authInfo.getValue());
  }
  return zk;
}
 

/**
 * Creates a new ZooKeeper connection.
 */
private ZooKeeper createZooKeeper() throws IOException {
  ZooKeeper zk = new ZooKeeper(zkStr, sessionTimeout, wrapWatcher(this));
  for (Map.Entry<String, byte[]> authInfo : authInfos.entries()) {
    zk.addAuthInfo(authInfo.getKey(), authInfo.getValue());
  }
  return zk;
}
 

@Test
public void deletePath() throws Exception {
	ZooKeeper zk = new ZooKeeper("127.0.0.1:2181", 3000, null);
	zk.addAuthInfo("digest", "ScheduleAdmin:password".getBytes());
	ZKTools.deleteTree(zk, "/uncode/schedule/task/taskObj#print");
	//ZKTools.deleteTree(zk, "/uncode/schedule");
	StringWriter writer = new StringWriter();
	ZKTools.printTree(zk, "/", writer, "\n");
	System.out.println(writer.getBuffer().toString());
}
 

@Test
public void testACL() throws Exception {
    ZooKeeper zk = new ZooKeeper("localhost:2181", 3000, new ScheduleWatcher(null));
    List<ACL> acls = new ArrayList<ACL>();
    zk.addAuthInfo("digest", "TestUser:password".getBytes());
    acls.add(new ACL(ZooDefs.Perms.ALL,
            new Id("digest", DigestAuthenticationProvider.generateDigest("TestUser:password"))));
    acls.add(new ACL(ZooDefs.Perms.READ, Ids.ANYONE_ID_UNSAFE));
    zk.create("/abc", new byte[0], acls, CreateMode.PERSISTENT);
    zk.getData("/abc", false, null);
}
 

@Test
public void deletePath() throws Exception {
    ZooKeeper zk = new ZooKeeper("localhost:2181", 3000, new ScheduleWatcher(null));
    zk.addAuthInfo("digest", "ScheduleAdmin:password".getBytes());

    ZKTools.deleteTree(zk, "/taobao-pamirs-schedule");
    StringWriter writer = new StringWriter();
    ZKTools.printTree(zk, "/", writer, "\n");
    System.out.println(writer.getBuffer().toString());
}
 

private ZooKeeper getZooKeeper(String zooKeeperConnectString, AuthMode authMode, byte[] authData) throws IOException {
    CountDownLatch connectionLatch = new CountDownLatch(1);
    ZooKeeper zooKeeper = new ZooKeeper(zooKeeperConnectString, 3000, watchedEvent -> {
        if (LOGGER.isDebugEnabled()) {
            LOGGER.debug("ZooKeeper server state changed to {} in {}", watchedEvent.getState(), zooKeeperConnectString);
        }
        if (watchedEvent.getType().equals(Watcher.Event.EventType.None) && watchedEvent.getState().equals(Watcher.Event.KeeperState.SyncConnected)) {
            connectionLatch.countDown();
        }
    });

    final boolean connected;
    try {
        connected = connectionLatch.await(5, TimeUnit.SECONDS);
    } catch (InterruptedException e) {
        closeZooKeeper(zooKeeper);
        Thread.currentThread().interrupt();
        throw new IOException(String.format("interrupted while waiting for ZooKeeper connection to %s", zooKeeperConnectString), e);
    }

    if (!connected) {
        closeZooKeeper(zooKeeper);
        throw new IOException(String.format("unable to connect to %s", zooKeeperConnectString));
    }

    if (authMode.equals(AuthMode.DIGEST)) {
        zooKeeper.addAuthInfo(SCHEME_DIGEST, authData);
    }
    return zooKeeper;
}
 

private ZooKeeper getZooKeeper(String zooKeeperConnectString, AuthMode authMode, byte[] authData) throws IOException {
    CountDownLatch connectionLatch = new CountDownLatch(1);
    ZooKeeper zooKeeper = new ZooKeeper(zooKeeperConnectString, 3000, watchedEvent -> {
        if (LOGGER.isDebugEnabled()) {
            LOGGER.debug("ZooKeeper server state changed to {} in {}", watchedEvent.getState(), zooKeeperConnectString);
        }
        if (watchedEvent.getType().equals(Watcher.Event.EventType.None) && watchedEvent.getState().equals(Watcher.Event.KeeperState.SyncConnected)) {
            connectionLatch.countDown();
        }
    });

    final boolean connected;
    try {
        connected = connectionLatch.await(5, TimeUnit.SECONDS);
    } catch (InterruptedException e) {
        closeZooKeeper(zooKeeper);
        Thread.currentThread().interrupt();
        throw new IOException(String.format("interrupted while waiting for ZooKeeper connection to %s", zooKeeperConnectString), e);
    }

    if (!connected) {
        closeZooKeeper(zooKeeper);
        throw new IOException(String.format("unable to connect to %s", zooKeeperConnectString));
    }

    if (authMode.equals(AuthMode.DIGEST)) {
        zooKeeper.addAuthInfo(SCHEME_DIGEST, authData);
    }
    return zooKeeper;
}
 

protected void initZookeeper() throws IOException {
    zooKeeper = new ZooKeeper(context.getServers(), context.getSessionTimeOut(), startWatcher());
    if (!Strings.isNullOrEmpty(context.getScheme())) {
        zooKeeper.addAuthInfo(context.getScheme(), context.getAuth());
    }
}
 

@Override
public void authenticate(ZooKeeper zooKeeper) {
    zooKeeper.addAuthInfo("digest", String.format("%s:%s", username, password).getBytes(UTF_8));
}
 

@Override
public void authenticate(ZooKeeper zooKeeper) {
    zooKeeper.addAuthInfo("digest", String.format("%s:%s", username, password).getBytes(UTF_8));
}
 

@Before
public void init() throws Exception {
    zoo = new ZooKeeper(HOST.concat(":" + CLIENT_PORT), TIME_OUT_MILLISECOND, null);
    zoo.addAuthInfo("digest", "yuzhouwan:com".getBytes());
    zooNoAuth = new ZooKeeper(HOST.concat(":" + CLIENT_PORT), TIME_OUT_MILLISECOND, null);
}