コード例 #1
0
  @Override
  protected void readColumns(
      DatabaseSnapshot snapshot, String schema, DatabaseMetaData databaseMetaData)
      throws SQLException, DatabaseException {
    findIntegerColumns(snapshot, schema);
    super.readColumns(snapshot, schema, databaseMetaData);

    /*
     * Code Description:
     * Finding all 'tablespace' attributes of column's PKs
     * */
    Database database = snapshot.getDatabase();
    Statement statement = null;
    ResultSet rs = null;
    try {
      statement =
          ((JdbcConnection) database.getConnection()).getUnderlyingConnection().createStatement();

      // Setting default schema name. Needed for correct statement generation
      if (schema == null) schema = database.convertRequestedSchemaToSchema(schema);

      String query =
          "select ui.tablespace_name TABLESPACE, ucc.table_name TABLE_NAME, ucc.column_name COLUMN_NAME FROM all_indexes ui , all_constraints uc , all_cons_columns ucc where uc.constraint_type = 'P' and ucc.constraint_name = uc.constraint_name and uc.index_name = ui.index_name and uc.owner = '"
              + schema
              + "' and ui.table_owner = '"
              + schema
              + "' and ucc.owner = '"
              + schema
              + "'";
      rs = statement.executeQuery(query);

      while (rs.next()) {
        Column column = snapshot.getColumn(rs.getString("TABLE_NAME"), rs.getString("COLUMN_NAME"));
        // setting up tablespace property to column, to configure it's PK-index
        if (column == null) {
          continue; // probably a different schema
        }
        column.setTablespace(rs.getString("TABLESPACE"));
      }
    } finally {
      if (rs != null) {
        try {
          rs.close();
        } catch (SQLException ignore) {
        }
      }
      if (statement != null) {
        try {
          statement.close();
        } catch (SQLException ignore) {
        }
      }
    }
  }
コード例 #2
0
  /**
   * Method finds all INTEGER columns in snapshot's database
   *
   * @param snapshot current database snapshot
   * @return String list with names of all INTEGER columns
   * @throws java.sql.SQLException execute statement error
   */
  private List<String> findIntegerColumns(DatabaseSnapshot snapshot, String schema)
      throws SQLException, DatabaseException {

    Database database = snapshot.getDatabase();
    // Setting default schema name. Needed for correct statement generation
    if (schema == null) {
      schema = database.convertRequestedSchemaToSchema(schema);
    }
    Statement statement =
        ((JdbcConnection) database.getConnection()).getUnderlyingConnection().createStatement();
    ResultSet integerListRS = null;
    // Finding all columns created as 'INTEGER'
    try {
      integerListRS =
          statement.executeQuery(
              "select TABLE_NAME, COLUMN_NAME from all_tab_columns where data_precision is null and data_scale = 0 and data_type = 'NUMBER' and owner = '"
                  + schema
                  + "'");
      while (integerListRS.next()) {
        integerList.add(
            integerListRS.getString("TABLE_NAME") + "." + integerListRS.getString("COLUMN_NAME"));
      }
    } finally {
      if (integerListRS != null) {
        try {
          integerListRS.close();
        } catch (SQLException ignore) {
        }
      }

      if (statement != null) {
        try {
          statement.close();
        } catch (SQLException ignore) {
        }
      }
    }

    return integerList;
  }