001: /*
002: * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
003: *
004: * Copyright 1997-2007 Sun Microsystems, Inc. All rights reserved.
005: *
006: * The contents of this file are subject to the terms of either the GNU
007: * General Public License Version 2 only ("GPL") or the Common
008: * Development and Distribution License("CDDL") (collectively, the
009: * "License"). You may not use this file except in compliance with the
010: * License. You can obtain a copy of the License at
011: * http://www.netbeans.org/cddl-gplv2.html
012: * or nbbuild/licenses/CDDL-GPL-2-CP. See the License for the
013: * specific language governing permissions and limitations under the
014: * License. When distributing the software, include this License Header
015: * Notice in each file and include the License file at
016: * nbbuild/licenses/CDDL-GPL-2-CP. Sun designates this
017: * particular file as subject to the "Classpath" exception as provided
018: * by Sun in the GPL Version 2 section of the License file that
019: * accompanied this code. If applicable, add the following below the
020: * License Header, with the fields enclosed by brackets [] replaced by
021: * your own identifying information:
022: * "Portions Copyrighted [year] [name of copyright owner]"
023: *
024: * Contributor(s):
025: *
026: * The Original Software is NetBeans. The Initial Developer of the Original
027: * Software is Sun Microsystems, Inc. Portions Copyright 1997-2007 Sun
028: * Microsystems, Inc. All Rights Reserved.
029: *
030: * If you wish your version of this file to be governed by only the CDDL
031: * or only the GPL Version 2, indicate your decision by adding
032: * "[Contributor] elects to include this software in this distribution
033: * under the [CDDL or GPL Version 2] license." If you do not indicate a
034: * single choice of license, a recipient has the option to distribute
035: * your version of this file under either the CDDL, the GPL Version 2 or
036: * to extend the choice of license to its licensees as provided above.
037: * However, if you add GPL Version 2 code and therefore, elected the GPL
038: * Version 2 license, then the option applies only if the new code is
039: * made subject to such option by the copyright holder.
040: */
041: package org.netbeans.modules.sql.framework.model.visitors;
042:
043: import java.util.HashSet;
044: import java.util.Iterator;
045: import java.util.Map;
046: import java.util.Set;
047:
048: import org.netbeans.modules.sql.framework.model.ColumnRef;
049: import org.netbeans.modules.sql.framework.model.SQLDBColumn;
050: import org.netbeans.modules.sql.framework.model.SQLInputObject;
051: import org.netbeans.modules.sql.framework.model.SQLObject;
052: import org.netbeans.modules.sql.framework.model.SQLOperator;
053:
054: import com.sun.sql.framework.exception.BaseException;
055:
056: /**
057: * @author Girish Patil
058: * @version Revision
059: */
060: public class ColumnsUsedInSQLOperatorVisitor {
061:
062: /**
063: * Visits each of Expression Object Nodes.
064: *
065: * @param operator SQLOperator
066: * @return Set of columns used in the expression.
067: * @throws BaseException
068: */
069: public static Set visit(SQLOperator operator) throws BaseException {
070: if (operator == null) {
071: return null;
072: }
073:
074: Set columnsNames = new HashSet();
075: Map inputMap = operator.getInputObjectMap();
076:
077: Set keys = null;
078: Iterator keyItr = null;
079: Object keyObject = null;
080: Object operandObject = null;
081:
082: if (inputMap != null) {
083: keys = inputMap.keySet();
084:
085: if (keys != null) {
086: keyItr = keys.iterator();
087: Set columnNamesFromChildOperators = null;
088:
089: while (keyItr.hasNext()) {
090: keyObject = keyItr.next();
091: if (keyObject != null) {
092: operandObject = inputMap.get(keyObject);
093:
094: if (operandObject instanceof SQLInputObject) {
095: SQLInputObject inputObj = (SQLInputObject) operandObject;
096: SQLObject sqlObject = inputObj
097: .getSQLObject();
098:
099: if (sqlObject instanceof SQLOperator) {
100: columnNamesFromChildOperators = visit((SQLOperator) sqlObject);
101: columnsNames
102: .addAll(columnNamesFromChildOperators);
103: } else {
104: if (sqlObject instanceof ColumnRef) {
105: ColumnRef conditionColumn = (ColumnRef) sqlObject;
106: SQLDBColumn column = (SQLDBColumn) conditionColumn
107: .getColumn();
108: String colName = column.getName();
109: columnsNames.add(colName);
110: }
111: }
112: }
113: }
114: }
115: }
116: }
117: return columnsNames;
118: }
119:
120: }
|