Search in sources :

Example 1 with GridTestJob

use of org.apache.ignite.GridTestJob in project ignite by apache.

the class GridRoundRobinLoadBalancingSpiMultipleNodesSelfTest method testMultipleTasks.

/**
     * @throws Exception If test failed.
     */
@SuppressWarnings({ "ObjectEquality" })
public void testMultipleTasks() throws Exception {
    ComputeTaskSession ses1 = new GridTestTaskSession(IgniteUuid.randomUuid());
    ComputeTaskSession ses2 = new GridTestTaskSession(IgniteUuid.randomUuid());
    List<ClusterNode> allNodes = (List<ClusterNode>) getSpiContext().nodes();
    // Initialize.
    getSpi().getBalancedNode(ses1, allNodes, new GridTestJob());
    getSpi().getBalancedNode(ses2, allNodes, new GridTestJob());
    List<UUID> orderedNodes1 = getSpi().getNodeIds(ses1);
    List<UUID> orderedNodes2 = getSpi().getNodeIds(ses2);
    assert orderedNodes1 != orderedNodes2;
    // Check the round-robin actually did circle.
    for (int i = 0; i < allNodes.size(); i++) {
        ClusterNode node1 = getSpi().getBalancedNode(ses1, allNodes, new GridTestJob());
        assert orderedNodes1.get(i) == node1.id();
        ClusterNode node2 = getSpi().getBalancedNode(ses2, allNodes, new GridTestJob());
        assert orderedNodes2.get(i) == node2.id();
        assert orderedNodes1.get(i) == orderedNodes2.get(i);
    }
    // Double-check.
    for (int i = 0; i < allNodes.size(); i++) {
        ClusterNode node1 = getSpi().getBalancedNode(ses1, allNodes, new GridTestJob());
        assert orderedNodes1.get(i) == node1.id();
        ClusterNode node2 = getSpi().getBalancedNode(ses2, allNodes, new GridTestJob());
        assert orderedNodes2.get(i) == node2.id();
        assert orderedNodes1.get(i) == orderedNodes2.get(i);
    }
    getSpiContext().triggerEvent(new TaskEvent(null, null, EVT_TASK_FINISHED, ses1.getId(), null, null, false, null));
    getSpiContext().triggerEvent(new TaskEvent(null, null, EVT_TASK_FAILED, ses2.getId(), null, null, false, null));
}
Also used : ClusterNode(org.apache.ignite.cluster.ClusterNode) GridTestTaskSession(org.apache.ignite.GridTestTaskSession) TaskEvent(org.apache.ignite.events.TaskEvent) ArrayList(java.util.ArrayList) List(java.util.List) UUID(java.util.UUID) GridTestJob(org.apache.ignite.GridTestJob) ComputeTaskSession(org.apache.ignite.compute.ComputeTaskSession)

Example 2 with GridTestJob

use of org.apache.ignite.GridTestJob in project ignite by apache.

the class GridAdaptiveLoadBalancingSpiSelfTest method testSingleNodeZeroWeight.

/**
     * @throws Exception If failed.
     */
@SuppressWarnings({ "ObjectEquality" })
public void testSingleNodeZeroWeight() throws Exception {
    GridTestNode node = (GridTestNode) getSpiContext().nodes().iterator().next();
    node.addAttribute("load", 0d);
    List<ClusterNode> nodes = Collections.singletonList((ClusterNode) node);
    ComputeTaskSession ses = new GridTestTaskSession(IgniteUuid.randomUuid());
    GridTestNode pick1 = (GridTestNode) getSpi().getBalancedNode(ses, nodes, new GridTestJob());
    pick1.setAttribute("used", true);
    assert nodes.contains(pick1);
    // Verify that same instance is returned every time.
    ClusterNode pick2 = getSpi().getBalancedNode(ses, nodes, new GridTestJob());
    assert pick1 == pick2;
}
Also used : ClusterNode(org.apache.ignite.cluster.ClusterNode) GridTestTaskSession(org.apache.ignite.GridTestTaskSession) GridTestJob(org.apache.ignite.GridTestJob) GridTestNode(org.apache.ignite.testframework.GridTestNode) ComputeTaskSession(org.apache.ignite.compute.ComputeTaskSession)

Example 3 with GridTestJob

use of org.apache.ignite.GridTestJob in project ignite by apache.

the class GridAdaptiveLoadBalancingSpiSelfTest method testSingleNodeSameSession.

