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.test.jdbcstub;
043:
044: import java.sql.SQLException;
045: import java.util.HashMap;
046: import java.util.Iterator;
047: import java.util.List;
048: import java.util.Map;
049: import org.netbeans.modules.j2ee.persistence.editor.completion.*;
050: import org.netbeans.test.stub.api.StubDelegate;
051:
052: /**
053: *
054: * @author Andrei Badea
055: */
056: public class ResultSetImpl extends StubDelegate {
057:
058: private List/*<List<Object>>*/columns;
059: private Map/*<String, List<Object>*/names2iterators = new HashMap();
060: private Map/*<String, Object>*/names2values; // current row values
061:
062: public ResultSetImpl(List columns) {
063: this .columns = columns;
064:
065: for (Iterator it = columns.iterator(); it.hasNext();) {
066: List column = (List) it.next();
067: Iterator columnIterator = column.iterator();
068: String columnName = columnIterator.next().toString();
069: names2iterators.put(columnName, columnIterator);
070: }
071: }
072:
073: public boolean next() {
074: if (names2values != null) {
075: names2values.clear();
076: } else {
077: names2values = new HashMap();
078: }
079:
080: Iterator it = names2iterators.entrySet().iterator();
081: if (!it.hasNext()) {
082: return false;
083: }
084:
085: while (it.hasNext()) {
086: Map.Entry entry = (Map.Entry) it.next();
087: String columnName = (String) entry.getKey();
088: Iterator columnIterator = (Iterator) entry.getValue();
089:
090: if (!columnIterator.hasNext()) {
091: return false;
092: }
093:
094: Object value = columnIterator.next();
095: names2values.put(columnName, value);
096: }
097:
098: return true;
099: }
100:
101: public Object getObject(String columnName) throws SQLException {
102: if (names2values == null) {
103: throw new SQLException(
104: "The next() method has not been called yet");
105: }
106: if (!names2values.containsKey(columnName)) {
107: throw new SQLException("Unknown column name " + columnName
108: + ".");
109: }
110: return names2values.get(columnName);
111: }
112:
113: public short getShort(String columnName) throws SQLException {
114: Object value = getObject(columnName);
115: if (value instanceof Short) {
116: return ((Short) value).shortValue();
117: } else {
118: throw new SQLException(value + "is not a short.");
119: }
120: }
121:
122: public int getInt(String columnName) throws SQLException {
123: Object value = getObject(columnName);
124: if (value instanceof Integer) {
125: return ((Integer) value).intValue();
126: } else {
127: throw new SQLException(value + " is not an int.");
128: }
129: }
130:
131: public boolean getBoolean(String columnName) throws SQLException {
132: Object value = getObject(columnName);
133: if (value instanceof Boolean) {
134: return ((Boolean) value).booleanValue();
135: } else {
136: throw new SQLException(value + " is not a boolean.");
137: }
138: }
139:
140: public String getString(String columnName) throws SQLException {
141: Object value = getObject(columnName);
142: return value != null ? value.toString() : null;
143: }
144:
145: public void close() {
146: }
147: }
|