/**
* 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.camel.model;
import java.util.Collections;
import java.util.Comparator;
import java.util.List;
import javax.xml.bind.annotation.XmlAccessType;
import javax.xml.bind.annotation.XmlAccessorType;
import javax.xml.bind.annotation.XmlAttribute;
import javax.xml.bind.annotation.XmlElement;
import javax.xml.bind.annotation.XmlRootElement;
import javax.xml.bind.annotation.XmlTransient;
import org.apache.camel.Expression;
import org.apache.camel.Processor;
import org.apache.camel.processor.SortProcessor;
import org.apache.camel.spi.RouteContext;
import org.apache.camel.util.ObjectHelper;
import static org.apache.camel.builder.ExpressionBuilder.bodyExpression;
/**
* Represents an XML <sort/> element
*/
@XmlRootElement(name = "sort")
@XmlAccessorType(XmlAccessType.FIELD)
public class SortDefinition extends OutputDefinition<SortDefinition> {
@XmlTransient
private Comparator<Object> comparator;
@XmlAttribute(required = false)
private String comparatorRef;
@XmlElement(name = "expression", required = false)
private ExpressionSubElementDefinition expression;
public SortDefinition() {
}
public SortDefinition(Expression expression) {
setExpression(expression);
}
public SortDefinition(Expression expression, Comparator<Object> comparator) {
this(expression);
this.comparator = comparator;
}
@Override
public String toString() {
return "sort[" + getExpression() + " by: " + (comparatorRef != null ? "ref:" + comparatorRef : comparator) + "]";
}
@Override
public String getShortName() {
return "sort";
}
@Override
@SuppressWarnings("unchecked")
public Processor createProcessor(RouteContext routeContext) throws Exception {
// lookup in registry
if (ObjectHelper.isNotEmpty(comparatorRef)) {
comparator = routeContext.getCamelContext().getRegistry().lookup(comparatorRef, Comparator.class);
}
// if no comparator then default on to string representation
if (comparator == null) {
comparator = new Comparator<Object>() {
public int compare(Object o1, Object o2) {
return ObjectHelper.compare(o1, o2);
}
};
}
// if no expression provided then default to body expression
if (getExpression() == null) {
setExpression(bodyExpression());
}
Expression exp = expression.getExpression();
// fallback to the type when its been initialized from JAXB using camel-spring
if (exp == null) {
exp = expression.getExpressionType();
}
return new SortProcessor(exp, getComparator());
}
@Override
@SuppressWarnings("unchecked")
public List<ProcessorDefinition> getOutputs() {
return Collections.EMPTY_LIST;
}
public Comparator<Object> getComparator() {
return comparator;
}
public void setComparator(Comparator<Object> comparator) {
this.comparator = comparator;
}
public String getComparatorRef() {
return comparatorRef;
}
public void setComparatorRef(String comparatorRef) {
this.comparatorRef = comparatorRef;
}
public ExpressionSubElementDefinition getExpression() {
return expression;
}
public void setExpression(Expression expression) {
this.expression = new ExpressionSubElementDefinition(expression);
}
public void setExpression(ExpressionSubElementDefinition expression) {
this.expression = expression;
}
/**
* Sets the comparator to use for sorting
*
* @param comparator the comparator to use for sorting
* @return the builder
*/
public SortDefinition comparator(Comparator<Object> comparator) {
setComparator(comparator);
return this;
}
/**
* Sets a reference to lookup for the comparator to use for sorting
*
* @param ref reference for the comparator
* @return the builder
*/
public SortDefinition comparatorRef(String ref) {
setComparatorRef(ref);
return this;
}
}