KickJava   Java API By Example, From Geeks To Geeks.

Java > Open Source Codes > com > quadcap > sql > meta > MetaPrimaryKeys


1 package com.quadcap.sql.meta;
2
3 /* Copyright 1999 - 2003 Quadcap Software. All rights reserved.
4  *
5  * This software is distributed under the Quadcap Free Software License.
6  * This software may be used or modified for any purpose, personal or
7  * commercial. Open Source redistributions are permitted. Commercial
8  * redistribution of larger works derived from, or works which bundle
9  * this software requires a "Commercial Redistribution License"; see
10  * http://www.quadcap.com/purchase.
11  *
12  * Redistributions qualify as "Open Source" under one of the following terms:
13  *
14  * Redistributions are made at no charge beyond the reasonable cost of
15  * materials and delivery.
16  *
17  * Redistributions are accompanied by a copy of the Source Code or by an
18  * irrevocable offer to provide a copy of the Source Code for up to three
19  * years at the cost of materials and delivery. Such redistributions
20  * must allow further use, modification, and redistribution of the Source
21  * Code under substantially the same terms as this license.
22  *
23  * Redistributions of source code must retain the copyright notices as they
24  * appear in each source code file, these license terms, and the
25  * disclaimer/limitation of liability set forth as paragraph 6 below.
26  *
27  * Redistributions in binary form must reproduce this Copyright Notice,
28  * these license terms, and the disclaimer/limitation of liability set
29  * forth as paragraph 6 below, in the documentation and/or other materials
30  * provided with the distribution.
31  *
32  * The Software is provided on an "AS IS" basis. No warranty is
33  * provided that the Software is free of defects, or fit for a
34  * particular purpose.
35  *
36  * Limitation of Liability. Quadcap Software shall not be liable
37  * for any damages suffered by the Licensee or any third party resulting
38  * from use of the Software.
39  */
40
41 import java.io.IOException JavaDoc;
42
43 import java.util.Iterator JavaDoc;
44
45 import java.sql.DatabaseMetaData JavaDoc;
46 import java.sql.ResultSetMetaData JavaDoc;
47 import java.sql.SQLException JavaDoc;
48
49 import com.quadcap.sql.Column;
50 import com.quadcap.sql.Constraint;
51 import com.quadcap.sql.Database;
52 import com.quadcap.sql.Expression;
53 import com.quadcap.sql.PrimaryKeyConstraint;
54 import com.quadcap.sql.Relation;
55 import com.quadcap.sql.Row;
56 import com.quadcap.sql.Session;
57 import com.quadcap.sql.StaticCursor;
58 import com.quadcap.sql.Table;
59
60 import com.quadcap.sql.index.Btree;
61 import com.quadcap.sql.index.BCursor;
62
63 import com.quadcap.sql.types.*;
64
65 import com.quadcap.util.Debug;
66
67 /**
68  * A Cursor supporting the <code>getPrimaryKeys</code> function.
69  *
70  * @author Stan Bailes
71  */
72 public class MetaPrimaryKeys extends MetaCursor {
73     static Column[] cols = {
74     new Column("TABLE_CAT", typeString), // 1
75 new Column("TABLE_SCHEM", typeString), // 2
76 new Column("TABLE_NAME", typeString), // 3
77 new Column("COLUMN_NAME", typeString), // 4
78 new Column("KEY_SEQ", typeShort), // 5
79 new Column("PK_NAME", typeString) // 6
80 };
81
82     static int[] sortColumns = { 4 };
83
84     public MetaPrimaryKeys(Session session, Expression predicate)
85     throws SQLException JavaDoc
86     {
87     super(session, predicate);
88     try {
89         addColumns(cols);
90         Database db = session.getDatabase();
91             session.getTableWriteLock("#Schema");
92             synchronized (db.getFile().getLock()) {
93                 Iterator JavaDoc iter = db.getRelationNameIterator();
94                 while (iter.hasNext()) {
95                     String JavaDoc name = (String JavaDoc)iter.next();
96                     Relation r = db.getRelation(name);
97                     if (r instanceof Table) {
98                         Table t = (Table)r;
99                         int num = t.getNumConstraints();
100                         for (int i = 0; i < num; i++) {
101                             Constraint c = t.getConstraint(i);
102                             if (c instanceof PrimaryKeyConstraint) {
103                                 doConstraint(t, (PrimaryKeyConstraint)c);
104                             }
105                         }
106                     }
107                 }
108             }
109         sort();
110     } catch (ValueException e) {
111         Debug.print(e);
112         SQLException JavaDoc te = new SQLException JavaDoc(e.toString(), "Q000N");
113         te.setNextException(e);
114         throw te;
115     } catch (IOException JavaDoc e) {
116         Debug.print(e);
117         throw new SQLException JavaDoc(e.toString(), "Q000O");
118     }
119     }
120
121     public int[] getSortColumns() {
122     return sortColumns;
123     }
124
125     void doConstraint(Table t, PrimaryKeyConstraint c) throws SQLException JavaDoc {
126     int[] colIndices = c.getColumns();
127     for (int i = 0; i < colIndices.length; i++) {
128         Column col = t.getColumn(colIndices[i]);
129         Row row = doColumn(t, c, i, col);
130         if (rowMatch(row)) {
131         addRow(row);
132         }
133     }
134     }
135
136     Row doColumn(Table t, PrimaryKeyConstraint c, int pos, Column col)
137     throws SQLException JavaDoc
138     {
139     Type type = col.getType();
140     Row row = new Row(6);
141     row.set(1, ValueNull.valueNull);
142     doTableName(2, row, t.getName());
143     row.set(4, new ValueString(col.getShortName()));
144     row.set(5, new ValueShort(pos+1));
145     row.set(6, new ValueString(c.getName()));
146     return row;
147     }
148
149 }
150
Popular Tags