public String[] getCleanSql() {
    if (cleanSql == null) {
      // loop over all foreign key constraints
      List dropForeignKeysSql = new ArrayList();
      List createForeignKeysSql = new ArrayList();
      Iterator iter = configuration.getTableMappings();
      while (iter.hasNext()) {
        Table table = (Table) iter.next();
        if (table.isPhysicalTable()) {
          Iterator subIter = table.getForeignKeyIterator();
          while (subIter.hasNext()) {
            ForeignKey fk = (ForeignKey) subIter.next();
            if (fk.isPhysicalConstraint()) {
              // collect the drop key constraint
              dropForeignKeysSql.add(
                  fk.sqlDropString(
                      dialect,
                      properties.getProperty(Environment.DEFAULT_CATALOG),
                      properties.getProperty(Environment.DEFAULT_SCHEMA)));
              createForeignKeysSql.add(
                  fk.sqlCreateString(
                      dialect,
                      mapping,
                      properties.getProperty(Environment.DEFAULT_CATALOG),
                      properties.getProperty(Environment.DEFAULT_SCHEMA)));
            }
          }
        }
      }

      List deleteSql = new ArrayList();
      iter = configuration.getTableMappings();
      while (iter.hasNext()) {
        Table table = (Table) iter.next();
        deleteSql.add("delete from " + table.getName());
      }

      List cleanSqlList = new ArrayList();
      cleanSqlList.addAll(dropForeignKeysSql);
      cleanSqlList.addAll(deleteSql);
      cleanSqlList.addAll(createForeignKeysSql);

      cleanSql = (String[]) cleanSqlList.toArray(new String[cleanSqlList.size()]);
    }
    return cleanSql;
  }
예제 #2
0
  /**
   * Returns a table's foreign keys and their columns as a Map from the key name to the ForeignKey
   * object.
   *
   * <p>A foreign key may not have a name. On such a database, 2 foreign keys must reference 2
   * different tables. Otherwise there's no way to tell them apart and the foreign key information
   * reported by DatabaseMetaData becomes ill-formed.
   */
  public static Map<String, ForeignKey> getForeignKeys(DatabaseMetaData metadata, String tableName)
      throws Exception {
    ResultSet keys =
        metadata.getImportedKeys(
            metadata.getConnection().getCatalog(), metadata.getUserName(), tableName);
    Map<String, ForeignKey> map = new HashMap<String, ForeignKey>();

    while (keys.next()) {
      String table = keys.getString(IMPORTED_PK_TAB_NAME);
      String name = keys.getString(IMPORTED_FK_KEY_NAME);
      if (name == null || name.length() == 0) name = "UNNAMED_FK_" + table;

      ForeignKey key = map.get(name);
      if (key == null) {
        map.put(name, key = new ForeignKey(table));
      }
      key.add(keys.getString(IMPORTED_FK_COL_NAME));
    }
    keys.close();
    return map;
  }
  @Override
  protected void readIndexes(
      DatabaseSnapshot snapshot, String schema, DatabaseMetaData databaseMetaData)
      throws DatabaseException, SQLException {
    Database database = snapshot.getDatabase();
    updateListeners("Reading indexes for " + database.toString() + " ...");

    String query =
        "select aic.index_name, 3 AS TYPE, aic.table_name, aic.column_name, aic.column_position AS ORDINAL_POSITION, null AS FILTER_CONDITION, ai.tablespace_name AS TABLESPACE, ai.uniqueness FROM all_ind_columns aic, all_indexes ai WHERE aic.table_owner='"
            + database.convertRequestedSchemaToSchema(schema)
            + "' and aic.index_name = ai.index_name ORDER BY INDEX_NAME, ORDINAL_POSITION";
    Statement statement = null;
    ResultSet rs = null;
    Map<String, Index> indexMap = null;
    try {
      statement =
          ((JdbcConnection) database.getConnection()).getUnderlyingConnection().createStatement();
      rs = statement.executeQuery(query);

      indexMap = new HashMap<String, Index>();
      while (rs.next()) {
        String indexName = convertFromDatabaseName(rs.getString("INDEX_NAME"));
        String tableName = rs.getString("TABLE_NAME");
        String tableSpace = rs.getString("TABLESPACE");
        String columnName = convertFromDatabaseName(rs.getString("COLUMN_NAME"));
        if (columnName == null) {
          // nothing to index, not sure why these come through sometimes
          continue;
        }
        short type = rs.getShort("TYPE");

        boolean nonUnique;

        String uniqueness = rs.getString("UNIQUENESS");

        if ("UNIQUE".equals(uniqueness)) {
          nonUnique = false;
        } else {
          nonUnique = true;
        }

        short position = rs.getShort("ORDINAL_POSITION");
        String filterCondition = rs.getString("FILTER_CONDITION");

        if (type == DatabaseMetaData.tableIndexStatistic) {
          continue;
        }

        Index index;
        if (indexMap.containsKey(indexName)) {
          index = indexMap.get(indexName);
        } else {
          index = new Index();
          Table table = snapshot.getTable(tableName);
          if (table == null) {
            continue; // probably different schema
          }
          index.setTable(table);
          index.setTablespace(tableSpace);
          index.setName(indexName);
          index.setUnique(!nonUnique);
          index.setFilterCondition(filterCondition);
          indexMap.put(indexName, index);
        }

        for (int i = index.getColumns().size(); i < position; i++) {
          index.getColumns().add(null);
        }
        index.getColumns().set(position - 1, columnName);
      }
    } finally {
      JdbcUtils.closeResultSet(rs);
      JdbcUtils.closeStatement(statement);
    }

    for (Map.Entry<String, Index> entry : indexMap.entrySet()) {
      snapshot.getIndexes().add(entry.getValue());
    }

    /*
     * marks indexes as "associated with" instead of "remove it"
     * Index should have associations with:
     * foreignKey, primaryKey or uniqueConstraint
     * */
    for (Index index : snapshot.getIndexes()) {
      for (PrimaryKey pk : snapshot.getPrimaryKeys()) {
        if (index.getTable().getName().equalsIgnoreCase(pk.getTable().getName())
            && index.getColumnNames().equals(pk.getColumnNames())) {
          index.addAssociatedWith(Index.MARK_PRIMARY_KEY);
        }
      }
      for (ForeignKey fk : snapshot.getForeignKeys()) {
        if (index.getTable().getName().equalsIgnoreCase(fk.getForeignKeyTable().getName())
            && index.getColumnNames().equals(fk.getForeignKeyColumns())) {
          index.addAssociatedWith(Index.MARK_FOREIGN_KEY);
        }
      }
      for (UniqueConstraint uc : snapshot.getUniqueConstraints()) {
        if (index.getTable().getName().equalsIgnoreCase(uc.getTable().getName())
            && index.getColumnNames().equals(uc.getColumnNames())) {
          index.addAssociatedWith(Index.MARK_UNIQUE_CONSTRAINT);
        }
      }
    }
  }