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:
042: package org.netbeans.modules.sql.framework.codegen.sqlserver;
043:
044: import java.util.ArrayList;
045: import java.util.Iterator;
046: import java.util.List;
047:
048: import org.apache.velocity.VelocityContext;
049: import org.netbeans.modules.sql.framework.codegen.AbstractDB;
050: import org.netbeans.modules.sql.framework.codegen.ResolvedMapping;
051: import org.netbeans.modules.sql.framework.codegen.StatementContext;
052: import org.netbeans.modules.sql.framework.codegen.base.BaseStatements;
053: import org.netbeans.modules.sql.framework.model.SQLConstants;
054: import org.netbeans.modules.sql.framework.model.TargetTable;
055:
056: import com.sun.sql.framework.exception.BaseException;
057:
058: /**
059: * @author Ritesh Adval
060: * @version $Revision$
061: */
062: public class SqlServerStatements extends BaseStatements {
063:
064: public SqlServerStatements(AbstractDB database) {
065: super (database);
066: }
067:
068: protected void populateContextForUpdate(TargetTable targetTable,
069: StatementContext context, VelocityContext vContext)
070: throws BaseException {
071: StatementContext localContext = new StatementContext();
072: if (context != null) {
073: localContext.putAll(context);
074: }
075:
076: vContext.put("targetTable", this .genFactory.generate(
077: targetTable, localContext));
078:
079: //SELECT START
080: localContext.putClientProperty(
081: StatementContext.USE_SOURCE_TABLE_ALIAS_NAME,
082: Boolean.TRUE);
083:
084: final boolean useKeyColumns = false;
085: List rMappings = createResolvedMappings(targetTable,
086: useKeyColumns, localContext);
087: ArrayList selectIdentifiers = new ArrayList();
088:
089: Iterator it = rMappings.iterator();
090: while (it.hasNext()) {
091: ResolvedMapping rm = (ResolvedMapping) it.next();
092: selectIdentifiers.add(rm.getSource());
093: selectIdentifiers.add(rm.getTarget());
094: }
095:
096: localContext.putClientProperty(
097: StatementContext.USE_TARGET_TABLE_ALIAS_NAME,
098: Boolean.TRUE);
099: vContext.put("fromContent", getFromStatementContentForTarget(
100: targetTable, SQLConstants.INNER_JOIN, localContext));
101:
102: vContext.put("useUpdateWhere", Boolean.FALSE);
103:
104: String condition = getWhereCondition(targetTable, localContext);
105: if (condition != null && !condition.equals("")) {
106: vContext.put("useUpdateWhere", Boolean.TRUE);
107: vContext.put("condition", condition);
108: }
109: //SELECT END
110:
111: vContext.put("mappings", rMappings);
112: }
113: }
|