/**
     * @throws Exception If failed.
     */
@SuppressWarnings({ "ObjectEquality" })
public void testSingleNodeSameSession() throws Exception {
    GridTestNode node = (GridTestNode) getSpiContext().nodes().iterator().next();
    node.addAttribute("load", 1d);
    List<ClusterNode> nodes = Collections.singletonList((ClusterNode) node);
    ComputeTaskSession ses = new GridTestTaskSession(IgniteUuid.randomUuid());
    GridTestNode pick1 = (GridTestNode) getSpi().getBalancedNode(ses, nodes, new GridTestJob());
    pick1.setAttribute("used", true);
    assert nodes.contains(pick1);
    // Verify that same instance is returned every time.
    ClusterNode pick2 = getSpi().getBalancedNode(ses, nodes, new GridTestJob());
    assert pick1 == pick2;
}
Also used : ClusterNode(org.apache.ignite.cluster.ClusterNode) GridTestTaskSession(org.apache.ignite.GridTestTaskSession) GridTestJob(org.apache.ignite.GridTestJob) GridTestNode(org.apache.ignite.testframework.GridTestNode) ComputeTaskSession(org.apache.ignite.compute.ComputeTaskSession)

Example 4 with GridTestJob

use of org.apache.ignite.GridTestJob in project ignite by apache.

the class GridRoundRobinLoadBalancingSpiNotPerTaskSelfTest method testBalancingOneNode.

/**
     * @throws Exception If test failed.
     */
public void testBalancingOneNode() throws Exception {
    ComputeTaskSession ses = new GridTestTaskSession();
    List<ClusterNode> allNodes = (List<ClusterNode>) getSpiContext().nodes();
    List<ClusterNode> balancedNode = Arrays.asList(allNodes.get(0));
    ClusterNode firstNode = getSpi().getBalancedNode(ses, balancedNode, new GridTestJob());
    ClusterNode secondNode = getSpi().getBalancedNode(ses, balancedNode, new GridTestJob());
    assertEquals(firstNode, secondNode);
}
Also used : ClusterNode(org.apache.ignite.cluster.ClusterNode) GridTestTaskSession(org.apache.ignite.GridTestTaskSession) ArrayList(java.util.ArrayList) List(java.util.List) GridTestJob(org.apache.ignite.GridTestJob) ComputeTaskSession(org.apache.ignite.compute.ComputeTaskSession)

Example 5 with GridTestJob

use of org.apache.ignite.GridTestJob in project ignite by apache.

the class GridRoundRobinTestUtils method checkCyclicBalancing.

/**
     * Performs two full cycles by round robin routine for check correct order.
     *
     * @param spi Load balancing SPI.
     * @param allNodes Topology nodes.
     * @param orderedNodes Balancing nodes.
     * @param ses Task session.
     */
static void checkCyclicBalancing(RoundRobinLoadBalancingSpi spi, List<ClusterNode> allNodes, List<UUID> orderedNodes, ComputeTaskSession ses) {
    ClusterNode firstNode = spi.getBalancedNode(ses, allNodes, new GridTestJob());
    int startIdx = firstBalancedNodeIndex(firstNode, orderedNodes);
    // Two full cycles by round robin routine.
    for (int i = 0; i < allNodes.size() * 2; i++) {
        int actualIdx = (startIdx + i + 1) % allNodes.size();
        ClusterNode nextNode = spi.getBalancedNode(ses, allNodes, new GridTestJob());
        assertEquals("Balancer returns node out of order", nextNode.id(), orderedNodes.get(actualIdx));
    }
}
Also used : ClusterNode(org.apache.ignite.cluster.ClusterNode) GridTestJob(org.apache.ignite.GridTestJob)

Aggregations

GridTestJob (org.apache.ignite.GridTestJob)15 ClusterNode (org.apache.ignite.cluster.ClusterNode)15 GridTestTaskSession (org.apache.ignite.GridTestTaskSession)14 GridTestNode (org.apache.ignite.testframework.GridTestNode)9 ComputeTaskSession (org.apache.ignite.compute.ComputeTaskSession)8 ArrayList (java.util.ArrayList)6 List (java.util.List)5 UUID (java.util.UUID)3 HashMap (java.util.HashMap)1 Map (java.util.Map)1 AtomicInteger (java.util.concurrent.atomic.AtomicInteger)1 IgniteException (org.apache.ignite.IgniteException)1 TaskEvent (org.apache.ignite.events.TaskEvent)1