de.opitzconsulting.orcas.diff.InlineIndexExtension Maven / Gradle / Ivy

Go to download
package de.opitzconsulting.orcas.diff;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.Map;

import de.opitzconsulting.origOrcasDsl.Index;
import de.opitzconsulting.origOrcasDsl.IndexExTable;
import de.opitzconsulting.origOrcasDsl.Model;
import de.opitzconsulting.origOrcasDsl.ModelElement;
import de.opitzconsulting.origOrcasDsl.Table;
import de.opitzconsulting.origOrcasDsl.impl.IndexImpl;

public class InlineIndexExtension
{
  public Model transformModel( Model pModel )
  {
    Map lCacheTableMap = new HashMap<>();

    for( ModelElement lModelElement : new ArrayList( pModel.getModel_elements() ) )
    {
      if( lModelElement instanceof IndexExTable )
      {
        IndexExTable lIndexExTable = (IndexExTable) lModelElement;

        Index lIndex = new IndexImpl();

        lIndex.setBitmap( lIndexExTable.getBitmap() );
        lIndex.setCompression( lIndexExTable.getCompression() );
        lIndex.setConsName( lIndexExTable.getIndex_name() );
        lIndex.setDomain_index_expression( lIndexExTable.getDomain_index_expression() );
        lIndex.setFunction_based_expression( lIndexExTable.getFunction_based_expression() );
        lIndex.setGlobal( lIndexExTable.getGlobal() );
        lIndex.setLogging( lIndexExTable.getLogging() );
        lIndex.setParallel( lIndexExTable.getParallel() );
        lIndex.setParallel_degree( lIndexExTable.getParallel_degree() );
        lIndex.setTablespace( lIndexExTable.getTablespace() );
        lIndex.setUnique( lIndexExTable.getUniqueness() );
        lIndex.getIndex_columns().addAll( ModelUtil.copyColumnRefs( lIndexExTable.getIndex_columns() ) );

        findTable( lCacheTableMap, pModel, lIndexExTable.getTable_name() ).getInd_uks().add( lIndex );

        pModel.getModel_elements().remove( lIndexExTable );
      }
    }

    return pModel;
  }

  private Table findTable( Map pCacheTableMap, Model pModel, String pTablename )
  {
    if( pCacheTableMap.containsKey( pTablename ) )
    {
      return pCacheTableMap.get( pTablename );
    }

    for( ModelElement lModelElement : pModel.getModel_elements() )
    {
      if( lModelElement instanceof Table )
      {
        if( ((Table) lModelElement).getName().equalsIgnoreCase( pTablename ) )
        {
          pCacheTableMap.put( pTablename, (Table) lModelElement );

          return (Table) lModelElement;
        }
      }
    }

    throw new IllegalArgumentException( "Table not found: " + pTablename );
  }
}