org.apache.hadoop.registry.client.api.RegistryConstants Java Examples

private static void initAndStart(){
    try{
        Configuration conf  = new YarnConfiguration();
        conf.set(RegistryConstants.KEY_REGISTRY_ZK_ROOT, YarnZkContant.ZK_AM_REGISTRY_ROOT);
        registryOperations = RegistryOperationsFactory.createInstance(YarnZkContant.ZK_AM_REGISTRY_JSTORM_YARN, conf);
        if (registryOperations instanceof RMRegistryOperationsService) {
            RMRegistryOperationsService rmRegOperations = (RMRegistryOperationsService) registryOperations;
            rmRegOperations.initUserRegistryAsync(StreamContant.HADOOP_USER_NAME);
        }
    }catch(Exception e){
    }
    if(registryOperations==null){
        throw new IllegalArgumentException("YarnZkClientBusiness registryOperations start is error，cann't connect yarn zk！！");
    }
    registryOperations.start();
}
 

/**
 * Buld the user path -switches to the system path if the user is "".
 * It also cross-converts the username to ascii via punycode
 * @param username username or ""
 * @return the path to the user
 */
public static String homePathForUser(String username) {
  Preconditions.checkArgument(username != null, "null user");

  // catch recursion
  if (username.startsWith(RegistryConstants.PATH_USERS)) {
    return username;
  }
  if (username.isEmpty()) {
    return RegistryConstants.PATH_SYSTEM_SERVICES;
  }

  // convert username to registry name
  String convertedName = convertUsername(username);

  return RegistryPathUtils.join(RegistryConstants.PATH_USERS,
      encodeForRegistry(convertedName));
}
 

/**
 * Buld the user path -switches to the system path if the user is "".
 * It also cross-converts the username to ascii via punycode
 * @param username username or ""
 * @return the path to the user
 */
public static String homePathForUser(String username) {
  Preconditions.checkArgument(username != null, "null user");

  // catch recursion
  if (username.startsWith(RegistryConstants.PATH_USERS)) {
    return username;
  }
  if (username.isEmpty()) {
    return RegistryConstants.PATH_SYSTEM_SERVICES;
  }

  // convert username to registry name
  String convertedName = convertUsername(username);

  return RegistryPathUtils.join(RegistryConstants.PATH_USERS,
      encodeForRegistry(convertedName));
}
 

/**
 * Create a service classpath
 * @param user username or ""
 * @param serviceClass service name
 * @return a full path
 */
public static String serviceclassPath(String user,
    String serviceClass) {
  String services = join(homePathForUser(user),
      RegistryConstants.PATH_USER_SERVICES);
  return join(services,
      serviceClass);
}
 

public YarnConfiguration createRegistryConfiguration() {
  YarnConfiguration conf = new YarnConfiguration();
  conf.setInt(RegistryConstants.KEY_REGISTRY_ZK_CONNECTION_TIMEOUT, 1000);
  conf.setInt(RegistryConstants.KEY_REGISTRY_ZK_RETRY_INTERVAL, 500);
  conf.setInt(RegistryConstants.KEY_REGISTRY_ZK_RETRY_TIMES, 10);
  conf.setInt(RegistryConstants.KEY_REGISTRY_ZK_RETRY_CEILING, 10);
  conf.set(RegistryConstants.KEY_REGISTRY_ZK_QUORUM,
      zookeeper.getConnectionString());
  return conf;
}
 

@Before
public void setupTestSecureRMRegistryOperations() throws Exception {
  startSecureZK();
  secureConf = new Configuration();
  secureConf.setBoolean(KEY_REGISTRY_SECURE, true);

  // create client conf containing the ZK quorum
  zkClientConf = new Configuration(secureZK.getConfig());
  zkClientConf.setBoolean(KEY_REGISTRY_SECURE, true);
  assertNotEmpty(zkClientConf.get(RegistryConstants.KEY_REGISTRY_ZK_QUORUM));

  // ZK is in charge
  secureConf.set(KEY_REGISTRY_SYSTEM_ACCOUNTS, "sasl:zookeeper@");
  zookeeperUGI = loginUGI(ZOOKEEPER, keytab_zk);
}
 

