de.akquinet.jbosscc.guttenbase.tools.AbstractTablesOperationTool 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...

There is a newer version: 2.1.0

package de.akquinet.jbosscc.guttenbase.tools;

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

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

/**
 * Will execute given SQL scriptlet on all tables or single table of given connector. The table name can be referenced with @TABLE@ place
 * holder.
 * 
 * 
 * © 2012 akquinet tech@spree
 * 
 * 
 * @Uses-Hint {@link TableNameMapperHint}
 * @author M. Dahm
 */
public abstract class AbstractTablesOperationTool {
	public static final String TABLE_PLACEHOLDER = "@TABLE@";

	protected final ConnectorRepository _connectorRepository;
	protected final ScriptExecutorTool _scriptExecutor;
	private final String _template;

	public AbstractTablesOperationTool(final ConnectorRepository connectorRepository, final String template) {
		assert connectorRepository != null : "connectorRepository != null";
		assert template != null : "template != null";
		_connectorRepository = connectorRepository;
		_template = template;
		_scriptExecutor = new ScriptExecutorTool(connectorRepository);
	}

	public void executeOnAllTables(final String connectorId, final boolean updateSchema, final boolean prepareTargetConnection)
			throws SQLException {
		final TableNameMapper tableNameMapper = _connectorRepository.getConnectorHint(connectorId, TableNameMapper.class).getValue();
		final List tables = TableOrderHint.getSortedTables(_connectorRepository, connectorId);
		final List sql = new ArrayList();

		for (final TableMetaData tableMetaData : tables) {
			final String tableName = tableNameMapper.mapTableName(tableMetaData);

			sql.add(_template.replaceAll(TABLE_PLACEHOLDER, tableName));
		}

		_scriptExecutor.executeScript(connectorId, updateSchema, prepareTargetConnection, sql);
	}

	public void executeOnTable(final String connectorId, final boolean updatesSchema, final boolean prepareTargetConnection,
			final TableMetaData tableMetaData) throws SQLException {
		final TableNameMapper tableNameMapper = _connectorRepository.getConnectorHint(connectorId, TableNameMapper.class).getValue();
		final String tableName = tableNameMapper.mapTableName(tableMetaData);

		_scriptExecutor.executeScript(connectorId, updatesSchema, prepareTargetConnection, _template.replaceAll(TABLE_PLACEHOLDER, tableName));
	}
}