jmx.org.apache.cassandra.JMXUtils Maven / Gradle / Ivy

package jmx.org.apache.cassandra;

import javax.management.remote.JMXConnector;
import javax.management.remote.JMXConnectorFactory;
import javax.rmi.ssl.SslRMIClientSocketFactory;
import java.nio.file.Paths;
import java.util.HashMap;
import java.util.Map;

public class JMXUtils {

    public static JMXConnector getJmxConnector(final CassandraJMXConnectionInfo jmxConnectionInfo) throws Exception {
        if (jmxConnectionInfo == null || jmxConnectionInfo.jmxServiceURL == null) {
            throw new IllegalArgumentException("passed JMXConnectionInfo is either null or its jmxServiceURL is null.");
        }

        final Map envMap = new HashMap<>();

        if (jmxConnectionInfo.jmxPassword != null && jmxConnectionInfo.jmxUser != null) {
            envMap.put(JMXConnector.CREDENTIALS, new String[]{
                jmxConnectionInfo.jmxUser,
                jmxConnectionInfo.jmxPassword});
        }

        if (jmxConnectionInfo.trustStore != null && jmxConnectionInfo.trustStorePassword != null) {

            if (!Paths.get(jmxConnectionInfo.trustStore).toFile().exists()) {
                throw new IllegalStateException(String.format("Specified truststore file for Cassandra %s does not exist!", jmxConnectionInfo.trustStore));
            }

            System.setProperty("javax.net.ssl.trustStore", jmxConnectionInfo.trustStore);
            System.setProperty("javax.net.ssl.trustStorePassword", jmxConnectionInfo.trustStorePassword);
            System.setProperty("javax.net.ssl.keyStore", jmxConnectionInfo.keyStore);
            System.setProperty("javax.net.ssl.keyStorePassword", jmxConnectionInfo.keyStorePassword);
            System.setProperty("ssl.enable", "true");
            System.setProperty("com.sun.management.jmxremote.ssl.need.client.auth", Boolean.toString(jmxConnectionInfo.clientAuth));
            System.setProperty("com.sun.management.jmxremote.registry.ssl", "true");

            envMap.put("com.sun.jndi.rmi.factory.socket", new SslRMIClientSocketFactory());
        }

        return envMap.isEmpty() ?
            JMXConnectorFactory.newJMXConnector(jmxConnectionInfo.jmxServiceURL, null)
            : JMXConnectorFactory.newJMXConnector(jmxConnectionInfo.jmxServiceURL, envMap);
    }
}