EjbqlOrderByVisitor.java

package org.objectweb.jonas_ejb.lib;


import java.util.ArrayList;
import java.util.Iterator;
import java.util.Map;
import java.util.Stack;

import org.objectweb.jonas_ejb.deployment.ejbql.ASTEJBQL;
import org.objectweb.jonas_ejb.deployment.ejbql.ASTCmpPathExpression;
import org.objectweb.jonas_ejb.deployment.ejbql.ASTOrderByClause;
import org.objectweb.jonas_ejb.deployment.ejbql.ASTOrderByItem;
import org.objectweb.jonas_ejb.deployment.ejbql.ASTPath;
import org.objectweb.jonas_ejb.deployment.ejbql.SimpleNode;

import org.objectweb.medor.api.Field;
import org.objectweb.medor.query.api.OrderField;
import org.objectweb.medor.query.api.QueryTreeField;
import org.objectweb.medor.query.lib.BasicOrderField;

public class EjbqlOrderByVisitor extends EjbqlAbstractVisitor  {

    Map fields;
    ArrayList orderFields;

    public EjbqlOrderByVisitor(ASTEJBQL ejbql, Map _fields) throws Exception {
        orderFields = new ArrayList();
        fields = _fields;
        visit(ejbql);
    }

    public OrderField[] getOrderFields() {
        OrderField[] ofs = new OrderField[orderFields.size()];
        Iterator itr = orderFields.iterator();
        for (int i=0; itr.hasNext(); i++) {
            ofs[i] = (OrderField) itr.next();
        }  
        return ofs;
    }


    public Object visit(ASTOrderByClause node, Object data) {
        visit((SimpleNode) node, data);
        return null;
    }

    public Object visit(ASTOrderByItem node, Object data) {
        visit((SimpleNode) node, data);
        QueryTreeField qtf = (QueryTreeField) ((Stack) data).pop();
        orderFields.add(new BasicOrderField(qtf, !node.asc));
        return null;
    }

    public Object visit(ASTCmpPathExpression node, Object data) {
        visit((SimpleNode) node, data);
        try {
            String path = (String) ((ASTPath) ((Stack) data).pop()).value;
            // FIXME check type for cmp field
            ((Stack) data).push((Field) fields.get(path));
        } catch (Exception e) {
            throw new VisitorException(e);
        }
        return null;
    }

    public Object visit(ASTPath node, Object data) {
        ((Stack) data).push(node);
        return null;
    }

}

---