コード例 #1
0
  protected List<ObjectValuePair<String, IndexMetadata>> getIndexesSQL(
      ClassLoader classLoader, String tableName) throws IOException {

    if (!PortalClassLoaderUtil.isPortalClassLoader(classLoader)) {
      List<ObjectValuePair<String, IndexMetadata>> objectValuePairs = new ArrayList<>();

      try (InputStream is = classLoader.getResourceAsStream("META-INF/sql/indexes.sql");
          Reader reader = new InputStreamReader(is);
          UnsyncBufferedReader unsyncBufferedReader = new UnsyncBufferedReader(reader)) {

        String line = null;

        while ((line = unsyncBufferedReader.readLine()) != null) {
          line = line.trim();

          if (line.isEmpty()) {
            continue;
          }

          IndexMetadata indexMetadata = IndexMetadataFactoryUtil.createIndexMetadata(line);

          if (tableName.equals(indexMetadata.getTableName())) {
            objectValuePairs.add(new ObjectValuePair<>(line, indexMetadata));
          }
        }
      }

      return objectValuePairs;
    }

    if (!_portalIndexesSQL.isEmpty()) {
      return _portalIndexesSQL.get(tableName);
    }

    try (InputStream is =
            classLoader.getResourceAsStream(
                "com/liferay/portal/tools/sql/dependencies/indexes.sql");
        Reader reader = new InputStreamReader(is);
        UnsyncBufferedReader unsyncBufferedReader = new UnsyncBufferedReader(reader)) {

      String line = null;

      while ((line = unsyncBufferedReader.readLine()) != null) {
        line = line.trim();

        if (line.isEmpty()) {
          continue;
        }

        IndexMetadata indexMetadata = IndexMetadataFactoryUtil.createIndexMetadata(line);

        List<ObjectValuePair<String, IndexMetadata>> objectValuePairs =
            _portalIndexesSQL.get(indexMetadata.getTableName());

        if (objectValuePairs == null) {
          objectValuePairs = new ArrayList<>();

          _portalIndexesSQL.put(indexMetadata.getTableName(), objectValuePairs);
        }

        objectValuePairs.add(new ObjectValuePair<>(line, indexMetadata));
      }
    }

    return _portalIndexesSQL.get(tableName);
  }
コード例 #2
0
  protected void alter(Class<?> tableClass, Alterable... alterables) throws Exception {

    try (LoggingTimer loggingTimer = new LoggingTimer()) {
      Field tableNameField = tableClass.getField("TABLE_NAME");

      String tableName = (String) tableNameField.get(null);

      DatabaseMetaData databaseMetaData = connection.getMetaData();

      try (ResultSet rs1 = databaseMetaData.getPrimaryKeys(null, null, tableName);
          ResultSet rs2 =
              databaseMetaData.getIndexInfo(
                  null, null, normalizeName(tableName, databaseMetaData), false, false)) {

        Set<String> primaryKeyNames = new HashSet<>();

        while (rs1.next()) {
          String primaryKeyName = StringUtil.toUpperCase(rs1.getString("PK_NAME"));

          if (primaryKeyName != null) {
            primaryKeyNames.add(primaryKeyName);
          }
        }

        Map<String, Set<String>> columnNamesMap = new HashMap<>();

        while (rs2.next()) {
          String indexName = StringUtil.toUpperCase(rs2.getString("INDEX_NAME"));

          if ((indexName == null) || primaryKeyNames.contains(indexName)) {

            continue;
          }

          Set<String> columnNames = columnNamesMap.get(indexName);

          if (columnNames == null) {
            columnNames = new HashSet<>();

            columnNamesMap.put(indexName, columnNames);
          }

          columnNames.add(StringUtil.toUpperCase(rs2.getString("COLUMN_NAME")));
        }

        for (Alterable alterable : alterables) {
          for (Map.Entry<String, Set<String>> entry : columnNamesMap.entrySet()) {

            if (alterable.shouldDropIndex(entry.getValue())) {
              runSQL("drop index " + entry.getKey() + " on " + tableName);
            }
          }

          runSQL(alterable.getSQL(tableName));

          List<ObjectValuePair<String, IndexMetadata>> objectValuePairs =
              getIndexesSQL(tableClass.getClassLoader(), tableName);

          if (objectValuePairs == null) {
            continue;
          }

          for (ObjectValuePair<String, IndexMetadata> objectValuePair : objectValuePairs) {

            IndexMetadata indexMetadata = objectValuePair.getValue();

            if (alterable.shouldAddIndex(Arrays.asList(indexMetadata.getColumnNames()))) {

              runSQLTemplateString(objectValuePair.getKey(), false, true);
            }
          }
        }
      } catch (SQLException sqle) {
        if (_log.isWarnEnabled()) {
          _log.warn("Fallback to recreating the table", sqle);
        }

        Field tableColumnsField = tableClass.getField("TABLE_COLUMNS");
        Field tableSQLCreateField = tableClass.getField("TABLE_SQL_CREATE");
        Field tableSQLAddIndexesField = tableClass.getField("TABLE_SQL_ADD_INDEXES");

        upgradeTable(
            tableName,
            (Object[][]) tableColumnsField.get(null),
            (String) tableSQLCreateField.get(null),
            (String[]) tableSQLAddIndexesField.get(null));
      }
    }
  }