/**
 * Create a service classpath
 * @param user username or ""
 * @param serviceClass service name
 * @return a full path
 */
public static String serviceclassPath(String user,
    String serviceClass) {
  String services = join(homePathForUser(user),
      RegistryConstants.PATH_USER_SERVICES);
  return join(services,
      serviceClass);
}
 

public YarnConfiguration createRegistryConfiguration() {
  YarnConfiguration conf = new YarnConfiguration();
  conf.setInt(RegistryConstants.KEY_REGISTRY_ZK_CONNECTION_TIMEOUT, 1000);
  conf.setInt(RegistryConstants.KEY_REGISTRY_ZK_RETRY_INTERVAL, 500);
  conf.setInt(RegistryConstants.KEY_REGISTRY_ZK_RETRY_TIMES, 10);
  conf.setInt(RegistryConstants.KEY_REGISTRY_ZK_RETRY_CEILING, 10);
  conf.set(RegistryConstants.KEY_REGISTRY_ZK_QUORUM,
      zookeeper.getConnectionString());
  return conf;
}
 

@Before
public void setupTestSecureRMRegistryOperations() throws Exception {
  startSecureZK();
  secureConf = new Configuration();
  secureConf.setBoolean(KEY_REGISTRY_SECURE, true);

  // create client conf containing the ZK quorum
  zkClientConf = new Configuration(secureZK.getConfig());
  zkClientConf.setBoolean(KEY_REGISTRY_SECURE, true);
  assertNotEmpty(zkClientConf.get(RegistryConstants.KEY_REGISTRY_ZK_QUORUM));

  // ZK is in charge
  secureConf.set(KEY_REGISTRY_SYSTEM_ACCOUNTS, "sasl:zookeeper@");
  zookeeperUGI = loginUGI(ZOOKEEPER, keytab_zk);
}
 

/**
 * Get the absolute path to where the service has registered itself.
 * This includes the base registry path
 * Null until the service is registered
 * @return the service registration path.
 */
public String getAbsoluteSelfRegistrationPath() {
  if (selfRegistrationPath == null) {
    return null;
  }
  String root = registryOperations.getConfig().getTrimmed(
      RegistryConstants.KEY_REGISTRY_ZK_ROOT,
      RegistryConstants.DEFAULT_ZK_REGISTRY_ROOT);
  return RegistryPathUtils.join(root, selfRegistrationPath);
}
 

@Test
public void testDefaultAClsValid() throws Throwable {
  registrySecurity.buildACLs(
      RegistryConstants.DEFAULT_REGISTRY_SYSTEM_ACCOUNTS,
      REALM_EXAMPLE_COM, ZooDefs.Perms.ALL);
}
 

@Test
public void testDefaultAClsValid() throws Throwable {
  registrySecurity.buildACLs(
      RegistryConstants.DEFAULT_REGISTRY_SYSTEM_ACCOUNTS,
      REALM_EXAMPLE_COM, ZooDefs.Perms.ALL);
}
 

/**
 * Create a path for listing components under a service
 * @param user username or ""
 * @param serviceClass service name
 * @param serviceName service name unique for that user and service class
 * @return a full path
 */
public static String componentListPath(String user,
    String serviceClass, String serviceName) {

  return join(servicePath(user, serviceClass, serviceName),
      RegistryConstants.SUBPATH_COMPONENTS);
}
 

/**
 * Create a path for listing components under a service
 * @param user username or ""
 * @param serviceClass service name
 * @param serviceName service name unique for that user and service class
 * @return a full path
 */
public static String componentListPath(String user,
    String serviceClass, String serviceName) {

  return join(servicePath(user, serviceClass, serviceName),
      RegistryConstants.SUBPATH_COMPONENTS);
}