All Downloads are FREE. Search and download functionalities are using the official Maven repository.
  • Log In
  • Register

    • org.modeshape.jcr.query.optimize.AddOrderingColumnsToSources Maven / Gradle / Ivy

    /*
 * ModeShape (http://www.modeshape.org)
 * See the COPYRIGHT.txt file distributed with this work for information
 * regarding copyright ownership.  Some portions may be licensed
 * to Red Hat, Inc. under one or more contributor license agreements.
 * See the AUTHORS.txt file in the distribution for a full listing of 
 * individual contributors.
 *
 * ModeShape is free software. Unless otherwise indicated, all code in ModeShape
 * is licensed to you 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.
 * 
 * ModeShape 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.modeshape.jcr.query.optimize;

import java.util.HashSet;
import java.util.LinkedList;
import java.util.List;
import java.util.Set;
import org.modeshape.jcr.query.QueryContext;
import org.modeshape.jcr.query.model.Column;
import org.modeshape.jcr.query.model.Ordering;
import org.modeshape.jcr.query.model.PropertyValue;
import org.modeshape.jcr.query.model.ReferenceValue;
import org.modeshape.jcr.query.model.SelectorName;
import org.modeshape.jcr.query.model.Visitor;
import org.modeshape.jcr.query.model.Visitors;
import org.modeshape.jcr.query.plan.PlanNode;
import org.modeshape.jcr.query.plan.PlanNode.Property;
import org.modeshape.jcr.query.plan.PlanNode.Type;

/**
 * An {@link OptimizerRule} that adds any missing columns required by the ordering specifications to the SORT node's PROJECT, and
 * to the appropriate access nodes.
 */
public class AddOrderingColumnsToSources implements OptimizerRule {

    public static final AddOrderingColumnsToSources INSTANCE = new AddOrderingColumnsToSources();

    @Override
    public PlanNode execute( QueryContext context,
                             PlanNode plan,
                             LinkedList ruleStack ) {
        final boolean includeSourceName = context.getHints().qualifyExpandedColumnNames;

        // For each of the SORT nodes ...
        for (PlanNode sortNode : plan.findAllAtOrBelow(Type.SORT)) {

            // Get the list of property and reference expressions that are used in the Ordering instances, using a visitor. Other
            // kinds of
            // dynamic operands in Ordering instances will not need to change the columns
            final Set sortColumns = new HashSet();
            Visitor columnVisitor = new Visitors.AbstractVisitor() {
                @Override
                public void visit( PropertyValue prop ) {
                    sortColumns.add(columnFor(prop.selectorName(), prop.getPropertyName(), includeSourceName));
                }

                @Override
                public void visit( ReferenceValue ref ) {
                    sortColumns.add(columnFor(ref.selectorName(), ref.getPropertyName(), includeSourceName));
                }
            };
            List orderBys = sortNode.getPropertyAsList(Property.SORT_ORDER_BY, Object.class);
            if (orderBys != null && !orderBys.isEmpty()) {
                for (Object ordering : orderBys) {
                    if (ordering instanceof Ordering) {
                        Visitors.visitAll((Ordering)ordering, columnVisitor);
                    }
                }
            }

            // Add each of the sort columns to the appropriate PROJECT nodes in the subtrees of this plan ...
            for (Column sortColumn : sortColumns) {
                addSortColumn(context, sortNode, sortColumn);
            }
        }
        return plan;
    }

    /**
     * Make sure that the supplied column is included in the {@link Property#PROJECT_COLUMNS projected columns} on the supplied
     * plan node or its children.
     * 
     * @param context the query context; may not be null
     * @param node the query plan node
     * @param sortColumn the column required by the sort
     */
    protected void addSortColumn( QueryContext context,
                                  PlanNode node,
                                  Column sortColumn ) {
        if (node.getSelectors().contains(sortColumn.selectorName())) {
            // Get the existing projected columns ...
            List columns = node.getPropertyAsList(Property.PROJECT_COLUMNS, Column.class);
            List types = node.getPropertyAsList(Property.PROJECT_COLUMN_TYPES, String.class);
            if (columns != null && addIfMissing(context, sortColumn, columns, types)) {
                node.setProperty(Property.PROJECT_COLUMNS, columns);
                node.setProperty(Property.PROJECT_COLUMN_TYPES, types);
            }
        }

        // Apply recursively ...
        for (PlanNode child : node) {
            addSortColumn(context, child, sortColumn);
        }
    }

    /**
     * Check the supplied list of columns for an existing column that matches the supplied {@link Column}, and if none is found
     * add the supplied Column to the list and add an appropriate type.
     * 
     * @param context the query context
     * @param column the column that will be added if not already in the list; may not be null
     * @param columns the list of columns; may not be null
     * @param columnTypes the list of column types; may not be null
     * @return true if the column was added (i.e., the lists were modified), or false if the lists were not modified
     */
    protected boolean addIfMissing( QueryContext context,
                                    Column column,
                                    List columns,
                                    List columnTypes ) {
        for (Column c : columns) {
            if (!c.selectorName().equals(column.selectorName())) continue;
            String cName = c.getPropertyName();
            if (cName.equals(column.getPropertyName()) || cName.equals(column.getColumnName())) return false;
            cName = c.getColumnName();
            if (cName.equals(column.getPropertyName()) || cName.equals(column.getColumnName())) return false;
        }
        columns.add(column);
        columnTypes.add(context.getTypeSystem().getDefaultType());
        return true;
    }

    protected Column columnFor( SelectorName selector,
                                String property,
                                boolean includeSourceName ) {
        String columnName = includeSourceName ? selector.getString() + "." + property : property;
        return new Column(selector, property, columnName);
    }

    @Override
    public String toString() {
        return getClass().getSimpleName();
    }
}
    
    
    
    

    




    
    
    © 2015 - 2025 Weber Informatics LLC | Privacy Policy