Examples of ZKHelixDataAccessor

Examples of org.apache.helix.manager.zk.ZKHelixDataAccessor

                                 String resourceGroupName,
                                 String resourceKey,
                                 String sessionId,
                                 String state)
  {
    ZKHelixDataAccessor accessor =
        new ZKHelixDataAccessor(clusterName, new ZkBaseDataAccessor(_gZkClient));
    Builder keyBuilder = accessor.keyBuilder();


    CurrentState curState = new CurrentState(resourceGroupName);
    curState.setState(resourceKey, state);
    curState.setSessionId(sessionId);
    curState.setStateModelDefRef("MasterSlave");
    accessor.setProperty(keyBuilder.currentState(instance, sessionId, resourceGroupName),
                         curState);
  }

View Full Code Here

Examples of org.apache.helix.manager.zk.ZKHelixDataAccessor

                            3, // replicas
                            "MasterSlave",
                            true); // do rebalance


    // add fake liveInstance
    ZKHelixDataAccessor accessor =
        new ZKHelixDataAccessor(clusterName, new ZkBaseDataAccessor<ZNRecord>(_gZkClient));
    Builder keyBuilder = new Builder(clusterName);
    LiveInstance liveInstance = new LiveInstance("localhost_12918");
    liveInstance.setSessionId("session_0");
    liveInstance.setHelixVersion("version_0");
    accessor.setProperty(keyBuilder.liveInstance("localhost_12918"), liveInstance);


    // drop without stop the process, should throw exception
    try
    {
      ClusterSetup.processCommandLineArgs(new String[] { "--zkSvr", ZK_ADDR,
          "--dropNode", clusterName, "localhost:12918" });
      Assert.fail("Should throw exception since localhost_12918 is still in LIVEINSTANCES/");
    }
    catch (Exception e)
    {
      // OK
    }
    accessor.removeProperty(keyBuilder.liveInstance("localhost_12918"));


    // drop without disable, should throw exception
    try
    {
      ClusterSetup.processCommandLineArgs(new String[] { "--zkSvr", ZK_ADDR,
          "--dropNode", clusterName, "localhost:12918" });
      Assert.fail("Should throw exception since localhost_12918 is enabled");
    }
    catch (Exception e)
    {
      // e.printStackTrace();
      // OK
    }


    // drop it
    ClusterSetup.processCommandLineArgs(new String[] { "--zkSvr", ZK_ADDR,
        "--enableInstance", clusterName, "localhost_12918", "false" });
    ClusterSetup.processCommandLineArgs(new String[] { "--zkSvr", ZK_ADDR, "--dropNode",
        clusterName, "localhost:12918" });


    Assert.assertNull(accessor.getProperty(keyBuilder.instanceConfig("localhost_12918")),
                      "Instance config should be dropped");
    Assert.assertFalse(_gZkClient.exists(PropertyPathConfig.getPath(PropertyType.INSTANCES,
                                                                    clusterName,
                                                                    "localhost_12918")),
                       "Instance/host should be dropped");

View Full Code Here

Examples of org.apache.helix.manager.zk.ZKHelixDataAccessor

    new HealthStatsAggregationTask(cmResult._manager).run();
  //sleep for a few seconds to give stats stage time to trigger
    Thread.sleep(3000);


    // other verifications go here
    ZKHelixDataAccessor accessor = new ZKHelixDataAccessor(clusterName, new ZkBaseDataAccessor(_zkClient));
    Builder keyBuilder = accessor.keyBuilder();


    //for (int i = 0; i < 1; i++) //change 1 back to 5
    //{
      //String instance = "localhost_" + (12918 + i);
      //String instance = "localhost_12918";
      ZNRecord record = accessor.getProperty(keyBuilder.alertStatus()).getRecord();
      Map<String, Map<String,String>> recMap = record.getMapFields();
      Set<String> keySet = recMap.keySet();
      Map<String,String> alertStatusMap = recMap.get(_alertStatusStr);
      String val = alertStatusMap.get(AlertValueAndStatus.VALUE_NAME);
      boolean fired = Boolean.parseBoolean(alertStatusMap.get(AlertValueAndStatus.FIRED_NAME));

View Full Code Here

