This method returns a list of entity types supported by the database + * adapter. This list will always be limited to Tables, Views and + * Sequences.
+ * + *Not all databases support all types. MySQL only supports + * Tables. Informix supports Tables and Views. Oracle and DB2 support + * Tables, Views and Sequences.
+ * + * @param connection + * @param schema - + * This parameter is somewhat bogus. It is used to determine if the + * adapter defines an SQL statement for finding entities of a + * particular types. + * @return + * @throws SQLException + */ + public String[] getEntityTypes(Connection connection, Schema schema) + throws SQLException { + + Set set = new HashSet(); + if (this.databaseAdapter.getShowTableQuery(schema.getName(), false) != null) { + set.add(IQuantumConstants.Table); + } else if (this.databaseAdapter.getShowViewQuery(schema.getName(), false) != null) { + set.add(IQuantumConstants.View); + } else if (this.databaseAdapter.getShowSequenceQuery(schema.getName(), false) != null) { + set.add(IQuantumConstants.Sequence); + } + + DatabaseMetaData metaData = connection.getMetaData(); + ResultSet resultSet = metaData.getTableTypes(); + while (resultSet.next()) { + String type = resultSet.getString("TABLE_TYPE"); + if (type != null) { + type = type.trim(); + } + if (STANDARD_TABLE_TYPES.contains(type)) { + set.add(type); + } + } + + return (String[]) set.toArray(new String[set.size()]); + } + + public String getInformation() throws SQLException { + try { + Connection connection = this.bookmark.getConnection(); + DatabaseMetaData metaData = connection.getMetaData(); + + return metaData == null ? null : metaData.getDatabaseProductName() + " " + + metaData.getDatabaseProductVersion(); + } catch (NotConnectedException e) { + // TODO: think about this... + return ""; + } + } + + /** + * Get a list of entities (tables, views, sequences) for a particular + * bookmark. This function is usually not redefined because it gives + * an external interface. You will usually redefine the getShowTableQuery(), + * getShowViewQuery(), etc. + * + * @param bookmark - + * the bookmark that describes the database that is being accessed. + * @param passwordFinder - + * a utility class that knows how to obtain a password, if required + * @param schema - + * the schema from which to extract + * @param type - + * the type ("VIEW", "TABLE", etc.) of entities to extract or null + * if all entity types should be extracted + * @return + * an array of entity objects representing the tables, views and sequences. + * @throws SQLException + */ + public Entity[] getEntities(Bookmark bookmark, Schema schema, String type) + throws SQLException, NotConnectedException { + Connection connection = bookmark.getConnection(); + Entity[] result = getEntities(bookmark, connection, schema, type); + return (result == null) ? new Entity[0] : result; + } + + protected Entity[] getEntities(Bookmark bookmark, Connection connection, Schema schema, String type) + throws SQLException { + + List list = new ArrayList(); + String[] types = (type == null) ? ALL_TYPES : new String[] { type }; + + for (int i = 0; i < types.length; i++) { + list.addAll(getEntitiesList(bookmark, connection, types[i], schema)); + } + + return (Entity[]) list.toArray(new Entity[list.size()]); + } + + protected List getEntitiesList(Bookmark bookmark, Connection connection, String type, Schema schema) + throws SQLException { + + String sql = getSQL(bookmark, type, schema); + List list = new ArrayList(); + SQLResults results = null; + if (sql != null) { + results = MultiSQLServer.getInstance().execute(connection, sql); + for (int i = 1, size = (results == null) ? 0 : results.getRowCount(); i <= size; i++) { + String schemaName = results.getColumnCount() == 1 + ? schema.getName() : results.getElement(1, i).toString(); + String tableName = results.getColumnCount() == 1 + ? results.getElement(1, i).toString() + : results.getElement(2, i).toString(); + if (tableName != null && tableName.length() > 0) { + Entity entity = EntityFactory.getInstance().create( + bookmark, schemaName, tableName, type); + if (entity != null) { + list.add(entity); + } + } + } + } + // If we have some results, we go back + if (results != null) return list; + // Else, we try the JDBC driver + DatabaseMetaData metaData = connection.getMetaData(); + // getTables needs a null schema to get all the schemas. So we don't pass a "" schema, but a null one + ResultSet set = null; + if (metaData.supportsSchemasInTableDefinitions()) + set = metaData.getTables(null, (schema != null) ? schema.getName() : null, "%", new String[] { type }); + else + set = metaData.getTables(null, null, "%", new String[] { type }); + + while (set.next()) { + String tempSchema = set.getString("TABLE_SCHEM"); + tempSchema = (tempSchema == null) ? "" : tempSchema.trim(); + String tableName = set.getString("TABLE_NAME"); + tableName = (tableName == null) ? "" : tableName.trim(); + + if (tableName != null && tableName.length() > 0) { + Entity entity = EntityFactory.getInstance().create(bookmark, tempSchema, tableName, type); + if (entity != null) { + list.add(entity); + } + } + } + set.close(); + return list; + } + + + private String getSQL(Bookmark bookmark, String type, Schema schema) { + if (Entity.TABLE_TYPE.equals(type)) { + return this.databaseAdapter.getShowTableQuery(schema.getName(), schema.isDefault()); + } else if (Entity.VIEW_TYPE.equals(type)) { + return this.databaseAdapter.getShowViewQuery(schema.getName(), schema.isDefault()); + } else if (Entity.SEQUENCE_TYPE.equals(type)) { + return this.databaseAdapter.getShowSequenceQuery(schema.getName(), schema.isDefault()); + } else { + return null; + } + } + + +}