| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You under the Apache License, Version 2.0 |
| * (the "License"); you may not use this file except in compliance with |
| * the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| package org.apache.ignite.internal; |
| |
| import java.io.Serializable; |
| import java.util.Arrays; |
| import java.util.Collections; |
| import java.util.List; |
| import java.util.Map; |
| import java.util.concurrent.atomic.AtomicBoolean; |
| import org.apache.ignite.Ignite; |
| import org.apache.ignite.IgniteException; |
| import org.apache.ignite.cluster.ClusterGroup; |
| import org.apache.ignite.cluster.ClusterNode; |
| import org.apache.ignite.cluster.ClusterTopologyException; |
| import org.apache.ignite.compute.ComputeJob; |
| import org.apache.ignite.compute.ComputeJobAdapter; |
| import org.apache.ignite.compute.ComputeJobResult; |
| import org.apache.ignite.compute.ComputeJobResultPolicy; |
| import org.apache.ignite.compute.ComputeTask; |
| import org.apache.ignite.compute.ComputeTaskSession; |
| import org.apache.ignite.compute.ComputeTaskSessionFullSupport; |
| import org.apache.ignite.compute.ComputeUserUndeclaredException; |
| import org.apache.ignite.configuration.IgniteConfiguration; |
| import org.apache.ignite.lang.IgnitePredicate; |
| import org.apache.ignite.resources.TaskSessionResource; |
| import org.apache.ignite.spi.failover.FailoverContext; |
| import org.apache.ignite.spi.failover.always.AlwaysFailoverSpi; |
| import org.apache.ignite.testframework.junits.common.GridCommonAbstractTest; |
| import org.apache.ignite.testframework.junits.common.GridCommonTest; |
| import org.jetbrains.annotations.NotNull; |
| import org.junit.Test; |
| |
| /** |
| * Test failover of a task with Node filter predicate. |
| */ |
| @GridCommonTest(group = "Kernal Self") |
| public class GridFailoverTaskWithPredicateSelfTest extends GridCommonAbstractTest { |
| /** First node's name. */ |
| private static final String NODE1 = "NODE1"; |
| |
| /** Second node's name. */ |
| private static final String NODE2 = "NODE2"; |
| |
| /** Third node's name. */ |
| private static final String NODE3 = "NODE3"; |
| |
| /** Predicate to exclude the second node from topology */ |
| private final IgnitePredicate<ClusterNode> p = new IgnitePredicate<ClusterNode>() { |
| @Override public boolean apply(ClusterNode e) { |
| return !NODE2.equals(e.attribute(IgniteNodeAttributes.ATTR_IGNITE_INSTANCE_NAME)); |
| } |
| }; |
| |
| /** Whether delegating fail over node was found or not. */ |
| private final AtomicBoolean routed = new AtomicBoolean(); |
| |
| /** Whether job execution failed with exception. */ |
| private final AtomicBoolean failed = new AtomicBoolean(); |
| |
| /** {@inheritDoc} */ |
| @Override protected IgniteConfiguration getConfiguration(String igniteInstanceName) throws Exception { |
| IgniteConfiguration cfg = super.getConfiguration(igniteInstanceName); |
| |
| cfg.setFailoverSpi(new AlwaysFailoverSpi() { |
| /** {@inheritDoc} */ |
| @Override public ClusterNode failover(FailoverContext ctx, List<ClusterNode> grid) { |
| ClusterNode failoverNode = super.failover(ctx, grid); |
| |
| if (failoverNode != null) |
| routed.set(true); |
| else |
| routed.set(false); |
| |
| return failoverNode; |
| } |
| }); |
| |
| return cfg; |
| } |
| |
| /** |
| * Tests that failover doesn't happen on two-node grid when the Task is applicable only for the first node |
| * and fails on it. |
| * |
| * @throws Exception If failed. |
| */ |
| @Test |
| public void testJobNotFailedOver() throws Exception { |
| failed.set(false); |
| routed.set(false); |
| |
| try { |
| Ignite ignite1 = startGrid(NODE1); |
| Ignite ignite2 = startGrid(NODE2); |
| |
| assert ignite1 != null; |
| assert ignite2 != null; |
| |
| compute(ignite1.cluster().forPredicate(p)).withTimeout(10000).execute(JobFailTask.class.getName(), "1"); |
| } |
| catch (ClusterTopologyException ignored) { |
| failed.set(true); |
| } |
| finally { |
| assertTrue(failed.get()); |
| assertFalse(routed.get()); |
| |
| stopGrid(NODE1); |
| stopGrid(NODE2); |
| } |
| } |
| |
| /** |
| * Tests that failover happens on three-node grid when the Task is applicable for the first node |
| * and fails on it, but is also applicable on another node. |
| * |
| * @throws Exception If failed. |
| */ |
| @Test |
| public void testJobFailedOver() throws Exception { |
| failed.set(false); |
| routed.set(false); |
| |
| try { |
| Ignite ignite1 = startGrid(NODE1); |
| Ignite ignite2 = startGrid(NODE2); |
| Ignite ignite3 = startGrid(NODE3); |
| |
| assert ignite1 != null; |
| assert ignite2 != null; |
| assert ignite3 != null; |
| |
| Integer res = (Integer)compute(ignite1.cluster().forPredicate(p)).withTimeout(10000). |
| execute(JobFailTask.class.getName(), "1"); |
| |
| assert res == 1; |
| } |
| catch (ClusterTopologyException ignored) { |
| failed.set(true); |
| } |
| finally { |
| assertFalse(failed.get()); |
| assertTrue(routed.get()); |
| |
| stopGrid(NODE1); |
| stopGrid(NODE2); |
| stopGrid(NODE3); |
| } |
| } |
| |
| /** |
| * Tests that in case of failover our predicate is intersected with projection |
| * (logical AND is performed). |
| * |
| * @throws Exception If error happens. |
| */ |
| @Test |
| public void testJobNotFailedOverWithStaticProjection() throws Exception { |
| failed.set(false); |
| routed.set(false); |
| |
| try { |
| Ignite ignite1 = startGrid(NODE1); |
| Ignite ignite2 = startGrid(NODE2); |
| Ignite ignite3 = startGrid(NODE3); |
| |
| assert ignite1 != null; |
| assert ignite2 != null; |
| assert ignite3 != null; |
| |
| // Get projection only for first 2 nodes. |
| ClusterGroup nodes = ignite1.cluster().forNodeIds(Arrays.asList( |
| ignite1.cluster().localNode().id(), |
| ignite2.cluster().localNode().id())); |
| |
| // On failover NODE3 shouldn't be taken into account. |
| Integer res = (Integer)compute(nodes.forPredicate(p)).withTimeout(10000). |
| execute(JobFailTask.class.getName(), "1"); |
| |
| assert res == 1; |
| } |
| catch (ClusterTopologyException ignored) { |
| failed.set(true); |
| } |
| finally { |
| assertTrue(failed.get()); |
| assertFalse(routed.get()); |
| |
| stopGrid(NODE1); |
| stopGrid(NODE2); |
| stopGrid(NODE3); |
| } |
| } |
| |
| /** */ |
| @ComputeTaskSessionFullSupport |
| private static class JobFailTask implements ComputeTask<String, Object> { |
| /** */ |
| @TaskSessionResource |
| private ComputeTaskSession ses; |
| |
| /** {@inheritDoc} */ |
| @NotNull @Override public Map<? extends ComputeJob, ClusterNode> map(List<ClusterNode> subgrid, String arg) { |
| ses.setAttribute("fail", true); |
| |
| return Collections.singletonMap(new ComputeJobAdapter(arg) { |
| /** {@inheritDoc} */ |
| @SuppressWarnings({"RedundantTypeArguments"}) |
| @Override public Serializable execute() { |
| boolean fail; |
| |
| try { |
| fail = ses.<String, Boolean>waitForAttribute("fail", 0); |
| } |
| catch (InterruptedException e) { |
| throw new IgniteException("Got interrupted while waiting for attribute to be set.", e); |
| } |
| |
| if (fail) { |
| ses.setAttribute("fail", false); |
| |
| throw new IgniteException("Job exception."); |
| } |
| |
| // This job does not return any result. |
| return Integer.parseInt(this.<String>argument(0)); |
| } |
| }, subgrid.get(0)); |
| } |
| |
| /** {@inheritDoc} */ |
| @Override public ComputeJobResultPolicy result(ComputeJobResult res, List<ComputeJobResult> received) { |
| if (res.getException() != null && !(res.getException() instanceof ComputeUserUndeclaredException)) |
| return ComputeJobResultPolicy.FAILOVER; |
| |
| return ComputeJobResultPolicy.REDUCE; |
| } |
| |
| /** {@inheritDoc} */ |
| @Override public Object reduce(List<ComputeJobResult> results) { |
| assert results.size() == 1; |
| |
| return results.get(0).getData(); |
| } |
| } |
| |
| } |