Examples of org.apache.helix.manager.zk.ZKHelixDataAccessor

    new HealthStatsAggregationTask(cmResult._manager).run();
  //sleep for a few seconds to give stats stage time to trigger
    Thread.sleep(3000);


    // other verifications go here
    ZKHelixDataAccessor accessor = new ZKHelixDataAccessor(clusterName, new ZkBaseDataAccessor(_zkClient));
    Builder keyBuilder = accessor.keyBuilder();
    //for (int i = 0; i < 1; i++) //change 1 back to 5
    //{
      //String instance = "localhost_" + (12918 + i);
      //String instance = "localhost_12918";
      ZNRecord record = accessor.getProperty(keyBuilder.alertStatus()).getRecord();
      Map<String, Map<String,String>> recMap = record.getMapFields();
      Set<String> keySet = recMap.keySet();
      Map<String,String> alertStatusMap = recMap.get(_alertStatusStr);
      String val = alertStatusMap.get(AlertValueAndStatus.VALUE_NAME);
      boolean fired = Boolean.parseBoolean(alertStatusMap.get(AlertValueAndStatus.FIRED_NAME));

View Full Code Here

Examples of org.apache.helix.manager.zk.ZKHelixDataAccessor

                                                                                          clusterName));
    Assert.assertTrue(result);


    // drop alert soon after adding, but leave enough time for alert to fire once
    // Thread.sleep(3000);
    ZKHelixDataAccessor accessor = new ZKHelixDataAccessor(clusterName, new ZkBaseDataAccessor(_zkClient));
    Builder keyBuilder = accessor.keyBuilder();


    new HealthStatsAggregationTask(cmResult._manager).run();
    String instance = "localhost_12918";
    ZNRecord record = accessor.getProperty(keyBuilder.alertStatus()).getRecord();
    Map<String, Map<String, String>> recMap = record.getMapFields();
    Set<String> keySet = recMap.keySet();
    Assert.assertTrue(keySet.size() > 0);


    _setupTool.getClusterManagementTool().dropAlert(clusterName, _alertStr);
    new HealthStatsAggregationTask(cmResult._manager).run();
    // other verifications go here
    // for (int i = 0; i < 1; i++) //change 1 back to 5
    // {
    // String instance = "localhost_" + (12918 + i);
    record = accessor.getProperty(keyBuilder.alertStatus()).getRecord();
    recMap = record.getMapFields();
    keySet = recMap.keySet();
    Assert.assertEquals(keySet.size(), 0);
    // }

View Full Code Here

Examples of org.apache.helix.manager.zk.ZKHelixDataAccessor

      // OK
  }


  // verify leader node
  BaseDataAccessor<ZNRecord> baseAccessor = new ZkBaseDataAccessor<ZNRecord>(_gZkClient);
  HelixDataAccessor accessor = new ZKHelixDataAccessor(grandClusterName, baseAccessor);
  LiveInstance controllerLeader = accessor.getProperty(accessor.keyBuilder()
          .controllerLeader());
  Assert.assertNotNull(controllerLeader,
          "controllerLeader should be either controller_9000 or controller_9001");
  Assert.assertTrue(controllerLeader.getInstanceName().startsWith("controller_900"));


  accessor = new ZKHelixDataAccessor(clusterName, baseAccessor);
  LiveInstance leader = accessor.getProperty(accessor.keyBuilder().controllerLeader());
  for (int i = 0; i < 20; i++) {
      if (leader != null) {
    break;
      }
      Thread.sleep(200);
      leader = accessor.getProperty(accessor.keyBuilder().controllerLeader());
  }
  Assert.assertTrue(leader.getInstanceName().startsWith("controller_900"));


  boolean verifyResult = ClusterStateVerifier
          .verifyByZkCallback(new MasterNbInExtViewVerifier(ZK_ADDR, clusterName));

View Full Code Here

