package org.geoserver.feature.retype;
import java.util.HashSet;
import java.util.Iterator;
import java.util.Set;
import org.geoserver.feature.RetypingFeatureCollection;
import org.geotools.data.DataUtilities;
import org.geotools.data.store.ReTypingFeatureCollection;
import org.geotools.factory.CommonFactoryFinder;
import org.geotools.filter.identity.FeatureIdImpl;
import org.geotools.filter.visitor.DuplicatingFilterVisitor;
import org.opengis.filter.Filter;
import org.opengis.filter.Id;
import org.opengis.filter.identity.FeatureId;
/**
* Takes a filter that eventually contains a fid filter and builds a new filter that contains
* the same fids but with a different prefix
* @author Andrea Aime
*/
class FidTransformeVisitor extends DuplicatingFilterVisitor {
private Object featureType;
private FeatureTypeMap map;
public FidTransformeVisitor(FeatureTypeMap map) {
super(CommonFactoryFinder.getFilterFactory2(null));
this.map = map;
}
public Object visit(Id filter, Object extraData) {
Set ids = filter.getIDs();
if (ids.isEmpty()) {
throw new IllegalArgumentException(
"Invalid fid filter provides, has no fids inside");
}
Filter external = null;
Set<FeatureId> fids = new HashSet<FeatureId>();
for (Iterator it = ids.iterator(); it.hasNext();) {
FeatureId id = new FeatureIdImpl((String) it.next());
FeatureId retyped = RetypingFeatureCollection.reTypeId(id, map.getFeatureType(), map.getOriginalFeatureType());
fids.add(retyped);
}
return ff.id(fids);
}
}