Examples of ZkClient


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

  public static ZkClient getZkClient(String zkServer)
  {
    // happy path that we cache the zkclient and it's still connected
    if (_zkClientMap.containsKey(zkServer))
    {
      ZkClient zkClient = _zkClientMap.get(zkServer);
      if (zkClient.getConnection().getZookeeperState() == States.CONNECTED)
      {
        return zkClient;
      }
    }

    synchronized (_zkClientMap)
    {
      // if we cache a stale zkclient, purge it
      if (_zkClientMap.containsKey(zkServer))
      {
        ZkClient zkClient = _zkClientMap.get(zkServer);
        if (zkClient.getConnection().getZookeeperState() != States.CONNECTED)
        {
          _zkClientMap.remove(zkServer);
        }
      }

      // get a new zkclient
      if (!_zkClientMap.containsKey(zkServer))
      {
        ZkClient zkClient = new ZkClient(zkServer, DEFAULT_SESSION_TIMEOUT,
            ZkClient.DEFAULT_CONNECTION_TIMEOUT, new ZNRecordSerializer());

        _zkClientMap.put(zkServer, zkClient);
      }
      return _zkClientMap.get(zkServer);
View Full Code Here

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

                                                             String instanceName,
                                                             String resourceGroup) throws JsonGenerationException,
      JsonMappingException,
      IOException
  {
    ZkClient zkClient = (ZkClient)getRequest().getAttributes().get(RestAdminApplication.ZKCLIENT);
    String instanceSessionId =
        ClusterRepresentationUtil.getInstanceSessionId(zkClient,
                                                       clusterName,
                                                       instanceName);
    Builder keyBuilder = new PropertyKey.Builder(clusterName);
View Full Code Here

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

  StringRepresentation getClustersRepresentation() throws JsonGenerationException,
      JsonMappingException,
      IOException
  {
    ZkClient zkClient =
        (ZkClient) getContext().getAttributes().get(RestAdminApplication.ZKCLIENT);
    ClusterSetup setupTool = new ClusterSetup(zkClient);
    List<String> clusters = setupTool.getClusterManagementTool().getClusters();

    ZNRecord clustersRecord = new ZNRecord("Clusters Summary");
View Full Code Here

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

  public void post(String zkServer,
                   Message message,
                   String clusterName,
                   String instanceName)
  {
    ZkClient client = new ZkClient(zkServer);
    client.setZkSerializer(new ZNRecordSerializer());
    String path = HelixUtil.getMessagePath(clusterName, instanceName) + "/" + message.getId();
    client.delete(path);
    ZNRecord record = client.readData(HelixUtil.getLiveInstancePath(clusterName, instanceName));
    message.setTgtSessionId(record.getSimpleField(LiveInstanceProperty.SESSION_ID.toString()).toString());
    message.setTgtName(record.getId());
    //System.out.println(message);
    client.createPersistent(path, message.getRecord());
  }
View Full Code Here

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

      }
      else if (command.equalsIgnoreCase(ClusterSetup.addCluster))
      {
        jsonParameters.verifyCommand(ClusterSetup.addCluster);

        ZkClient zkClient =
            (ZkClient) getContext().getAttributes().get(RestAdminApplication.ZKCLIENT);
        ClusterSetup setupTool = new ClusterSetup(zkClient);
        setupTool.addCluster(jsonParameters.getParameter(JsonParameters.CLUSTER_NAME),
                             false);
      }
View Full Code Here

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

    final String zkAddr = args[0];
    final int numNodes = Integer.parseInt(args[1]);
    final String clusterName = DEFAULT_CLUSTER_NAME;

    ZkClient zkclient = null;
    try
    {
      zkclient = new ZkClient(zkAddr, ZkClient.DEFAULT_SESSION_TIMEOUT,
          ZkClient.DEFAULT_CONNECTION_TIMEOUT, new ZNRecordSerializer());
      ZKHelixAdmin admin = new ZKHelixAdmin(zkclient);

      // add cluster
      admin.addCluster(clusterName, true);

      // add state model definition
      StateModelConfigGenerator generator = new StateModelConfigGenerator();
      admin.addStateModelDef(clusterName, DEFAULT_STATE_MODEL,
          new StateModelDefinition(generator.generateConfigForOnlineOffline()));
      // addNodes
      for (int i = 0; i < numNodes; i++)
      {
        String port = "" +(12001+ i);
        String serverId = "localhost_"+ port;
        InstanceConfig config = new InstanceConfig(serverId);
        config.setHostName("localhost");
        config.setPort(port);
        config.setInstanceEnabled(true);
        admin.addInstance(clusterName, config);
      }
      // add resource "repository" which has 1 partition
      String resourceName = DEFAULT_RESOURCE_NAME;
      admin.addResource(clusterName, resourceName, DEFAULT_PARTITION_NUMBER,
          DEFAULT_STATE_MODEL, IdealStateModeProperty.AUTO.toString());
      admin.rebalance(clusterName, resourceName, 1);

    } finally
    {
      if (zkclient != null)
      {
        zkclient.close();
      }
    }
  }
View Full Code Here

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

  }
 
  StringRepresentation getStateModelsRepresentation() throws JsonGenerationException, JsonMappingException, IOException
  {
    String clusterName = (String)getRequest().getAttributes().get("clusterName");
    ZkClient zkClient = (ZkClient)getContext().getAttributes().get(RestAdminApplication.ZKCLIENT);
    ClusterSetup setupTool = new ClusterSetup(zkClient);
   
    List<String> models = setupTool.getClusterManagementTool().getStateModelDefs(clusterName);
   
    ZNRecord modelDefinitions = new ZNRecord("modelDefinitions");
View Full Code Here

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

  public void acceptRepresentation(Representation entity)
  {
    try
    {
      String clusterName = (String)getRequest().getAttributes().get("clusterName");
      ZkClient zkClient = (ZkClient)getContext().getAttributes().get(RestAdminApplication.ZKCLIENT);;
     
      JsonParameters jsonParameters = new JsonParameters(entity);
      String command = jsonParameters.getCommand();

       
View Full Code Here

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

      IOException
  {
    String clusterName = (String) getRequest().getAttributes().get("clusterName");
    String instanceName = (String) getRequest().getAttributes().get("instanceName");
    Builder keyBuilder = new PropertyKey.Builder(clusterName);
    ZkClient zkClient =
        (ZkClient) getContext().getAttributes().get(RestAdminApplication.ZKCLIENT);

    String message =
        ClusterRepresentationUtil.getClusterPropertyAsString(zkClient,
                                                             clusterName,
View Full Code Here

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

        jsonParameters.verifyCommand(ClusterSetup.enableInstance);

        boolean enabled =
            Boolean.parseBoolean(jsonParameters.getParameter(JsonParameters.ENABLED));

        ZkClient zkClient =
            (ZkClient) getContext().getAttributes().get(RestAdminApplication.ZKCLIENT);
        ClusterSetup setupTool = new ClusterSetup(zkClient);
        setupTool.getClusterManagementTool().enableInstance(clusterName,
                                                            instanceName,
                                                            enabled);
      }
      else if (command.equalsIgnoreCase(ClusterSetup.enablePartition))
      {
        jsonParameters.verifyCommand(ClusterSetup.enablePartition);
        boolean enabled =
             Boolean.parseBoolean(jsonParameters.getParameter(JsonParameters.ENABLED));

        String[] partitions =
            jsonParameters.getParameter(JsonParameters.PARTITION).split(";");
        String resource =
            jsonParameters.getParameter(JsonParameters.RESOURCE);

        ZkClient zkClient =
            (ZkClient) getContext().getAttributes().get(RestAdminApplication.ZKCLIENT);
        ClusterSetup setupTool = new ClusterSetup(zkClient);
        setupTool.getClusterManagementTool().enablePartition(enabled,
                                                             clusterName,
                                                             instanceName,
                                                             resource,
                                                             Arrays.asList(partitions));
      }
      else if (command.equalsIgnoreCase(ClusterSetup.resetPartition))
      {
        jsonParameters.verifyCommand(ClusterSetup.resetPartition);

        String resource =
            jsonParameters.getParameter(JsonParameters.RESOURCE);

        ZkClient zkClient =
            (ZkClient) getContext().getAttributes().get(RestAdminApplication.ZKCLIENT);
        ClusterSetup setupTool = new ClusterSetup(zkClient);
        String[] partitionNames =
            jsonParameters.getParameter(JsonParameters.PARTITION).split("\\s+");
        setupTool.getClusterManagementTool()
                 .resetPartition(clusterName,
                                 instanceName,
                                 resource,
                                 Arrays.asList(partitionNames));
      }
      else if (command.equalsIgnoreCase(ClusterSetup.resetInstance))
      {
        jsonParameters.verifyCommand(ClusterSetup.resetInstance);

        ZkClient zkClient =
            (ZkClient) getContext().getAttributes().get(RestAdminApplication.ZKCLIENT);
        ClusterSetup setupTool = new ClusterSetup(zkClient);
        setupTool.getClusterManagementTool().resetInstance(clusterName,
                                                           Arrays.asList(instanceName));
      }
View Full Code Here
TOP
Copyright © 2018 www.massapi.com. All rights reserved.
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.