Commit 9dadc8c4 authored by chiaming2000's avatar chiaming2000
Browse files

Admin cli: provide a way to override (admin cli request timeout) at

runtime the same way as a regular Kinetic client.
parent 440246c5
Loading
Loading
Loading
Loading
+8 −1
Original line number Diff line number Diff line
@@ -44,6 +44,7 @@ import kinetic.admin.Limits;
import kinetic.admin.Statistics;
import kinetic.admin.Temperature;
import kinetic.admin.Utilization;
import kinetic.client.ClientConfiguration;
import kinetic.client.KineticException;

import com.google.protobuf.ByteString;
@@ -75,6 +76,12 @@ public class KineticAdminCLI {
    private static KineticAdminClient kineticAdminClient = null;
    private final Map<String, List<String>> legalArguments = new HashMap<String, List<String>>();

    /**
     * provide a way to override default request timeout at runtime.
     */
    private static long DEFAULT_REQUEST_TIMEOUT = Integer.getInteger(
            ClientConfiguration.DEFAULT_TIMEOUT_PROP_NAME, 180000).longValue();

    public KineticAdminCLI() throws KineticException {
        String rootArg = "-help";
        List<String> subArgs = new ArrayList<String>();
@@ -153,7 +160,7 @@ public class KineticAdminCLI {
    public void init(String host, String tlsPort, String clusterVersion)
            throws KineticException {
        AdminClientConfiguration adminClientConfig = new AdminClientConfiguration();
        adminClientConfig.setRequestTimeoutMillis(180000);
        adminClientConfig.setRequestTimeoutMillis(DEFAULT_REQUEST_TIMEOUT);
        if (host != null && !host.isEmpty()) {
            validateHost(host);
            adminClientConfig.setHost(host);