Java tutorial
/* * 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 com.wenyu.clustertools; import com.wenyu.utils.AsyncTask; import com.wenyu.utils.ClusterToolNodeProbe; import com.wenyu.utils.Constants; import io.airlift.command.Arguments; import io.airlift.command.Command; import io.airlift.command.Option; import org.apache.cassandra.tools.NodeProbe; import org.apache.cassandra.tools.NodeTool.NodeToolCmd; import java.util.HashMap; import java.util.Map; import java.util.concurrent.ExecutorService; import java.util.concurrent.Executors; import java.util.concurrent.Future; import java.util.concurrent.TimeUnit; import static org.apache.commons.lang3.StringUtils.EMPTY; @Command(name = "setstreamthroughput", description = "Set the Mb/s throughput cap for streaming in the system, or 0 to disable throttling") public class SetStreamThroughput extends ClusterToolCmd { @Arguments(title = "stream_throughput", usage = "<value_in_mb>", description = "Value in Mb, 0 to disable throttling", required = true) private Integer streamThroughput = null; @Option(title = "parallel executor", name = { "-p", "--par-jobs" }, description = "Number of threads to get timeout of all nodes.") private int parallel = 1; @Override public void execute() { ExecutorService executor = Executors.newFixedThreadPool(parallel); Map<Node, Future<String>> futures = new HashMap<>(); for (ClusterToolCmd.Node node : nodes) { futures.put(node, executor.submit(new Executor(node))); } for (Map.Entry<ClusterToolCmd.Node, Future<String>> future : futures.entrySet()) { try { String result = future.getValue().get(Constants.MAX_PARALLEL_WAIT_IN_SEC, TimeUnit.SECONDS); System.out.println(result); } catch (Exception ex) { System.out .println(String.format("%s failed with error: %s", future.getKey().server, ex.toString())); ex.printStackTrace(); } } } private class Executor extends AsyncTask<String> { private ClusterToolCmd.Node node; public Executor(ClusterToolCmd.Node node) { this.node = node; } @Override public String execute() { ClusterToolNodeProbe probe = connect(node); probe.setStreamThroughput(streamThroughput); return "Finished set stream throughput on " + node.server + " to " + streamThroughput + "Mb/s"; } } }