/*
* 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.synapse.mediators.throttle;
import org.apache.axiom.om.OMAttribute;
import org.apache.axiom.om.OMElement;
import org.apache.synapse.Mediator;
import org.apache.synapse.config.xml.AbstractMediatorFactory;
import org.apache.synapse.config.xml.SequenceMediatorFactory;
import org.apache.synapse.config.xml.XMLConfigConstants;
import javax.xml.namespace.QName;
import java.util.Properties;
/**
* The Factory for create throttle mediator- key or InLine XMl need to provide
*
* <pre>
* <throttle [onReject="string"] [onAccept="string"] id="string">
* (<policy key="string"/> | <policy>..</policy>)
* <onReject>..</onReject>?
* <onAccept>..</onAccept>?
* </throttle>
* </pre>
*/
public class ThrottleMediatorFactory extends AbstractMediatorFactory {
/**
* The Tag Name for throttle
*/
private static final QName TAG_NAME
= new QName(XMLConfigConstants.SYNAPSE_NAMESPACE, "throttle");
protected Mediator createSpecificMediator(OMElement elem, Properties properties) {
ThrottleMediator throttleMediator = new ThrottleMediator();
OMElement policy = elem.getFirstChildWithName(
new QName(XMLConfigConstants.SYNAPSE_NAMESPACE, "policy"));
if (policy != null) {
OMAttribute key = policy.getAttribute(new QName(XMLConfigConstants.NULL_NAMESPACE, "key"));
if (key != null) {
String keyValue = key.getAttributeValue();
if (keyValue != null && !"".equals(keyValue)) {
throttleMediator.setPolicyKey(keyValue);
} else {
handleException("key attribute should have a value ");
}
} else {
OMElement inLine = policy.getFirstElement();
if (inLine != null) {
throttleMediator.setInLinePolicy(inLine);
}
}
}
// after successfully creating the mediator
// set its common attributes such as tracing etc
processAuditStatus(throttleMediator,elem);
String id = elem.getAttributeValue(new QName(XMLConfigConstants.NULL_NAMESPACE, "id"));
if (id != null && !"".equals(id)) {
throttleMediator.setId(id.trim());
} else {
handleException("Idy attribute must have defined ");
}
SequenceMediatorFactory mediatorFactory = new SequenceMediatorFactory();
OMAttribute onReject = elem.getAttribute(
new QName(XMLConfigConstants.NULL_NAMESPACE, XMLConfigConstants.ONREJECT));
if (onReject != null) {
String onRejectValue = onReject.getAttributeValue();
if (onRejectValue != null) {
throttleMediator.setOnRejectSeqKey(onRejectValue.trim());
}
} else {
OMElement onRejectMediatorElement = elem.getFirstChildWithName(
new QName(XMLConfigConstants.SYNAPSE_NAMESPACE, XMLConfigConstants.ONREJECT));
if (onRejectMediatorElement != null) {
throttleMediator.setOnRejectMediator(mediatorFactory.createAnonymousSequence(
onRejectMediatorElement, properties));
}
}
OMAttribute onAccept = elem.getAttribute(
new QName(XMLConfigConstants.NULL_NAMESPACE, XMLConfigConstants.ONACCEPT));
if (onAccept != null) {
String onAcceptValue = onAccept.getAttributeValue();
if (onAcceptValue != null) {
throttleMediator.setOnAcceptSeqKey(onAcceptValue);
}
} else {
OMElement onAcceptMediatorElement = elem.getFirstChildWithName(
new QName(XMLConfigConstants.SYNAPSE_NAMESPACE, XMLConfigConstants.ONACCEPT));
if (onAcceptMediatorElement != null) {
throttleMediator.setOnAcceptMediator(mediatorFactory.createAnonymousSequence(
onAcceptMediatorElement, properties));
}
}
return throttleMediator;
}
public QName getTagQName() {
return TAG_NAME;
}
}