Examples of org.apache.helix.model.ClusterConstraints

org.apache.helix.model.ClusterConstraints

    ZKHelixDataAccessor accessor =
        new ZKHelixDataAccessor(clusterName, new ZkBaseDataAccessor(_gZkClient));
    Builder keyBuilder = accessor.keyBuilder();


    accessor.setProperty(keyBuilder.constraint(ConstraintType.STATE_CONSTRAINT.toString()),
                         new ClusterConstraints(record));


    record =
        accessor.getProperty(keyBuilder.constraint(ConstraintType.STATE_CONSTRAINT.toString()))
                .getRecord();
    ClusterConstraints constraint = new ClusterConstraints(record);
    // System.out.println("constraint: " + constraint);


    // state1: hit rule2
    Map<ConstraintAttribute, String> stateAttr1 =
        new HashMap<ConstraintAttribute, String>();
    stateAttr1.put(ConstraintAttribute.STATE, "MASTER");
    stateAttr1.put(ConstraintAttribute.RESOURCE, "TestDB");


    Set<ConstraintItem> matches = constraint.match(stateAttr1);
    System.out.println(stateAttr1 + " matches(" + matches.size() + "): " + matches);
    Assert.assertEquals(matches.size(), 3);
    Assert.assertTrue(contains(matches, constraint0));
    Assert.assertTrue(contains(matches, constraint1));
    Assert.assertTrue(contains(matches, constraint2));


    // matches = selectConstraints(matches, stateAttr1);
    // System.out.println(stateAttr1 + " matches(" + matches.size() + "): " + matches);
    // Assert.assertEquals(matches.size(), 2);
    // Assert.assertTrue(contains(matches, constraint0));
    // Assert.assertTrue(contains(matches, constraint1));


    // state2: not hit any rules
    Map<ConstraintAttribute, String> stateAttr2 =
        new HashMap<ConstraintAttribute, String>();
    stateAttr2.put(ConstraintAttribute.STATE, "MASTER");
    stateAttr2.put(ConstraintAttribute.RESOURCE, "MyDB");


    matches = constraint.match(stateAttr2);
    System.out.println(stateAttr2 + " matches(" + matches.size() + "): " + matches);
    Assert.assertEquals(matches.size(), 2);
    Assert.assertTrue(contains(matches, constraint0));
    Assert.assertTrue(contains(matches, constraint2));

View Full Code Here

    builder.addConstraintAttribute(attribute, value);
    return this;
  }
  
  public ClusterConstraints build() {
    ClusterConstraints constraints = new ClusterConstraints(_constraintType);
    
    for (String constraintId : _constraintBuilderMap.keySet()) {
      ConstraintItemBuilder builder = _constraintBuilderMap.get(constraintId);
      constraints.addConstraintItem(constraintId, builder.build());
    }
    return constraints;
  }

View Full Code Here

    record.getMapField("constraint5").put("CONSTRAINT_VALUE", "3");
    ConstraintItem constraint5 = new ConstraintItem(record.getMapField("constraint5"));


    Builder keyBuilder = accessor.keyBuilder();
    accessor.setProperty(keyBuilder.constraint(ConstraintType.MESSAGE_CONSTRAINT.toString()),
                         new ClusterConstraints(record));


    // ClusterConstraints constraint =
    // accessor.getProperty(ClusterConstraints.class,
    // PropertyType.CONFIGS,
    // ConfigScopeProperty.CONSTRAINT.toString(),
    // ConstraintType.MESSAGE_CONSTRAINT.toString());
    ClusterConstraints constraint =
        accessor.getProperty(keyBuilder.constraint(ConstraintType.MESSAGE_CONSTRAINT.toString()));


    MessageThrottleStage throttleStage = new MessageThrottleStage();


    // test constraintSelection
    // message1: hit contraintSelection rule1 and rule2
    Message msg1 =
        createMessage(MessageType.STATE_TRANSITION,
                      "msgId-001",
                      "OFFLINE",
                      "SLAVE",
                      "TestDB",
                      "localhost_0");


    Map<ConstraintAttribute, String> msgAttr =
        ClusterConstraints.toConstraintAttributes(msg1);
    Set<ConstraintItem> matches = constraint.match(msgAttr);
    System.out.println(msg1 + " matches(" + matches.size() + "): " + matches);
    Assert.assertEquals(matches.size(), 5);
    Assert.assertTrue(containsConstraint(matches, constraint0));
    Assert.assertTrue(containsConstraint(matches, constraint1));
    Assert.assertTrue(containsConstraint(matches, constraint2));
    Assert.assertTrue(containsConstraint(matches, constraint4));
    Assert.assertTrue(containsConstraint(matches, constraint5));


    matches = throttleStage.selectConstraints(matches, msgAttr);
    System.out.println(msg1 + " matches(" + matches.size() + "): " + matches);
    Assert.assertEquals(matches.size(), 2);
    Assert.assertTrue(containsConstraint(matches, constraint1));
    Assert.assertTrue(containsConstraint(matches, constraint5));


    // message2: hit contraintSelection rule1, rule2, and rule3
    Message msg2 =
        createMessage(MessageType.STATE_TRANSITION,
                      "msgId-002",
                      "OFFLINE",
                      "SLAVE",
                      "TestDB",
                      "localhost_1");


    msgAttr = ClusterConstraints.toConstraintAttributes(msg2);
    matches = constraint.match(msgAttr);
    System.out.println(msg2 + " matches(" + matches.size() + "): " + matches);
    Assert.assertEquals(matches.size(), 5);
    Assert.assertTrue(containsConstraint(matches, constraint0));
    Assert.assertTrue(containsConstraint(matches, constraint1));
    Assert.assertTrue(containsConstraint(matches, constraint2));

