de.akquinet.jbosscc.guttenbase.configuration.impl.MsSqlTargetDatabaseConfiguration 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.configuration.impl;

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

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

/**
 * Implementation for MS Server SQL data base.
 * 
 * 
 * © 2012 akquinet tech@spree
 * 
 * 
 * @Uses-Hint {@link TableNameMapperHint}
 * @author M. Dahm
 */
public class MsSqlTargetDatabaseConfiguration extends DefaultTargetDatabaseConfiguration {
	public MsSqlTargetDatabaseConfiguration(final ConnectorRepository connectorRepository) {
		super(connectorRepository);
	}

	/**
	 * {@inheritDoc}
	 */
	@Override
	public void initializeTargetConnection(final Connection connection, final String connectorId) throws SQLException {
		connection.setAutoCommit(false);

		disableTableForeignKeys(connection, connectorId, getTableMetaData(connectorId));
	}

	/**
	 * {@inheritDoc}
	 */
	@Override
	public void finalizeTargetConnection(final Connection connection, final String connectorId) throws SQLException {
		enableTableForeignKeys(connection, connectorId, getTableMetaData(connectorId));
	}

	/**
	 * {@inheritDoc}
	 */
	@Override
	public void beforeInsert(final Connection connection, final String connectorId, final TableMetaData table) throws SQLException {
		setIdentityInsert(connection, connectorId, true, table);
	}

	/**
	 * {@inheritDoc}
	 */
	@Override
	public void afterInsert(final Connection connection, final String connectorId, final TableMetaData table) throws SQLException {
		setIdentityInsert(connection, connectorId, false, table);
	}

	private List getTableMetaData(final String connectorId) throws SQLException {
		return _connectorRepository.getDatabaseMetaData(connectorId).getTableMetaData();
	}

	private void disableTableForeignKeys(final Connection connection, final String connectorId, final List tableMetaData)
			throws SQLException {
		setTableForeignKeys(connection, connectorId, tableMetaData, false);
	}

	private void enableTableForeignKeys(final Connection connection, final String connectorId, final List tableMetaData)
			throws SQLException {
		setTableForeignKeys(connection, connectorId, tableMetaData, true);
	}

	private void setTableForeignKeys(final Connection connection, final String connectorId, final List tableMetaDatas,
			final boolean enable) throws SQLException {
		final TableNameMapper tableNameMapper = _connectorRepository.getConnectorHint(connectorId, TableNameMapper.class).getValue();

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

			executeSQL(connection, "ALTER TABLE " + tableName + (enable ? " CHECK CONSTRAINT ALL" : " NOCHECK CONSTRAINT ALL"));
		}
	}

	private void setIdentityInsert(final Connection connection, final String connectorId, final boolean enable,
			final TableMetaData tableMetaData) throws SQLException {
		final TableNameMapper tableNameMapper = _connectorRepository.getConnectorHint(connectorId, TableNameMapper.class).getValue();
		final String tableName = tableNameMapper.mapTableName(tableMetaData);

		if (hasIdentityColumn(tableMetaData)) {
			executeSQL(connection, "SET IDENTITY_INSERT " + tableName + " " + (enable ? "ON" : "OFF"));
		}
	}

	private boolean hasIdentityColumn(final TableMetaData tableMetaData) {
		for (final ColumnMetaData columnMetaData : tableMetaData.getColumnMetaData()) {
			if (columnMetaData.getColumnTypeName().contains("IDENTITY")) {
				return true;
			}
		}

		return false;
	}
}