/** Method declaration */
private void refreshTree() {
tTree.removeAll();
try {
int color_table = Color.yellow.getRGB();
int color_column = Color.orange.getRGB();
int color_index = Color.red.getRGB();
tTree.addRow("", dMeta.getURL(), "-", 0);
String usertables[] = {"TABLE"};
ResultSet result = dMeta.getTables(null, null, null, usertables);
Vector tables = new Vector();
// sqlbob@users Added remarks.
Vector remarks = new Vector();
while (result.next()) {
tables.addElement(result.getString(3));
remarks.addElement(result.getString(5));
}
result.close();
for (int i = 0; i < tables.size(); i++) {
String name = (String) tables.elementAt(i);
String key = "tab-" + name + "-";
tTree.addRow(key, name, "+", color_table);
// sqlbob@users Added remarks.
String remark = (String) remarks.elementAt(i);
if ((remark != null) && !remark.trim().equals("")) {
tTree.addRow(key + "r", " " + remark);
}
ResultSet col = dMeta.getColumns(null, null, name, null);
while (col.next()) {
String c = col.getString(4);
String k1 = key + "col-" + c + "-";
tTree.addRow(k1, c, "+", color_column);
String type = col.getString(6);
tTree.addRow(k1 + "t", "Type: " + type);
boolean nullable = col.getInt(11) != DatabaseMetaData.columnNoNulls;
tTree.addRow(k1 + "n", "Nullable: " + nullable);
}
col.close();
tTree.addRow(key + "ind", "Indices", "+", 0);
ResultSet ind = dMeta.getIndexInfo(null, null, name, false, false);
String oldiname = null;
while (ind.next()) {
boolean nonunique = ind.getBoolean(4);
String iname = ind.getString(6);
String k2 = key + "ind-" + iname + "-";
if ((oldiname == null || !oldiname.equals(iname))) {
tTree.addRow(k2, iname, "+", color_index);
tTree.addRow(k2 + "u", "Unique: " + !nonunique);
oldiname = iname;
}
String c = ind.getString(9);
tTree.addRow(k2 + "c-" + c + "-", c);
}
ind.close();
}
tTree.addRow("p", "Properties", "+", 0);
tTree.addRow("pu", "User: "******"pr", "ReadOnly: " + cConn.isReadOnly());
tTree.addRow("pa", "AutoCommit: " + cConn.getAutoCommit());
tTree.addRow("pd", "Driver: " + dMeta.getDriverName());
tTree.addRow("pp", "Product: " + dMeta.getDatabaseProductName());
tTree.addRow("pv", "Version: " + dMeta.getDatabaseProductVersion());
} catch (SQLException e) {
tTree.addRow("", "Error getting metadata:", "-", 0);
tTree.addRow("-", e.getMessage());
tTree.addRow("-", e.getSQLState());
}
tTree.update();
}
protected void initIndexes(ConnectionProvider cp, String tbl) {
if (cp != null)
try {
boolean unique;
DatabaseMetaData dmd = cp.getDatabaseMetaData();
String shortTableName;
if (tbl == null) shortTableName = getName().getName();
else shortTableName = tbl;
ResultSet rs;
// rs = dmd.getIndexInfo(cp.getConnection().getCatalog(),
// dmd.getUserName().trim(), shortTableName, false, true);
rs =
dmd.getIndexInfo(
cp.getConnection().getCatalog(), cp.getSchema(), shortTableName, false, true);
String name, columnName;
boolean unq;
LinkedList idxs = new LinkedList();
if (rs != null) {
HashMap rset = new HashMap();
String uniqueStr;
while (rs.next()) {
name = rs.getString("INDEX_NAME"); // NOI18N
columnName = rs.getString("COLUMN_NAME"); // NOI18N
if (columnName != null) columnName = columnName.trim();
unq = rs.getBoolean("NON_UNIQUE"); // NOI18N
// hack for PostgreSQL bug 3480: the driver returns quotes around quoted column names
if (columnName != null
&& columnName.length() >= 2
&& columnName.startsWith("\"")
&& columnName.endsWith("\"")) { // NOI18N
columnName = columnName.substring(1, columnName.length() - 1);
}
if (name == null) continue;
else name = name.trim();
if (unq) idxs.add(name + "." + columnName + ".false"); // NOI18N
else idxs.add(name + "." + columnName + ".true"); // NOI18N
}
rs.close();
}
String info;
int start, end;
for (int i = 0; i < idxs.size(); i++) {
info = idxs.get(i).toString();
start = info.indexOf('.'); // NOI18N
end = info.lastIndexOf('.'); // NOI18N
name = info.substring(0, start);
if ((info.substring(end + 1)).equals("true")) // NOI18N
unique = true;
else unique = false;
if (indexes.find(DBIdentifier.create(name)) != null) continue;
IndexElementImpl iei = new IndexElementImpl(this, name, unique);
IndexElement[] ie = {new IndexElement(iei, (TableElement) element)};
iei.initColumns(idxs);
changeIndexes(ie, DBElement.Impl.ADD);
}
} catch (Exception exc) {
if (Boolean.getBoolean("netbeans.debug.exceptions")) // NOI18N
exc.printStackTrace();
}
}
protected boolean checkExistence(Connection conn, DatabaseObjectType type, String... params)
throws SQLException {
boolean result = false;
DatabaseMetaData metadata = null;
PreparedStatement pstmt = null;
BaseSchema baseSchema = catalogStore.getSchema();
if (params == null || params.length < 1) {
throw new IllegalArgumentException("checkExistence function needs at least one argument.");
}
switch (type) {
case DATA:
metadata = conn.getMetaData();
ResultSet data =
metadata.getUDTs(
null,
baseSchema.getSchemaName() != null && !baseSchema.getSchemaName().isEmpty()
? baseSchema.getSchemaName().toUpperCase()
: null,
params[0].toUpperCase(),
null);
result = data.next();
CatalogUtil.closeQuietly(data);
break;
case FUNCTION:
metadata = conn.getMetaData();
ResultSet functions =
metadata.getFunctions(
null,
baseSchema.getSchemaName() != null && !baseSchema.getSchemaName().isEmpty()
? baseSchema.getSchemaName().toUpperCase()
: null,
params[0].toUpperCase());
result = functions.next();
CatalogUtil.closeQuietly(functions);
break;
case INDEX:
if (params.length != 2) {
throw new IllegalArgumentException(
"Finding index object is needed two strings, table name and index name");
}
pstmt = getExistQuery(conn, type);
if (pstmt != null) {
result = checkExistenceByQuery(pstmt, baseSchema, params);
} else {
metadata = conn.getMetaData();
ResultSet indexes =
metadata.getIndexInfo(
null,
baseSchema.getSchemaName() != null && !baseSchema.getSchemaName().isEmpty()
? baseSchema.getSchemaName().toUpperCase()
: null,
params[0].toUpperCase(),
false,
true);
while (indexes.next()) {
if (indexes.getString("INDEX_NAME").equals(params[1].toUpperCase())) {
result = true;
break;
}
}
CatalogUtil.closeQuietly(indexes);
}
break;
case TABLE:
pstmt = getExistQuery(conn, type);
if (pstmt != null) {
result = checkExistenceByQuery(pstmt, baseSchema, params);
} else {
metadata = conn.getMetaData();
ResultSet tables =
metadata.getTables(
null,
baseSchema.getSchemaName() != null && !baseSchema.getSchemaName().isEmpty()
? baseSchema.getSchemaName().toUpperCase()
: null,
params[0].toUpperCase(),
new String[] {"TABLE"});
result = tables.next();
CatalogUtil.closeQuietly(tables);
}
break;
case DOMAIN:
case OPERATOR:
case RULE:
case SEQUENCE:
case TRIGGER:
case VIEW:
pstmt = getExistQuery(conn, type);
if (pstmt == null) {
throw new TajoInternalError(
"Finding "
+ type
+ " type of database object is not supported on this database system.");
}
result = checkExistenceByQuery(pstmt, baseSchema, params);
break;
}
return result;
}
/* Clear all existing nodes from the tree model and rebuild from scratch.
*/
protected void refreshTree() {
DefaultMutableTreeNode propertiesNode;
DefaultMutableTreeNode leaf;
// First clear the existing tree by simply enumerating
// over the root node's children and removing them one by one.
while (treeModel.getChildCount(rootNode) > 0) {
DefaultMutableTreeNode child = (DefaultMutableTreeNode) treeModel.getChild(rootNode, 0);
treeModel.removeNodeFromParent(child);
child.removeAllChildren();
child.removeFromParent();
}
treeModel.nodeStructureChanged(rootNode);
treeModel.reload();
tScrollPane.repaint();
// Now rebuild the tree below its root
try {
// Start by naming the root node from its URL:
rootNode.setUserObject(dMeta.getURL());
// get metadata about user tables by building a vector of table names
String usertables[] = {"TABLE", "GLOBAL TEMPORARY", "VIEW"};
ResultSet result = dMeta.getTables(null, null, null, usertables);
Vector tables = new Vector();
// sqlbob@users Added remarks.
Vector remarks = new Vector();
while (result.next()) {
tables.addElement(result.getString(3));
remarks.addElement(result.getString(5));
}
result.close();
// For each table, build a tree node with interesting info
for (int i = 0; i < tables.size(); i++) {
String name = (String) tables.elementAt(i);
DefaultMutableTreeNode tableNode = makeNode(name, rootNode);
ResultSet col = dMeta.getColumns(null, null, name, null);
// sqlbob@users Added remarks.
String remark = (String) remarks.elementAt(i);
if ((remark != null) && !remark.trim().equals("")) {
makeNode(remark, tableNode);
}
// With a child for each column containing pertinent attributes
while (col.next()) {
String c = col.getString(4);
DefaultMutableTreeNode columnNode = makeNode(c, tableNode);
String type = col.getString(6);
makeNode("Type: " + type, columnNode);
boolean nullable = col.getInt(11) != DatabaseMetaData.columnNoNulls;
makeNode("Nullable: " + nullable, columnNode);
}
col.close();
DefaultMutableTreeNode indexesNode = makeNode("Indices", tableNode);
ResultSet ind = dMeta.getIndexInfo(null, null, name, false, false);
String oldiname = null;
// A child node to contain each index - and its attributes
while (ind.next()) {
DefaultMutableTreeNode indexNode = null;
boolean nonunique = ind.getBoolean(4);
String iname = ind.getString(6);
if ((oldiname == null || !oldiname.equals(iname))) {
indexNode = makeNode(iname, indexesNode);
makeNode("Unique: " + !nonunique, indexNode);
oldiname = iname;
}
// And the ordered column list for index components
makeNode(ind.getString(9), indexNode);
}
ind.close();
}
// Finally - a little additional metadata on this connection
propertiesNode = makeNode("Properties", rootNode);
makeNode("User: "******"ReadOnly: " + cConn.isReadOnly(), propertiesNode);
makeNode("AutoCommit: " + cConn.getAutoCommit(), propertiesNode);
makeNode("Driver: " + dMeta.getDriverName(), propertiesNode);
makeNode("Product: " + dMeta.getDatabaseProductName(), propertiesNode);
makeNode("Version: " + dMeta.getDatabaseProductVersion(), propertiesNode);
} catch (SQLException se) {
propertiesNode = makeNode("Error getting metadata:", rootNode);
makeNode(se.getMessage(), propertiesNode);
makeNode(se.getSQLState(), propertiesNode);
}
treeModel.nodeStructureChanged(rootNode);
treeModel.reload();
tScrollPane.repaint();
}
