This commit is contained in:
Dejan Bosanac 2015-09-04 15:50:52 +02:00
parent c7b93d1232
commit 73f9131a62
10 changed files with 138 additions and 181 deletions

View File

@ -20,6 +20,7 @@ import java.io.BufferedReader;
import java.io.IOException; import java.io.IOException;
import java.io.InputStream; import java.io.InputStream;
import java.io.InputStreamReader; import java.io.InputStreamReader;
import java.net.ConnectException;
import java.util.List; import java.util.List;
import org.apache.activemq.ActiveMQConnectionMetaData; import org.apache.activemq.ActiveMQConnectionMetaData;
@ -159,4 +160,20 @@ public abstract class AbstractCommand implements Command {
} }
} }
} }
protected void handleException(Throwable exception, String serviceUrl) {
Throwable cause = exception.getCause();
while (true) {
Throwable next = cause.getCause();
if (next == null) {
break;
}
cause = next;
}
if (cause instanceof ConnectException) {
context.printInfo("Broker not available at: " + serviceUrl);
} else {
context.printException(new RuntimeException("Failed to execute " + getName() + " task. Reason: " + exception));
}
}
} }

View File

@ -20,6 +20,7 @@ import java.io.File;
import java.io.IOException; import java.io.IOException;
import java.lang.management.ManagementFactory; import java.lang.management.ManagementFactory;
import java.lang.reflect.Method; import java.lang.reflect.Method;
import java.net.ConnectException;
import java.net.MalformedURLException; import java.net.MalformedURLException;
import java.net.URL; import java.net.URL;
import java.net.URLClassLoader; import java.net.URLClassLoader;
@ -385,6 +386,9 @@ public abstract class AbstractJmxCommand extends AbstractCommand {
public void execute(List<String> tokens) throws Exception { public void execute(List<String> tokens) throws Exception {
try { try {
super.execute(tokens); super.execute(tokens);
} catch (Throwable exception) {
handleException(exception, jmxServiceUrl.toString());
return;
}finally { }finally {
closeJmxConnection(); closeJmxConnection();
} }

View File

@ -25,6 +25,7 @@ import java.util.StringTokenizer;
import javax.jms.Destination; import javax.jms.Destination;
import org.apache.activemq.ActiveMQConnectionFactory;
import org.apache.activemq.command.ActiveMQQueue; import org.apache.activemq.command.ActiveMQQueue;
import org.apache.activemq.command.ActiveMQTopic; import org.apache.activemq.command.ActiveMQTopic;
import org.apache.activemq.console.util.AmqMessagesUtil; import org.apache.activemq.console.util.AmqMessagesUtil;
@ -39,11 +40,11 @@ public class AmqBrowseCommand extends AbstractAmqCommand {
public static final String VIEW_GROUP_BODY = "body:"; public static final String VIEW_GROUP_BODY = "body:";
protected String[] helpFile = new String[] { protected String[] helpFile = new String[] {
"Task Usage: Main browse --amqurl <broker url> [browse-options] <destinations>", "Task Usage: Main browse [browse-options] <destinations>",
"Description: Display selected destination's messages.", "Description: Display selected destination's messages.",
"", "",
"Browse Options:", "Browse Options:",
" --amqurl <url> Set the broker URL to connect to.", " --amqurl <url> Set the broker URL to connect to. Default tcp://localhost:61616",
" --msgsel <msgsel1,msglsel2> Add to the search list messages matched by the query similar to", " --msgsel <msgsel1,msglsel2> Add to the search list messages matched by the query similar to",
" the messages selector format.", " the messages selector format.",
" --factory <className> Load className as the javax.jms.ConnectionFactory to use for creating connections.", " --factory <className> Load className as the javax.jms.ConnectionFactory to use for creating connections.",
@ -116,8 +117,7 @@ public class AmqBrowseCommand extends AbstractAmqCommand {
// If no broker url specified // If no broker url specified
if (getBrokerUrl() == null) { if (getBrokerUrl() == null) {
context.printException(new IllegalStateException("No broker url specified. Use the --amqurl option to specify a broker url.")); setBrokerUrl(ActiveMQConnectionFactory.DEFAULT_BROKER_BIND_URL);
return;
} }
// Display the messages for each destination // Display the messages for each destination
@ -151,9 +151,8 @@ public class AmqBrowseCommand extends AbstractAmqCommand {
context.printMessage(AmqMessagesUtil.filterMessagesView(addMsgs, groupViews, queryViews)); context.printMessage(AmqMessagesUtil.filterMessagesView(addMsgs, groupViews, queryViews));
} }
} catch (Exception e) { } catch (Exception exception) {
context.printException(new RuntimeException("Failed to execute browse task. Reason: " + e)); handleException(exception, getBrokerUrl().toString());
throw new Exception(e);
} }
} }

View File

@ -97,7 +97,6 @@ public class BrowseCommand extends AbstractJmxCommand {
* @throws Exception * @throws Exception
*/ */
protected void runTask(List<String> tokens) throws Exception { protected void runTask(List<String> tokens) throws Exception {
try {
// If there is no queue name specified, let's select all // If there is no queue name specified, let's select all
if (tokens.isEmpty()) { if (tokens.isEmpty()) {
tokens.add("*"); tokens.add("*");
@ -113,10 +112,6 @@ public class BrowseCommand extends AbstractJmxCommand {
context.printMessage(JmxMBeansUtil.filterMessagesView(messages, groupViews, queryViews)); context.printMessage(JmxMBeansUtil.filterMessagesView(messages, groupViews, queryViews));
} }
} }
} catch (Exception e) {
context.printException(new RuntimeException("Failed to execute browse task. Reason: " + e));
throw new Exception(e);
}
} }
/** /**

View File

@ -66,8 +66,6 @@ public class DstatCommand extends AbstractJmxCommand {
*/ */
@Override @Override
protected void runTask(List<String> tokens) throws Exception { protected void runTask(List<String> tokens) throws Exception {
try {
if (tokens.contains("topics")) { if (tokens.contains("topics")) {
displayTopicStats(); displayTopicStats();
} else if (tokens.contains("queues")) { } else if (tokens.contains("queues")) {
@ -75,12 +73,6 @@ public class DstatCommand extends AbstractJmxCommand {
} else { } else {
displayAllDestinations(); displayAllDestinations();
} }
// Iterate through the queue names
} catch (Exception e) {
context.printException(new RuntimeException("Failed to execute dstat task. Reason: " + e.getMessage(), e));
throw new Exception(e);
}
} }
@SuppressWarnings("unchecked") @SuppressWarnings("unchecked")

View File

@ -55,14 +55,9 @@ public class ListCommand extends AbstractJmxCommand {
* @throws Exception * @throws Exception
*/ */
protected void runTask(List tokens) throws Exception { protected void runTask(List tokens) throws Exception {
try {
Set<String> propsView = new HashSet<String>(); Set<String> propsView = new HashSet<String>();
propsView.add("brokerName"); propsView.add("brokerName");
context.printMBean(JmxMBeansUtil.filterMBeansView(JmxMBeansUtil.getAllBrokers(createJmxConnection()), propsView)); context.printMBean(JmxMBeansUtil.filterMBeansView(JmxMBeansUtil.getAllBrokers(createJmxConnection()), propsView));
} catch (Exception e) {
context.printException(new RuntimeException("Failed to execute list task. Reason: " + e));
throw new Exception(e);
}
} }
/** /**

View File

@ -83,7 +83,6 @@ public class PurgeCommand extends AbstractJmxCommand {
*/ */
@Override @Override
protected void runTask(List<String> tokens) throws Exception { protected void runTask(List<String> tokens) throws Exception {
try {
// If there is no queue name specified, let's select all // If there is no queue name specified, let's select all
if (tokens.isEmpty()) { if (tokens.isEmpty()) {
tokens.add("*"); tokens.add("*");
@ -135,10 +134,6 @@ public class PurgeCommand extends AbstractJmxCommand {
} }
} }
} }
} catch (Exception e) {
context.printException(new RuntimeException("Failed to execute purge task. Reason: " + e));
throw new Exception(e);
}
} }

View File

@ -116,7 +116,6 @@ public class QueryCommand extends AbstractJmxCommand {
* @throws Exception * @throws Exception
*/ */
protected void runTask(List<String> tokens) throws Exception { protected void runTask(List<String> tokens) throws Exception {
try {
// Query for the mbeans to add // Query for the mbeans to add
Map<Object, List> addMBeans = JmxMBeansUtil.queryMBeansAsMap(createJmxConnection(), queryAddObjects, queryViews); Map<Object, List> addMBeans = JmxMBeansUtil.queryMBeansAsMap(createJmxConnection(), queryAddObjects, queryViews);
// Query for the mbeans to sub // Query for the mbeans to sub
@ -130,10 +129,6 @@ public class QueryCommand extends AbstractJmxCommand {
} else { } else {
context.print(doInvoke(addMBeans.keySet(), opAndParams)); context.print(doInvoke(addMBeans.keySet(), opAndParams));
} }
} catch (Exception e) {
context.printException(new RuntimeException("Failed to execute query task. Reason: " + e));
throw new Exception(e);
}
} }
private Collection doInvoke(Set<Object> mBeans, List<String> opAndParams) throws Exception { private Collection doInvoke(Set<Object> mBeans, List<String> opAndParams) throws Exception {

View File

@ -69,7 +69,6 @@ public class ShutdownCommand extends AbstractJmxCommand {
* @throws Exception * @throws Exception
*/ */
protected void runTask(List brokerNames) throws Exception { protected void runTask(List brokerNames) throws Exception {
try {
Collection mbeans; Collection mbeans;
// Stop all brokers // Stop all brokers
@ -112,10 +111,6 @@ public class ShutdownCommand extends AbstractJmxCommand {
// Stop all brokers in set // Stop all brokers in set
stopBrokers(createJmxConnection(), mbeans); stopBrokers(createJmxConnection(), mbeans);
} catch (Exception e) {
context.printException(new RuntimeException("Failed to execute stop task. Reason: " + e));
throw new Exception(e);
}
} }
/** /**

View File

@ -657,37 +657,7 @@ case "$1" in
invoke_kill invoke_kill
exit $? exit $?
;; ;;
decrypt|encrypt|create)
invoke_task checknotforrunning
exit $?
;;
export)
invoke_task checkfornotrunning
exit $?
;;
query|bstat|dstat|purge)
# Only check for a running broker if "--jmxurl" is part of the COMMANDLINE_ARGS
if (echo "$COMMANDLINE_ARGS"|grep -q -- "--jmxurl");then
invoke_task checknotforrunning
RET="$?"
else
invoke_task checkforrunning
RET="$?"
fi
exit $RET
;;
browse)
# Only check for a running broker if "--amqurl" is part of the COMMANDLINE_ARGS
if (echo "$COMMANDLINE_ARGS"|grep -q -- "--amqurl");then
invoke_task checknotforrunning
RET="$?"
else
invoke_task checkforrunning
RET="$?"
fi
exit $RET
;;
*) *)
invoke_task checkforrunning invoke_task checknotforrunning
exit $? exit $?
esac esac