Examples of org.apache.helix.manager.zk.ZKHelixDataAccessor

  ClusterSetup.processCommandLineArgs(command.split("\\s+"));
  Thread.sleep(500);


  // save ideal state
  BaseDataAccessor<ZNRecord> baseAccessor = new ZkBaseDataAccessor<ZNRecord>(_gZkClient);
  HelixDataAccessor accessor = new ZKHelixDataAccessor(clusterName, baseAccessor);
  IdealState idealState = accessor.getProperty(accessor.keyBuilder().idealStates("db_11"));
  ZNRecordJsonSerializer serializer = new ZNRecordJsonSerializer();


  String tmpDir = System.getProperty("java.io.tmpdir");
  if (tmpDir == null) {
      tmpDir = "/tmp";
  }
  final String tmpIdealStateFile = tmpDir + "/" + clusterName + "_idealState.log";
  FileWriter fos = new FileWriter(tmpIdealStateFile);
  PrintWriter pw = new PrintWriter(fos);
  pw.write(new String(serializer.serialize(idealState.getRecord())));
  pw.close();


  command = "-zkSvr localhost:2183 -dropResource " + clusterName + " db_11 ";
  ClusterSetup.processCommandLineArgs(command.split("\\s+"));


  boolean verifyResult = ClusterStateVerifier
          .verifyByZkCallback(new BestPossAndExtViewZkVerifier(ZK_ADDR, clusterName));
  Assert.assertTrue(verifyResult);


  command = "-zkSvr localhost:2183 -addIdealState " + clusterName + " db_11 "
          + tmpIdealStateFile;
  ClusterSetup.processCommandLineArgs(command.split("\\s+"));


  verifyResult = ClusterStateVerifier.verifyByZkCallback(new BestPossAndExtViewZkVerifier(
          ZK_ADDR, clusterName));
  Assert.assertTrue(verifyResult);


  IdealState idealState2 = accessor.getProperty(accessor.keyBuilder().idealStates("db_11"));
  Assert.assertTrue(idealState2.getRecord().equals(idealState.getRecord()));


  // clean up
  // for (int i = 0; i < 2; i++) {
  // controllers[i].syncStop();

View Full Code Here

Examples of org.apache.helix.manager.zk.ZKHelixDataAccessor

  command = "-zkSvr localhost:2183 -swapInstance " + clusterName
          + " localhost_1232 localhost_12320";
  ClusterSetup.processCommandLineArgs(command.split("\\s+"));


  BaseDataAccessor<ZNRecord> baseAccessor = new ZkBaseDataAccessor<ZNRecord>(_gZkClient);
  HelixDataAccessor accessor = new ZKHelixDataAccessor(clusterName, baseAccessor);
  String path = accessor.keyBuilder().instanceConfig("localhost_1232").getPath();
  Assert.assertFalse(_gZkClient.exists(path), path
          + " should not exist since localhost_1232 has been swapped by localhost_12320");


  System.out.println("END " + clusterName + " at " + new Date(System.currentTimeMillis()));
    }

View Full Code Here

Examples of org.apache.helix.manager.zk.ZKHelixDataAccessor

  command = "-zkSvr localhost:2183 -activateCluster " + clusterName + " " + grandClusterName
          + " false";
  ClusterSetup.processCommandLineArgs(command.split("\\s+"));


  BaseDataAccessor<ZNRecord> baseAccessor = new ZkBaseDataAccessor<ZNRecord>(_gZkClient);
  HelixDataAccessor accessor = new ZKHelixDataAccessor(clusterName, baseAccessor);
  String path = accessor.keyBuilder().controllerLeader().getPath();
  for (int i = 0; i < 10; i++) {
      Thread.sleep(1000);
      if (!_gZkClient.exists(path)) {
    break;
      }

View Full Code Here

Examples of org.apache.helix.manager.zk.ZKHelixDataAccessor

      _resourceName = resourceName;
    }
    @Override
    public boolean verify()
    {
      HelixDataAccessor accessor = new ZKHelixDataAccessor( _clusterName, new ZkBaseDataAccessor(_client));
      Builder keyBuilder = accessor.keyBuilder();
      int numberOfPartitions = accessor.getProperty(keyBuilder.idealStates(_resourceName)).getRecord().getListFields().size();
      ClusterDataCache cache = new ClusterDataCache();
      cache.refresh(accessor);
      String masterValue = cache.getStateModelDef(cache.getIdealState(_resourceName).getStateModelDefRef()).getStatesPriorityList().get(0);
      int replicas = Integer.parseInt(cache.getIdealState(_resourceName).getReplicas());
      return verifyBalanceExternalView(accessor.getProperty(keyBuilder.externalView(_resourceName)).getRecord(), numberOfPartitions, masterValue, replicas, cache.getLiveInstances().size());
    }

View Full Code Here

0 1 2 3 4 5

TOP

All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.