| /* |
| * Licensed 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.karaf.cellar.shell; |
| |
| import org.apache.karaf.cellar.core.Node; |
| import org.apache.karaf.cellar.core.shell.completer.AllNodeCompleter; |
| import org.apache.karaf.cellar.utils.ping.Ping; |
| import org.apache.karaf.shell.api.action.Argument; |
| import org.apache.karaf.shell.api.action.Command; |
| import org.apache.karaf.shell.api.action.Completion; |
| import org.apache.karaf.shell.api.action.lifecycle.Service; |
| |
| import java.util.Arrays; |
| import java.util.HashSet; |
| |
| @Command(scope = "cluster", name = "node-ping", description = "Ping a cluster node") |
| @Service |
| public class NodePingCommand extends ClusterCommandSupport { |
| |
| private static Long TIMEOUT = 10000L; |
| |
| @Argument(index = 0, name = "node", description = "The ID or alias of the node to ping", required = true, multiValued = false) |
| @Completion(AllNodeCompleter.class) |
| String nodeIdOrAlias; |
| |
| @Argument(index = 1, name = "iterations", description = "The number of iterations to perform", required = false, multiValued = false) |
| Integer iterations = 10; |
| |
| @Argument(index = 2, name = "interval", description = "The time in millis to wait between iteration", required = false, multiValued = false) |
| Long interval = 1000L; |
| |
| @Override |
| protected Object doExecute() throws Exception { |
| Node node = clusterManager.findNodeByIdOrAlias(nodeIdOrAlias); |
| if (node == null) { |
| System.out.println("Cluster node " + nodeIdOrAlias + " doesn't exist"); |
| return null; |
| } |
| |
| System.out.println("PING " + node.getId()); |
| try { |
| for (int i = 1; i <= iterations; i++) { |
| Long start = System.currentTimeMillis(); |
| Ping ping = new Ping(clusterManager.generateId()); |
| ping.setTimeout(timeout * 1000); |
| ping.setDestination(new HashSet(Arrays.asList(node))); |
| executionContext.execute(ping); |
| Long stop = System.currentTimeMillis(); |
| Long delay = stop - start; |
| if (delay >= TIMEOUT) { |
| System.err.println(String.format("TIMEOUT %s %s", i, node.getId())); |
| } else { |
| System.out.println(String.format("from %s: req=%s time=%s ms", i, node.getId(), delay)); |
| } |
| Thread.sleep(interval); |
| } |
| } catch (InterruptedException e) { |
| // nothing to do |
| } catch (Exception e) { |
| throw e; |
| } |
| return null; |
| } |
| |
| } |