/*
* Copyright 2013 Red Hat Inc. and/or its affiliates and other contributors.
*
* Licensed 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.switchyard.as7.extension.admin;
import static org.jboss.as.controller.descriptions.ModelDescriptionConstants.NAME;
import static org.switchyard.as7.extension.SwitchYardModelConstants.APPLICATION_NAME;
import javax.xml.namespace.QName;
import org.jboss.as.controller.OperationContext;
import org.jboss.as.controller.OperationFailedException;
import org.jboss.as.controller.OperationStepHandler;
import org.jboss.dmr.ModelNode;
import org.jboss.msc.service.ServiceController;
import org.switchyard.admin.Application;
import org.switchyard.admin.Reference;
import org.switchyard.admin.Service;
import org.switchyard.admin.SwitchYard;
import org.switchyard.as7.extension.services.SwitchYardAdminService;
/**
* SwitchYardSubsystemShowMetrics
*
* Operation for resetting metrics for services/references deployed on the
* SwitchYard subsystem.
*/
public final class SwitchYardSubsystemResetMetrics implements OperationStepHandler {
/**
* The global instance for this operation.
*/
public static final SwitchYardSubsystemResetMetrics INSTANCE = new SwitchYardSubsystemResetMetrics();
private SwitchYardSubsystemResetMetrics() {
// forbidden inheritance
}
/*
* (non-Javadoc)
*
* @see
* org.jboss.as.controller.OperationStepHandler#execute(org.jboss.as.controller
* .OperationContext, org.jboss.dmr.ModelNode)
*/
@Override
public void execute(final OperationContext context, final ModelNode operation) throws OperationFailedException {
context.addStep(new OperationStepHandler() {
@Override
public void execute(final OperationContext context, final ModelNode operation)
throws OperationFailedException {
final ServiceController<?> controller = context.getServiceRegistry(false).getRequiredService(
SwitchYardAdminService.SERVICE_NAME);
SwitchYard switchYard = SwitchYard.class.cast(controller.getService().getValue());
if (operation.hasDefined(NAME)) {
final QName serviceQName = QName.valueOf(operation.get(NAME).asString());
final QName applicationQName = operation.hasDefined(APPLICATION_NAME) ? QName.valueOf(operation
.get(APPLICATION_NAME).asString()) : null;
APPLICATION_LOOP: for (Application application : switchYard.getApplications()) {
if (applicationQName == null || applicationQName.equals(application.getName())) {
for (Service service : application.getServices()) {
if (serviceQName.equals(service.getName())) {
// XXX: we should really be doing this on
// the service
service.getPromotedService().resetMessageMetrics();
if (applicationQName == null) {
continue APPLICATION_LOOP;
} else {
break APPLICATION_LOOP;
}
}
}
for (Reference reference : application.getReferences()) {
if (serviceQName.equals(reference.getName())) {
reference.resetMessageMetrics();
if (applicationQName == null) {
continue APPLICATION_LOOP;
} else {
break APPLICATION_LOOP;
}
}
}
}
}
} else {
switchYard.resetMessageMetrics();
}
context.stepCompleted();
}
}, OperationContext.Stage.RUNTIME);
context.stepCompleted();
}
}