コード例 #1
0
  @Test
  public void test() {
    InMemScheduleConext context = new InMemScheduleConext();

    StreamPartition partition = new StreamPartition();
    partition.setType(StreamPartition.Type.GLOBAL);
    partition.setStreamId("s1");
    partition.setColumns(Arrays.asList("f1", "f2"));

    StreamGroup group = new StreamGroup();
    group.addStreamPartition(partition);

    {
      TestTopologyMgmtService mgmtService = new TestTopologyMgmtService(3, 3, "prefix-time1", true);
      SameTopologySlotStrategy strategy = new SameTopologySlotStrategy(context, group, mgmtService);
      List<WorkSlot> slots = strategy.reserveWorkSlots(5, false, new HashMap<String, Object>());
      Assert.assertEquals(0, slots.size());
      Assert.assertEquals(1, context.getTopologies().size());
    }

    {
      TestTopologyMgmtService mgmtService = new TestTopologyMgmtService(5, 5, "prefix-time2", true);
      SameTopologySlotStrategy strategy = new SameTopologySlotStrategy(context, group, mgmtService);
      List<WorkSlot> slots = strategy.reserveWorkSlots(5, false, new HashMap<String, Object>());
      Assert.assertEquals(5, slots.size());
      LOG.info(slots.get(0).getTopologyName());
      Assert.assertEquals(2, context.getTopologies().size());
      Assert.assertEquals(2, context.getTopologyUsages().size());

      // assert all on same topology
      for (WorkSlot ws : slots) {
        Assert.assertEquals(slots.get(0).getTopologyName(), ws.getTopologyName());
      }
      Iterator<TopologyUsage> it = context.getTopologyUsages().values().iterator();
      TopologyUsage usage = it.next();
      for (AlertBoltUsage u : usage.getAlertUsages().values()) {
        Assert.assertTrue(u.getPartitions().size() == 0);
        Assert.assertTrue(u.getQueueSize() == 0);
      }
      // assert
      usage = it.next();
      for (AlertBoltUsage u : usage.getAlertUsages().values()) {
        LOG.info(u.getBoltId());
        Assert.assertTrue(u.getPartitions().size() == 0);
        Assert.assertTrue(u.getBoltId(), u.getQueueSize() == 0);
      }
    }
  }
コード例 #2
0
  @SuppressWarnings("unused")
  @Test
  public void test2_overlap() {
    InMemScheduleConext context = new InMemScheduleConext();

    StreamPartition partition = new StreamPartition();
    partition.setType(StreamPartition.Type.GLOBAL);
    partition.setStreamId("s1");
    partition.setColumns(Arrays.asList("f1", "f2"));
    StreamGroup sg = new StreamGroup();
    sg.addStreamPartition(partition, false);

    MonitoredStream ms1 = new MonitoredStream(sg);

    TestTopologyMgmtService mgmtService = new TestTopologyMgmtService(5, 5, "prefix-3", true);

    String topo1 = null;
    String bolt1 = null;
    WorkQueueBuilder wrb = new WorkQueueBuilder(context, mgmtService);
    StreamWorkSlotQueue queue = wrb.createQueue(ms1, false, 5, new HashMap<String, Object>());
    {
      Assert.assertEquals(5, queue.getWorkingSlots().size());
      topo1 = queue.getWorkingSlots().get(0).getTopologyName();
      bolt1 = queue.getWorkingSlots().get(0).getBoltId();
      Assert.assertEquals(1, context.getTopologies().size());
      Assert.assertEquals(1, context.getTopologyUsages().size());
      LOG.info(queue.getWorkingSlots().get(0).getTopologyName());
      for (WorkSlot ws : queue.getWorkingSlots()) {
        Assert.assertEquals(topo1, ws.getTopologyName());
      }

      TopologyUsage usage = context.getTopologyUsages().values().iterator().next();
      for (AlertBoltUsage u : usage.getAlertUsages().values()) {
        Assert.assertTrue(u.getPartitions().size() > 0);
        Assert.assertTrue(u.getBoltId(), u.getQueueSize() > 0);
      }
    }

    // second partition
    StreamPartition partition2 = new StreamPartition();
    partition2.setType(StreamPartition.Type.GLOBAL);
    partition2.setStreamId("s2");
    partition2.setColumns(Arrays.asList("f1", "f2"));

    StreamGroup sg2 = new StreamGroup();
    sg2.addStreamPartition(partition2);
    MonitoredStream ms2 = new MonitoredStream(sg2);
    queue = wrb.createQueue(ms2, false, 5, new HashMap<String, Object>());
    {
      Assert.assertEquals(5, queue.getWorkingSlots().size());
      Assert.assertEquals(2, context.getTopologies().size());
      Assert.assertEquals(2, context.getTopologyUsages().size());

      String topo2 = queue.getWorkingSlots().get(0).getTopologyName();
      String bolt2 = queue.getWorkingSlots().get(0).getBoltId();
      for (WorkSlot ws : queue.getWorkingSlots()) {
        Assert.assertEquals(topo2, ws.getTopologyName());
      }
      Assert.assertNotEquals(topo1, topo2);
    }
  }