de.akquinet.jbosscc.guttenbase.tools.AbstractSequenceCreationTool Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of GuttenBase Show documentation
Copying done right...
The newest version!
package de.akquinet.jbosscc.guttenbase.tools;

import de.akquinet.jbosscc.guttenbase.hints.TableOrderHint;
import de.akquinet.jbosscc.guttenbase.mapping.TableMapper;
import de.akquinet.jbosscc.guttenbase.meta.TableMetaData;
import de.akquinet.jbosscc.guttenbase.repository.ConnectorRepository;

import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;

/**
 * Create auto-increment sequences for table IDs.
 * 
 * 
 * © 2012-2034 akquinet tech@spree
 * 
 *
 * @author M. Dahm
 * Uses {@link de.akquinet.jbosscc.guttenbase.hints.TableMapperHint}
 * Uses {@link de.akquinet.jbosscc.guttenbase.hints.EntityTableCheckerHint} to look for entity classes, i.e. classes that may use
 * an ID sequence
 */
@SuppressWarnings("RedundantThrows")
public abstract class AbstractSequenceCreationTool {
  protected final ConnectorRepository _connectorRepository;
  protected final ScriptExecutorTool _scriptExecutor;

  public AbstractSequenceCreationTool(final ConnectorRepository connectorRepository) {
    assert connectorRepository != null : "connectorRepository != null";
    _connectorRepository = connectorRepository;
    _scriptExecutor = new ScriptExecutorTool(connectorRepository);
  }

  public void createSequences(final String connectorId, long start, long incrementBy) throws SQLException {
    final List tableMetaDatas = TableOrderHint.getSortedTables(_connectorRepository, connectorId);
    final EntityTableChecker entityTableChecker = _connectorRepository.getConnectorHint(connectorId, EntityTableChecker.class).getValue();
    final TableMapper tableMapper = _connectorRepository.getConnectorHint(connectorId, TableMapper.class).getValue();
    final List updateClauses = new ArrayList<>();

    for (final TableMetaData tableMetaData : tableMetaDatas) {
      if (entityTableChecker.isEntityTable(tableMetaData)) {
        final String tableName = tableMapper.mapTableName(tableMetaData, tableMetaData.getDatabaseMetaData());
        final String sequenceName = getSequenceName(tableName);

        updateClauses.addAll(getCreateSequenceClauses(tableName, getIdColumn(tableMetaData), sequenceName, start, incrementBy));
      }
    }

    _scriptExecutor.executeScript(connectorId, false, false, updateClauses);
  }

  protected abstract String getIdColumn(final TableMetaData tableMetaData);

  protected abstract List getCreateSequenceClauses(final String tableName, String idColumn, final String sequenceName, final long start, final long incrementBy);

  protected abstract String getSequenceName(final String tableName);
}