package org.apache.ddlutils.alteration; | |
/* | |
* Licensed to the Apache Software Foundation (ASF) under one | |
* or more contributor license agreements. See the NOTICE file | |
* distributed with this work for additional information | |
* regarding copyright ownership. The ASF licenses this file | |
* to you under the Apache License, Version 2.0 (the | |
* "License"); you may not use this file except in compliance | |
* with the License. You may obtain a copy of the License at | |
* | |
* http://www.apache.org/licenses/LICENSE-2.0 | |
* | |
* Unless required by applicable law or agreed to in writing, | |
* software distributed under the License is distributed on an | |
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY | |
* KIND, either express or implied. See the License for the | |
* specific language governing permissions and limitations | |
* under the License. | |
*/ | |
import org.apache.ddlutils.model.Column; | |
import org.apache.ddlutils.model.Database; | |
import org.apache.ddlutils.model.Table; | |
/** | |
* Represents the change of the primary key of a table. | |
* | |
* @version $Revision: $ | |
*/ | |
public class PrimaryKeyChange extends TableChangeImplBase | |
{ | |
/** The names of the columns making up the new primary key. */ | |
private String[] _newPrimaryKeyColumns; | |
/** | |
* Creates a new change object. | |
* | |
* @param tableName The name of the table whose primary key is to be changed | |
* @param newPrimaryKeyColumns The names of the columns making up the new primary key | |
*/ | |
public PrimaryKeyChange(String tableName, String[] newPrimaryKeyColumns) | |
{ | |
super(tableName); | |
_newPrimaryKeyColumns = newPrimaryKeyColumns; | |
} | |
/** | |
* Returns the names of the columns making up the new primary key. | |
* | |
* @return The column names | |
*/ | |
public String[] getNewPrimaryKeyColumns() | |
{ | |
return _newPrimaryKeyColumns; | |
} | |
/** | |
* {@inheritDoc} | |
*/ | |
public void apply(Database model, boolean caseSensitive) | |
{ | |
Table table = findChangedTable(model, caseSensitive); | |
Column[] pkCols = table.getPrimaryKeyColumns(); | |
for (int idx = 0; idx < pkCols.length; idx++) | |
{ | |
pkCols[idx].setPrimaryKey(false); | |
} | |
for (int idx = 0; idx < _newPrimaryKeyColumns.length; idx++) | |
{ | |
Column column = table.findColumn(_newPrimaryKeyColumns[idx], caseSensitive); | |
column.setPrimaryKey(true); | |
} | |
} | |
} |