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-2006 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.db.explorer.actions;
043:
044: import java.sql.ResultSet;
045: import java.text.MessageFormat;
046: import java.util.HashMap;
047: import java.util.HashSet;
048: import java.util.Iterator;
049: import java.util.Vector;
050: import org.netbeans.modules.db.explorer.DbUtilities;
051:
052: import org.openide.DialogDisplayer;
053: import org.openide.NotifyDescriptor;
054: import org.openide.nodes.Node;
055:
056: import org.netbeans.lib.ddl.impl.CreateIndex;
057: import org.netbeans.lib.ddl.impl.DriverSpecification;
058: import org.netbeans.lib.ddl.impl.DropIndex;
059: import org.netbeans.lib.ddl.impl.Specification;
060:
061: import org.netbeans.modules.db.explorer.dlg.ColumnItem;
062: import org.netbeans.modules.db.explorer.dlg.LabeledComboDialog;
063: import org.netbeans.modules.db.explorer.nodes.DatabaseNode;
064: import org.netbeans.modules.db.explorer.infos.DatabaseNodeInfo;
065:
066: public class AddToIndexAction extends DatabaseAction {
067: static final long serialVersionUID = -1416260930649261633L;
068:
069: protected boolean enable(Node[] activatedNodes) {
070: return (activatedNodes != null && activatedNodes.length == 1);
071: }
072:
073: public void performAction(Node[] activatedNodes) {
074: Node node;
075: if (activatedNodes != null && activatedNodes.length > 0)
076: node = activatedNodes[0];
077: else
078: return;
079:
080: try {
081: DatabaseNodeInfo info = (DatabaseNodeInfo) node
082: .getCookie(DatabaseNodeInfo.class);
083: DatabaseNodeInfo nfo = info.getParent(nodename);
084:
085: String tablename = (String) nfo.get(DatabaseNode.TABLE);
086:
087: Specification spec = (Specification) nfo.getSpecification();
088: DriverSpecification drvSpec = info.getDriverSpecification();
089: String index = (String) nfo.get(DatabaseNode.INDEX);
090:
091: // List columns used in current index (do not show)
092: HashSet ixrm = new HashSet();
093:
094: drvSpec.getIndexInfo(tablename, false, true);
095: ResultSet rs = drvSpec.getResultSet();
096: HashMap rset = new HashMap();
097: boolean isUQ = false;
098: String ixname;
099: while (rs.next()) {
100: rset = drvSpec.getRow();
101: ixname = (String) rset.get(new Integer(6));
102:
103: if (!index.equals(ixname))
104: continue;
105:
106: String colname = (String) rset.get(new Integer(9));
107: ixrm.add(colname);
108:
109: String val = (String) rset.get(new Integer(4));
110: if (val.equals("1"))
111: isUQ = false;
112: else
113: isUQ = !(Boolean.valueOf(val).booleanValue());
114:
115: rset.clear();
116: }
117: rs.close();
118:
119: // List columns not present in current index
120: Vector cols = new Vector(5);
121:
122: drvSpec.getColumns(tablename, "%");
123: rs = drvSpec.getResultSet();
124: while (rs.next()) {
125: rset = drvSpec.getRow();
126: String colname = (String) rset.get(new Integer(4));
127: if (!ixrm.contains(colname))
128: cols.add(colname);
129: rset.clear();
130: }
131: rs.close();
132: if (cols.size() == 0)
133: throw new Exception(bundle().getString(
134: "EXC_NoUsableColumnInPlace")); // NOI18N
135:
136: // Create and execute command
137:
138: LabeledComboDialog dlg = new LabeledComboDialog(bundle()
139: .getString("AddToIndexTitle"), bundle().getString(
140: "AddToIndexLabel"), cols); // NOI18N
141: String selectedCol;
142: if (dlg.run()) {
143: AddToIndexDDL ddl = new AddToIndexDDL(spec,
144: (String) info.get(DatabaseNodeInfo.SCHEMA),
145: tablename);
146:
147: selectedCol = (String) dlg.getSelectedItem();
148: ixrm.add(selectedCol);
149: ddl.execute(index, isUQ, ixrm);
150: }
151:
152: } catch (Exception exc) {
153: DbUtilities.reportError(bundle().getString(
154: "ERR_UnableToAddColumn"), exc.getMessage()); // NOI18N
155: }
156: }
157: }
|