java/client/src/main/java/org/apache/qpid/client/AMQBrokerDetails.java - qpid - Git at Google

 /*
  *
  * 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.qpid.client;

 import java.net.URI;
 import java.net.URISyntaxException;
 import java.util.HashMap;
 import java.util.Map;

 import org.apache.qpid.jms.BrokerDetails;
 import org.apache.qpid.transport.ConnectionSettings;
 import org.apache.qpid.url.URLHelper;
 import org.apache.qpid.url.URLSyntaxException;

 public class AMQBrokerDetails implements BrokerDetails
 {
     private String _host;
     private int _port;
     private String _transport;

     private Map<String, String> _options = new HashMap<String, String>();

     public AMQBrokerDetails(){}

     public AMQBrokerDetails(String url) throws URLSyntaxException
     {

         // URL should be of format tcp://host:port?option='value',option='value'
         try
         {
             URI connection = new URI(url);

             String transport = connection.getScheme();

             // Handles some defaults to minimise changes to existing broker URLS e.g. localhost
             if (transport != null)
             {
                 //todo this list of valid transports should be enumerated somewhere
                 if (!(transport.equalsIgnoreCase(BrokerDetails.TCP)))
                 {
                     if (transport.equalsIgnoreCase("localhost"))
                     {
                         connection = new URI(DEFAULT_TRANSPORT + "://" + url);
                         transport = connection.getScheme();
                     }
                     else
                     {
                         if (url.charAt(transport.length()) == ':' && url.charAt(transport.length() + 1) != '/')
                         {
                             //Then most likely we have a host:port value
                             connection = new URI(DEFAULT_TRANSPORT + "://" + url);
                             transport = connection.getScheme();
                         }
                         else
                         {
                             throw URLHelper.parseError(0, transport.length(), "Unknown transport", url);
                         }
                     }
                 }
                 else if (url.indexOf("//") == -1)
                 {
                     throw new URLSyntaxException(url, "Missing '//' after the transport In broker URL",transport.length()+1,1);
                 }
             }
             else
             {
                 //Default the transport
                 connection = new URI(DEFAULT_TRANSPORT + "://" + url);
                 transport = connection.getScheme();
             }

             if (transport == null)
             {
                 throw URLHelper.parseError(-1, "Unknown transport in broker URL:'"
                         + url + "' Format: " + URL_FORMAT_EXAMPLE, "");
             }

             setTransport(transport);

             String host = connection.getHost();

             // Fix for Java 1.5
             if (host == null)
             {
                 host = "";

                 String auth = connection.getAuthority();
                 if (auth != null)
                 {
                     // contains both host & port myhost:5672
                     if (auth.contains(":"))
                     {
                         host = auth.substring(0,auth.indexOf(":"));
                     }
                     else
                     {
                         host = auth;
                     }
                 }

             }

             setHost(host);

             int port = connection.getPort();

             if (port == -1)
             {
                 // Fix for when there is port data but it is not automatically parseable by getPort().
                 String auth = connection.getAuthority();

                 if (auth != null && auth.contains(":"))
                 {
                     int start = auth.indexOf(":") + 1;
                     int end = start;
                     boolean looking = true;
                     boolean found = false;
                     // Throw an URL exception if the port number is not specified
                     if (start == auth.length())
                     {
                         throw URLHelper.parseError(connection.toString().indexOf(auth) + end - 1,
                                 connection.toString().indexOf(auth) + end, "Port number must be specified",
                                 connection.toString());
                     }
                     //Walk the authority looking for a port value.
                     while (looking)
                     {
                         try
                         {
                             end++;
                             Integer.parseInt(auth.substring(start, end));

                             if (end >= auth.length())
                             {
                                 looking = false;
                                 found = true;
                             }
                         }
                         catch (NumberFormatException nfe)
                         {
                             looking = false;
                         }

                     }
                     if (found)
                     {
                         setPort(Integer.parseInt(auth.substring(start, end)));
                     }
                     else
                     {
                         throw URLHelper.parseError(connection.toString().indexOf(connection.getAuthority()) + end - 1,
                                              "Illegal character in port number", connection.toString());
                     }

                 }
                 else
                 {
                     setPort(DEFAULT_PORT);
                 }
             }
             else
             {
                 setPort(port);
             }

             String queryString = connection.getQuery();

             URLHelper.parseOptions(_options, queryString);

             //Fragment is #string (not used)
         }
         catch (URISyntaxException uris)
         {
             if (uris instanceof URLSyntaxException)
             {
                 throw(URLSyntaxException) uris;
             }

             throw URLHelper.parseError(uris.getIndex(), uris.getReason(), uris.getInput());
         }
     }

     public AMQBrokerDetails(String host, int port)
     {
         _host = host;
         _port = port;
     }

     public String getHost()
     {
         return _host;
     }

     public void setHost(String _host)
     {
         this._host = _host;
     }

     public int getPort()
     {
         return _port;
     }

     public void setPort(int _port)
     {
         this._port = _port;
     }

     public String getTransport()
     {
         return _transport;
     }

     public void setTransport(String _transport)
     {
         this._transport = _transport;
     }


     public String getProperty(String key)
     {
         return _options.get(key);
     }

     public void setProperty(String key, String value)
     {
         _options.put(key, value);
     }

     public long getTimeout()
     {
         if (_options.containsKey(OPTIONS_CONNECT_TIMEOUT))
         {
             try
             {
                 return Long.parseLong(_options.get(OPTIONS_CONNECT_TIMEOUT));
             }
             catch (NumberFormatException nfe)
             {
                 //Do nothing as we will use the default below.
             }
         }

         return BrokerDetails.DEFAULT_CONNECT_TIMEOUT;
     }

     public boolean getBooleanProperty(String propName)
     {
         if (_options.containsKey(propName))
         {
             return Boolean.parseBoolean(_options.get(propName));
         }

         return false;
     }

     public void setTimeout(long timeout)
     {
         setProperty(OPTIONS_CONNECT_TIMEOUT, Long.toString(timeout));
     }

     public String toString()
     {
         StringBuffer sb = new StringBuffer();

         sb.append(_transport);
         sb.append("://");
         sb.append(_host);
         sb.append(':');
         sb.append(_port);

         sb.append(printOptionsURL());

         return sb.toString();
     }

     public boolean equals(Object o)
     {
         if (!(o instanceof BrokerDetails))
         {
             return false;
         }

         BrokerDetails bd = (BrokerDetails) o;

         return _host.equalsIgnoreCase(bd.getHost()) &&
                (_port == bd.getPort()) &&
                _transport.equalsIgnoreCase(bd.getTransport());
         //TODO do we need to compare all the options as well?
     }

     @Override
     public int hashCode()
     {
         int result = _host != null ? _host.hashCode() : 0;
         result = 31 * result + _port;
         result = 31 * result + (_transport != null ? _transport.hashCode() : 0);
         return result;
     }

     private String printOptionsURL()
     {
         StringBuffer optionsURL = new StringBuffer();

         optionsURL.append('?');

         if (!(_options.isEmpty()))
         {

             for (String key : _options.keySet())
             {
                 optionsURL.append(key);

                 optionsURL.append("='");

                 optionsURL.append(_options.get(key));

                 optionsURL.append("'");

                 optionsURL.append(URLHelper.DEFAULT_OPTION_SEPERATOR);
             }
         }

         //removeKey the extra DEFAULT_OPTION_SEPERATOR or the '?' if there are no options
         optionsURL.deleteCharAt(optionsURL.length() - 1);

         return optionsURL.toString();
     }

     public static String checkTransport(String broker)
     {
         if ((!broker.contains("://")))
         {
             return "tcp://" + broker;
         }
         else
         {
             return broker;
         }
     }

     public Map<String, String> getProperties()
     {
         return _options;
     }

     public void setProperties(Map<String, String> props)
     {
         _options = props;
     }

     public ConnectionSettings buildConnectionSettings()
     {
         ConnectionSettings conSettings = new ConnectionSettings();

         conSettings.setHost(getHost());
         conSettings.setPort(getPort());

         // ------------ sasl options ---------------
         if (getProperty(BrokerDetails.OPTIONS_SASL_MECHS) != null)
         {
             conSettings.setSaslMechs(
                     getProperty(BrokerDetails.OPTIONS_SASL_MECHS));
         }

         // Sun SASL Kerberos client uses the
         // protocol + servername as the service key.

         if (getProperty(BrokerDetails.OPTIONS_SASL_PROTOCOL_NAME) != null)
         {
             conSettings.setSaslProtocol(
                     getProperty(BrokerDetails.OPTIONS_SASL_PROTOCOL_NAME));
         }


         if (getProperty(BrokerDetails.OPTIONS_SASL_SERVER_NAME) != null)
         {
             conSettings.setSaslServerName(
                     getProperty(BrokerDetails.OPTIONS_SASL_SERVER_NAME));
         }

         conSettings.setUseSASLEncryption(
                 getBooleanProperty(BrokerDetails.OPTIONS_SASL_ENCRYPTION));

         // ------------- ssl options ---------------------
         conSettings.setUseSSL(getBooleanProperty(BrokerDetails.OPTIONS_SSL));

         if (getProperty(BrokerDetails.OPTIONS_TRUST_STORE) != null)
         {
             conSettings.setTrustStorePath(
                     getProperty(BrokerDetails.OPTIONS_TRUST_STORE));
         }

         if (getProperty(BrokerDetails.OPTIONS_TRUST_STORE_PASSWORD) != null)
         {
             conSettings.setTrustStorePassword(
                     getProperty(BrokerDetails.OPTIONS_TRUST_STORE_PASSWORD));
         }

         if (getProperty(BrokerDetails.OPTIONS_KEY_STORE) != null)
         {
             conSettings.setKeyStorePath(
                     getProperty(BrokerDetails.OPTIONS_KEY_STORE));
         }

         if (getProperty(BrokerDetails.OPTIONS_KEY_STORE_PASSWORD) != null)
         {
             conSettings.setKeyStorePassword(
                     getProperty(BrokerDetails.OPTIONS_KEY_STORE_PASSWORD));
         }

         if (getProperty(BrokerDetails.OPTIONS_SSL_CERT_ALIAS) != null)
         {
             conSettings.setCertAlias(
                     getProperty(BrokerDetails.OPTIONS_SSL_CERT_ALIAS));
         }
         // ----------------------------

         conSettings.setVerifyHostname(getBooleanProperty(BrokerDetails.OPTIONS_SSL_VERIFY_HOSTNAME));

         if (getProperty(BrokerDetails.OPTIONS_TCP_NO_DELAY) != null)
         {
             conSettings.setTcpNodelay(
                     getBooleanProperty(BrokerDetails.OPTIONS_TCP_NO_DELAY));
         }

         return conSettings;
     }
 }
	/*
	*
	* 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.qpid.client;

	import java.net.URI;
	import java.net.URISyntaxException;
	import java.util.HashMap;
	import java.util.Map;

	import org.apache.qpid.jms.BrokerDetails;
	import org.apache.qpid.transport.ConnectionSettings;
	import org.apache.qpid.url.URLHelper;
	import org.apache.qpid.url.URLSyntaxException;

	public class AMQBrokerDetails implements BrokerDetails
	{
	private String _host;
	private int _port;
	private String _transport;

	private Map<String, String> _options = new HashMap<String, String>();

	public AMQBrokerDetails(){}

	public AMQBrokerDetails(String url) throws URLSyntaxException
	{

	// URL should be of format tcp://host:port?option='value',option='value'
	try
	{
	URI connection = new URI(url);

	String transport = connection.getScheme();

	// Handles some defaults to minimise changes to existing broker URLS e.g. localhost
	if (transport != null)
	{
	//todo this list of valid transports should be enumerated somewhere
	if (!(transport.equalsIgnoreCase(BrokerDetails.TCP)))
	{
	if (transport.equalsIgnoreCase("localhost"))
	{
	connection = new URI(DEFAULT_TRANSPORT + "://" + url);
	transport = connection.getScheme();
	}
	else
	{
	if (url.charAt(transport.length()) == ':' && url.charAt(transport.length() + 1) != '/')
	{
	//Then most likely we have a host:port value
	connection = new URI(DEFAULT_TRANSPORT + "://" + url);
	transport = connection.getScheme();
	}
	else
	{
	throw URLHelper.parseError(0, transport.length(), "Unknown transport", url);
	}
	}
	}
	else if (url.indexOf("//") == -1)
	{
	throw new URLSyntaxException(url, "Missing '//' after the transport In broker URL",transport.length()+1,1);
	}
	}
	else
	{
	//Default the transport
	connection = new URI(DEFAULT_TRANSPORT + "://" + url);
	transport = connection.getScheme();
	}

	if (transport == null)
	{
	throw URLHelper.parseError(-1, "Unknown transport in broker URL:'"
	+ url + "' Format: " + URL_FORMAT_EXAMPLE, "");
	}

	setTransport(transport);

	String host = connection.getHost();

	// Fix for Java 1.5
	if (host == null)
	{
	host = "";

	String auth = connection.getAuthority();
	if (auth != null)
	{
	// contains both host & port myhost:5672
	if (auth.contains(":"))
	{
	host = auth.substring(0,auth.indexOf(":"));
	}
	else
	{
	host = auth;
	}
	}

	}

	setHost(host);

	int port = connection.getPort();

	if (port == -1)
	{
	// Fix for when there is port data but it is not automatically parseable by getPort().
	String auth = connection.getAuthority();

	if (auth != null && auth.contains(":"))
	{
	int start = auth.indexOf(":") + 1;
	int end = start;
	boolean looking = true;
	boolean found = false;
	// Throw an URL exception if the port number is not specified
	if (start == auth.length())
	{
	throw URLHelper.parseError(connection.toString().indexOf(auth) + end - 1,
	connection.toString().indexOf(auth) + end, "Port number must be specified",
	connection.toString());
	}
	//Walk the authority looking for a port value.
	while (looking)
	{
	try
	{
	end++;
	Integer.parseInt(auth.substring(start, end));

	if (end >= auth.length())
	{
	looking = false;
	found = true;
	}
	}
	catch (NumberFormatException nfe)
	{
	looking = false;
	}

	}
	if (found)
	{
	setPort(Integer.parseInt(auth.substring(start, end)));
	}
	else
	{
	throw URLHelper.parseError(connection.toString().indexOf(connection.getAuthority()) + end - 1,
	"Illegal character in port number", connection.toString());
	}

	}
	else
	{
	setPort(DEFAULT_PORT);
	}
	}
	else
	{
	setPort(port);
	}

	String queryString = connection.getQuery();

	URLHelper.parseOptions(_options, queryString);

	//Fragment is #string (not used)
	}
	catch (URISyntaxException uris)
	{
	if (uris instanceof URLSyntaxException)
	{
	throw(URLSyntaxException) uris;
	}

	throw URLHelper.parseError(uris.getIndex(), uris.getReason(), uris.getInput());
	}
	}

	public AMQBrokerDetails(String host, int port)
	{
	_host = host;
	_port = port;
	}

	public String getHost()
	{
	return _host;
	}

	public void setHost(String _host)
	{
	this._host = _host;
	}

	public int getPort()
	{
	return _port;
	}

	public void setPort(int _port)
	{
	this._port = _port;
	}

	public String getTransport()
	{
	return _transport;
	}

	public void setTransport(String _transport)
	{
	this._transport = _transport;
	}


	public String getProperty(String key)
	{
	return _options.get(key);
	}

	public void setProperty(String key, String value)
	{
	_options.put(key, value);
	}

	public long getTimeout()
	{
	if (_options.containsKey(OPTIONS_CONNECT_TIMEOUT))
	{
	try
	{
	return Long.parseLong(_options.get(OPTIONS_CONNECT_TIMEOUT));
	}
	catch (NumberFormatException nfe)
	{
	//Do nothing as we will use the default below.
	}
	}

	return BrokerDetails.DEFAULT_CONNECT_TIMEOUT;
	}

	public boolean getBooleanProperty(String propName)
	{
	if (_options.containsKey(propName))
	{
	return Boolean.parseBoolean(_options.get(propName));
	}

	return false;
	}

	public void setTimeout(long timeout)
	{
	setProperty(OPTIONS_CONNECT_TIMEOUT, Long.toString(timeout));
	}

	public String toString()
	{
	StringBuffer sb = new StringBuffer();

	sb.append(_transport);
	sb.append("://");
	sb.append(_host);
	sb.append(':');
	sb.append(_port);

	sb.append(printOptionsURL());

	return sb.toString();
	}

	public boolean equals(Object o)
	{
	if (!(o instanceof BrokerDetails))
	{
	return false;
	}

	BrokerDetails bd = (BrokerDetails) o;

	return _host.equalsIgnoreCase(bd.getHost()) &&
	(_port == bd.getPort()) &&
	_transport.equalsIgnoreCase(bd.getTransport());
	//TODO do we need to compare all the options as well?
	}

	@Override
	public int hashCode()
	{
	int result = _host != null ? _host.hashCode() : 0;
	result = 31 * result + _port;
	result = 31 * result + (_transport != null ? _transport.hashCode() : 0);
	return result;
	}

	private String printOptionsURL()
	{
	StringBuffer optionsURL = new StringBuffer();

	optionsURL.append('?');

	if (!(_options.isEmpty()))
	{

	for (String key : _options.keySet())
	{
	optionsURL.append(key);

	optionsURL.append("='");

	optionsURL.append(_options.get(key));

	optionsURL.append("'");

	optionsURL.append(URLHelper.DEFAULT_OPTION_SEPERATOR);
	}
	}

	//removeKey the extra DEFAULT_OPTION_SEPERATOR or the '?' if there are no options
	optionsURL.deleteCharAt(optionsURL.length() - 1);

	return optionsURL.toString();
	}

	public static String checkTransport(String broker)
	{
	if ((!broker.contains("://")))
	{
	return "tcp://" + broker;
	}
	else
	{
	return broker;
	}
	}

	public Map<String, String> getProperties()
	{
	return _options;
	}

	public void setProperties(Map<String, String> props)
	{
	_options = props;
	}

	public ConnectionSettings buildConnectionSettings()
	{
	ConnectionSettings conSettings = new ConnectionSettings();

	conSettings.setHost(getHost());
	conSettings.setPort(getPort());

	// ------------ sasl options ---------------
	if (getProperty(BrokerDetails.OPTIONS_SASL_MECHS) != null)
	{
	conSettings.setSaslMechs(
	getProperty(BrokerDetails.OPTIONS_SASL_MECHS));
	}

	// Sun SASL Kerberos client uses the
	// protocol + servername as the service key.

	if (getProperty(BrokerDetails.OPTIONS_SASL_PROTOCOL_NAME) != null)
	{
	conSettings.setSaslProtocol(
	getProperty(BrokerDetails.OPTIONS_SASL_PROTOCOL_NAME));
	}


	if (getProperty(BrokerDetails.OPTIONS_SASL_SERVER_NAME) != null)
	{
	conSettings.setSaslServerName(
	getProperty(BrokerDetails.OPTIONS_SASL_SERVER_NAME));
	}

	conSettings.setUseSASLEncryption(
	getBooleanProperty(BrokerDetails.OPTIONS_SASL_ENCRYPTION));

	// ------------- ssl options ---------------------
	conSettings.setUseSSL(getBooleanProperty(BrokerDetails.OPTIONS_SSL));

	if (getProperty(BrokerDetails.OPTIONS_TRUST_STORE) != null)
	{
	conSettings.setTrustStorePath(
	getProperty(BrokerDetails.OPTIONS_TRUST_STORE));
	}

	if (getProperty(BrokerDetails.OPTIONS_TRUST_STORE_PASSWORD) != null)
	{
	conSettings.setTrustStorePassword(
	getProperty(BrokerDetails.OPTIONS_TRUST_STORE_PASSWORD));
	}

	if (getProperty(BrokerDetails.OPTIONS_KEY_STORE) != null)
	{
	conSettings.setKeyStorePath(
	getProperty(BrokerDetails.OPTIONS_KEY_STORE));
	}

	if (getProperty(BrokerDetails.OPTIONS_KEY_STORE_PASSWORD) != null)
	{
	conSettings.setKeyStorePassword(
	getProperty(BrokerDetails.OPTIONS_KEY_STORE_PASSWORD));
	}

	if (getProperty(BrokerDetails.OPTIONS_SSL_CERT_ALIAS) != null)
	{
	conSettings.setCertAlias(
	getProperty(BrokerDetails.OPTIONS_SSL_CERT_ALIAS));
	}
	// ----------------------------

	conSettings.setVerifyHostname(getBooleanProperty(BrokerDetails.OPTIONS_SSL_VERIFY_HOSTNAME));

	if (getProperty(BrokerDetails.OPTIONS_TCP_NO_DELAY) != null)
	{
	conSettings.setTcpNodelay(
	getBooleanProperty(BrokerDetails.OPTIONS_TCP_NO_DELAY));
	}

	return conSettings;
	}
	}