View Full Code Here

          + ". Requires ClusterDataCache|RESOURCES|MESSAGES_SELECTED");
    }


    MessageThrottleStageOutput output = new MessageThrottleStageOutput();


        ClusterConstraints constraint = cache.getConstraint(ConstraintType.MESSAGE_CONSTRAINT);
    Map<String, Integer> throttleCounterMap = new HashMap<String, Integer>();


    if (constraint != null)
    {
      // go through all pending messages, they should be counted but not throttled

View Full Code Here

      ZKHelixAdmin tool = new ZKHelixAdmin(_gZkClient);
      tool.addCluster(clusterName, true);
      Assert.assertTrue(ZKUtil.isClusterSetup(clusterName, _gZkClient), "Cluster should be setup");


      // test admin.getMessageConstraints()
      ClusterConstraints constraints = tool.getConstraints(clusterName, ConstraintType.MESSAGE_CONSTRAINT);
      Assert.assertNull(constraints, "message-constraint should NOT exist for cluster: " + className);


      // remove non-exist constraint
      try {
        tool.removeConstraint(clusterName, ConstraintType.MESSAGE_CONSTRAINT, "constraint1");
        // will leave a null message-constraint znode on zk
      } catch (Exception e) {
        Assert.fail("Should not throw exception when remove a non-exist constraint.");
      }


      // add a message constraint
      ConstraintItemBuilder builder = new ConstraintItemBuilder();
      builder.addConstraintAttribute(ConstraintAttribute.RESOURCE.toString(), "MyDB")
             .addConstraintAttribute(ConstraintAttribute.CONSTRAINT_VALUE.toString(), "1");
      tool.setConstraint(clusterName, ConstraintType.MESSAGE_CONSTRAINT, "constraint1", builder.build());


      HelixDataAccessor accessor = new ZKHelixDataAccessor(clusterName, new ZkBaseDataAccessor<ZNRecord>(_gZkClient));
      PropertyKey.Builder keyBuilder = new PropertyKey.Builder(clusterName);
      constraints = accessor.getProperty(keyBuilder.constraint(ConstraintType.MESSAGE_CONSTRAINT.toString()));
      Assert.assertNotNull(constraints, "message-constraint should exist");
      ConstraintItem item = constraints.getConstraintItem("constraint1");
      Assert.assertNotNull(item, "message-constraint for constraint1 should exist");
      Assert.assertEquals(item.getConstraintValue(), "1");
      Assert.assertEquals(item.getAttributeValue(ConstraintAttribute.RESOURCE), "MyDB");
      
      // test admin.getMessageConstraints()
      constraints = tool.getConstraints(clusterName, ConstraintType.MESSAGE_CONSTRAINT);
      Assert.assertNotNull(constraints, "message-constraint should exist");
      item = constraints.getConstraintItem("constraint1");
      Assert.assertNotNull(item, "message-constraint for constraint1 should exist");
      Assert.assertEquals(item.getConstraintValue(), "1");
      Assert.assertEquals(item.getAttributeValue(ConstraintAttribute.RESOURCE), "MyDB");
      


      // remove a exist message-constraint
      tool.removeConstraint(clusterName, ConstraintType.MESSAGE_CONSTRAINT, "constraint1");
      constraints = accessor.getProperty(keyBuilder.constraint(ConstraintType.MESSAGE_CONSTRAINT.toString()));
      Assert.assertNotNull(constraints, "message-constraint should exist");
      item = constraints.getConstraintItem("constraint1");
      Assert.assertNull(item, "message-constraint for constraint1 should NOT exist");


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

View Full Code Here

    if (clusterName == null || constraintType == null) {
      throw new IllegalArgumentException("fail to get constraint. missing clusterName|constraintType");
    }


    ConstraintType type = ConstraintType.valueOf(constraintType);
    ClusterConstraints constraints = _admin.getConstraints(clusterName, type);
    ZNRecordSerializer serializer = new ZNRecordSerializer();
    return new String(serializer.serialize(constraints.getRecord()));
  }

View Full Code Here

    baseAccessor.update(path, new DataUpdater<ZNRecord>()
    {
      @Override
      public ZNRecord update(ZNRecord currentData)
      {
        ClusterConstraints constraints = currentData == null? 
            new ClusterConstraints(constraintType) : new ClusterConstraints(currentData);


        constraints.addConstraintItem(constraintId, constraintItem);
        return constraints.getRecord();
      }
    }, AccessOption.PERSISTENT);
  }

View Full Code Here

    {
      @Override
      public ZNRecord update(ZNRecord currentData)
      {
        if (currentData != null) {
          ClusterConstraints constraints = new ClusterConstraints(currentData);


          constraints.removeConstraintItem(constraintId);
          return constraints.getRecord();
        }
        return null;
      }
    }, AccessOption.PERSISTENT);
  }

View Full Code Here

0 1 2 3 4 5

TOP

Related Classes of org.apache.helix.model.ClusterConstraints

org.apache.helix.api.accessor.ClusterAccessor

org.apache.helix.api.config.ClusterConfig

org.apache.helix.api.config.ClusterConfig$Builder

org.apache.helix.api.config.ClusterConfig$Delta

org.apache.helix.controller.GenericHelixController

org.apache.helix.controller.stages.MessageThrottleStage

org.apache.helix.controller.stages.TestMessageThrottleStage

org.apache.helix.manager.zk.TestZkHelixAdmin

org.apache.helix.manager.zk.ZKHelixAdmin

org.apache.helix.model.builder.ClusterConstraintsBuilder

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.