| /* |
| * 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.p2p; |
| |
| import java.util.HashMap; |
| import java.util.List; |
| import java.util.Map; |
| import java.util.UUID; |
| import org.apache.ignite.Ignite; |
| import org.apache.ignite.IgniteLogger; |
| import org.apache.ignite.cluster.ClusterNode; |
| import org.apache.ignite.compute.ComputeJob; |
| import org.apache.ignite.compute.ComputeJobResult; |
| import org.apache.ignite.compute.ComputeTaskAdapter; |
| import org.apache.ignite.compute.gridify.GridifyArgument; |
| import org.apache.ignite.resources.IgniteInstanceResource; |
| import org.apache.ignite.resources.LoggerResource; |
| import org.jetbrains.annotations.NotNull; |
| |
| /** |
| * P2P test task. |
| */ |
| public class GridP2PTestTask extends ComputeTaskAdapter<Object, Integer> { |
| /** */ |
| public static final String TASK_NAME = GridP2PTestTask.class.getName(); |
| |
| /** */ |
| @LoggerResource |
| private IgniteLogger log; |
| |
| /** Ignite instance. */ |
| @IgniteInstanceResource |
| private Ignite ignite; |
| |
| /** {@inheritDoc} */ |
| @NotNull @Override public Map<? extends ComputeJob, ClusterNode> map(List<ClusterNode> subgrid, Object arg) { |
| assert subgrid != null; |
| assert !subgrid.isEmpty(); |
| |
| Integer arg1 = null; |
| |
| if (arg instanceof GridifyArgument) |
| arg1 = (Integer)((GridifyArgument)arg).getMethodParameters()[0]; |
| else if (arg instanceof Integer) |
| arg1 = (Integer)arg; |
| else |
| assert false : "Failed to map task (unknown argument type) [type=" + arg.getClass() + ", val=" + arg + ']'; |
| |
| Map<ComputeJob, ClusterNode> map = new HashMap<>(subgrid.size()); |
| |
| UUID nodeId = ignite != null ? ignite.configuration().getNodeId() : null; |
| |
| for (ClusterNode node : subgrid) |
| if (!node.id().equals(nodeId)) |
| map.put(new GridP2PTestJob(arg1), node); |
| |
| return map; |
| } |
| |
| /** {@inheritDoc} */ |
| @Override public Integer reduce(List<ComputeJobResult> results) { |
| assert results.size() == 1 : "Results [received=" + results.size() + ", expected=" + 1 + ']'; |
| |
| ComputeJobResult res = results.get(0); |
| |
| if (log.isInfoEnabled()) |
| log.info("Got job result for aggregation: " + res); |
| |
| if (res.getException() != null) |
| throw res.getException(); |
| |
| return res.getData(); |
| } |
| } |