@Test() public void testAdministrator() throws Exception { System.out.println( "START " + className + ".testAdministrator() at " + new Date(System.currentTimeMillis())); final String clusterName = CLUSTER_PREFIX + "_" + className + "_admin"; // basic test if (_gZkClient.exists("/" + clusterName)) { _gZkClient.deleteRecursive("/" + clusterName); } ZKHelixManager admin = new ZKHelixManager(clusterName, null, InstanceType.ADMINISTRATOR, ZK_ADDR); TestHelper.setupEmptyCluster(_gZkClient, clusterName); admin.connect(); AssertJUnit.assertTrue(admin.isConnected()); HelixAdmin adminTool = admin.getClusterManagmentTool(); // ConfigScope scope = new ConfigScopeBuilder().forCluster(clusterName) // .forResource("testResource").forPartition("testPartition").build(); HelixConfigScope scope = new HelixConfigScopeBuilder(ConfigScopeProperty.PARTITION) .forCluster(clusterName) .forResource("testResource") .forPartition("testPartition") .build(); Map<String, String> properties = new HashMap<String, String>(); properties.put("pKey1", "pValue1"); properties.put("pKey2", "pValue2"); adminTool.setConfig(scope, properties); properties = adminTool.getConfig(scope, Arrays.asList("pKey1", "pKey2")); Assert.assertEquals(properties.size(), 2); Assert.assertEquals(properties.get("pKey1"), "pValue1"); Assert.assertEquals(properties.get("pKey2"), "pValue2"); admin.disconnect(); AssertJUnit.assertFalse(admin.isConnected()); System.out.println( "END " + className + ".testAdministrator() at " + new Date(System.currentTimeMillis())); }
public static void updateResourceConfigsFor( Map<String, String> newConfigs, String resourceName, String clusterName, HelixAdmin admin) { final HelixConfigScope scope = getResourceScopeFor(clusterName, resourceName); admin.setConfig(scope, newConfigs); }