/*
* JBoss, Home of Professional Open Source.
* Copyright 2011, Red Hat, Inc., and individual contributors
* as indicated by the @author tags. See the copyright.txt file in the
* distribution for a full listing of individual contributors.
*
* This is free software; you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License as
* published by the Free Software Foundation; either version 2.1 of
* the License, or (at your option) any later version.
*
* This software is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this software; if not, write to the Free
* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
*/
package org.jboss.as.messaging;
import static org.jboss.as.controller.SimpleAttributeDefinitionBuilder.create;
import static org.jboss.as.messaging.OperationDefinitionHelper.createNonEmptyStringAttribute;
import static org.jboss.as.messaging.OperationDefinitionHelper.runtimeReadOnlyOperation;
import static org.jboss.as.messaging.QueueDefinition.forwardToRuntimeQueue;
import static org.jboss.dmr.ModelType.INT;
import static org.jboss.dmr.ModelType.LIST;
import static org.jboss.dmr.ModelType.LONG;
import static org.jboss.dmr.ModelType.STRING;
import java.util.ArrayList;
import java.util.List;
import org.hornetq.api.core.management.QueueControl;
import org.hornetq.api.core.management.ResourceNames;
import org.hornetq.core.server.HornetQServer;
import org.jboss.as.controller.AttributeDefinition;
import org.jboss.as.controller.OperationContext;
import org.jboss.as.controller.OperationFailedException;
import org.jboss.as.controller.descriptions.ResourceDescriptionResolver;
import org.jboss.as.controller.operations.validation.AllowedValuesValidator;
import org.jboss.as.controller.operations.validation.ModelTypeValidator;
import org.jboss.as.controller.registry.ManagementResourceRegistration;
import org.jboss.dmr.ModelNode;
import org.jboss.dmr.ModelType;
/**
* Handler for runtime operations that invoke on a HornetQ {@link org.hornetq.api.core.management.QueueControl}.
*
* @author Brian Stansberry (c) 2011 Red Hat Inc.
*/
public class QueueControlHandler extends AbstractQueueControlHandler<QueueControl> {
public static final QueueControlHandler INSTANCE = new QueueControlHandler();
public static final String LIST_SCHEDULED_MESSAGES = "list-scheduled-messages";
public static final String LIST_SCHEDULED_MESSAGES_AS_JSON = "list-scheduled-messages-as-json";
private static final AttributeDefinition MESSAGE_ID = create(CommonAttributes.MESSAGE_ID, LONG)
.build();
private static class TypeValidator extends ModelTypeValidator implements AllowedValuesValidator {
private TypeValidator() {
super(ModelType.INT);
}
@Override
public List<ModelNode> getAllowedValues() {
List<ModelNode> values = new ArrayList<>();
values.add(new ModelNode(0));
values.add(new ModelNode(2));
values.add(new ModelNode(3));
values.add(new ModelNode(4));
values.add(new ModelNode(5));
values.add(new ModelNode(6));
return values;
}
};
private QueueControlHandler() {
}
@Override
protected AttributeDefinition getMessageIDAttributeDefinition() {
return MESSAGE_ID;
}
@Override
protected AttributeDefinition[] getReplyMessageParameterDefinitions() {
return new AttributeDefinition[] {
createNonEmptyStringAttribute("messageID"),
createNonEmptyStringAttribute("userID"),
createNonEmptyStringAttribute("address"),
create("type", INT)
.setValidator(new TypeValidator())
.build(),
create("durable", INT)
.build(),
create("expiration", LONG)
.build(),
create("priority", INT)
.setValidator(PRIORITY_VALIDATOR)
.build()
};
}
@Override
public void registerOperations(ManagementResourceRegistration registry, ResourceDescriptionResolver resolver) {
super.registerOperations(registry, resolver);
registry.registerOperationHandler(runtimeReadOnlyOperation(LIST_SCHEDULED_MESSAGES, resolver)
.setReplyType(LIST)
.setReplyParameters(getReplyMessageParameterDefinitions())
.build(),
this);
registry.registerOperationHandler(runtimeReadOnlyOperation(LIST_SCHEDULED_MESSAGES_AS_JSON, resolver)
.setReplyType(STRING)
.build(),
this);
}
/*
* Do not check whether the queue resource exists.
*
* In the core queue resource does not exist, the {@link #executeRuntimeStep(org.jboss.as.controller.OperationContext, org.jboss.dmr.ModelNode)}
* will forward the operation to the corresponding runtime-queue.
*/
@Override
protected boolean resourceMustExist(OperationContext context, ModelNode operation) {
return false;
}
@Override
protected void executeRuntimeStep(OperationContext context, ModelNode operation) throws OperationFailedException {
if (forwardToRuntimeQueue(context, operation, INSTANCE)) {
context.stepCompleted();
} else {
super.executeRuntimeStep(context, operation);
}
}
@Override
protected Object handleAdditionalOperation(String operationName, ModelNode operation, OperationContext context,
QueueControl queueControl) throws OperationFailedException {
try {
if (LIST_SCHEDULED_MESSAGES.equals(operationName)) {
String json = queueControl.listScheduledMessagesAsJSON();
context.getResult().set(ModelNode.fromJSONString(json));
} else if (LIST_SCHEDULED_MESSAGES_AS_JSON.equals(operationName)) {
context.getResult().set(queueControl.listScheduledMessagesAsJSON());
} else {
// Bug
throwUnimplementedOperationException(operationName);
}
} catch (RuntimeException e) {
throw e;
} catch (Exception e) {
context.getFailureDescription().set(e.getLocalizedMessage());
}
return null;
}
@Override
protected void revertAdditionalOperation(String operationName, ModelNode operation, OperationContext context, QueueControl queueControl, Object handback) {
// no-op
}
@Override
protected DelegatingQueueControl<QueueControl> getQueueControl(HornetQServer hqServer, String queueName) {
final QueueControl control = QueueControl.class.cast(hqServer.getManagementService().getResource(ResourceNames.CORE_QUEUE + queueName));
if (control == null) {
return null;
}
return new DelegatingQueueControl<QueueControl>() {
public QueueControl getDelegate() {
return control;
}
@Override
public String listMessagesAsJSON(String filter) throws Exception {
return control.listMessagesAsJSON(filter);
}
@Override
public long countMessages(String filter) throws Exception {
return control.countMessages(filter);
}
@Override
public boolean removeMessage(ModelNode id) throws Exception {
return control.removeMessage(id.asLong());
}
@Override
public int removeMessages(String filter) throws Exception {
return control.removeMessages(filter);
}
@Override
public int expireMessages(String filter) throws Exception {
return control.expireMessages(filter);
}
@Override
public boolean expireMessage(ModelNode id) throws Exception {
return control.expireMessage(id.asLong());
}
@Override
public boolean sendMessageToDeadLetterAddress(ModelNode id) throws Exception {
return control.sendMessageToDeadLetterAddress(id.asLong());
}
@Override
public int sendMessagesToDeadLetterAddress(String filter) throws Exception {
return control.sendMessagesToDeadLetterAddress(filter);
}
@Override
public boolean changeMessagePriority(ModelNode id, int priority) throws Exception {
return control.changeMessagePriority(id.asLong(), priority);
}
@Override
public int changeMessagesPriority(String filter, int priority) throws Exception {
return control.changeMessagesPriority(filter, priority);
}
@Override
public boolean moveMessage(ModelNode id, String otherQueue) throws Exception {
return control.moveMessage(id.asLong(), otherQueue);
}
@Override
public boolean moveMessage(ModelNode id, String otherQueue, boolean rejectDuplicates) throws Exception {
return control.moveMessage(id.asLong(), otherQueue, rejectDuplicates);
}
@Override
public int moveMessages(String filter, String otherQueue) throws Exception {
return control.moveMessages(filter, otherQueue);
}
@Override
public int moveMessages(String filter, String otherQueue, boolean rejectDuplicates) throws Exception {
return control.moveMessages(filter, otherQueue, rejectDuplicates);
}
@Override
public String listMessageCounter() throws Exception {
return control.listMessageCounter();
}
@Override
public void resetMessageCounter() throws Exception {
control.resetMessageCounter();
}
@Override
public String listMessageCounterAsHTML() throws Exception {
return control.listMessageCounterAsHTML();
}
@Override
public String listMessageCounterHistory() throws Exception {
return control.listMessageCounterHistory();
}
@Override
public String listMessageCounterHistoryAsHTML() throws Exception {
return control.listMessageCounterHistoryAsHTML();
}
@Override
public void pause() throws Exception {
control.pause();
}
@Override
public void resume() throws Exception {
control.resume();
}
@Override
public String listConsumersAsJSON() throws Exception {
return control.listConsumersAsJSON();
}
};
}
}