liquibase.ext.mssql.change.AddPrimaryKeyChangeMSSQL Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of liquibase-mssql Show documentation

Liquibase extensions for MS SQL Server.

There is a newer version: 1.6.4

package liquibase.ext.mssql.change;

import liquibase.change.ChangeMetaData;
import liquibase.change.DatabaseChange;
import liquibase.change.core.AddPrimaryKeyChange;
import liquibase.database.Database;
import liquibase.ext.mssql.database.MSSQLDatabase;
import liquibase.ext.mssql.statement.AddPrimaryKeyStatementMSSQL;
import liquibase.statement.SqlStatement;
import liquibase.statement.core.AddPrimaryKeyStatement;

import java.util.ArrayList;
import java.util.List;

@DatabaseChange(name="addPrimaryKey", description = "Adds creates a primary key out of an existing column or set of columns.", priority = ChangeMetaData.PRIORITY_DATABASE, appliesTo = "column")
public class AddPrimaryKeyChangeMSSQL extends AddPrimaryKeyChange {

  private Integer fillFactor;

  public Integer getFillFactor() {
    return fillFactor;
  }

  public void setFillFactor(Integer fillFactor) {
    this.fillFactor = fillFactor;
  }

  @Override
  public SqlStatement[] generateStatements(Database database) {
    SqlStatement[] statements = super.generateStatements(database);
    if (!MSSQLDatabase.PRODUCT_NAME.equals(database.getDatabaseProductName())) {
    	return statements;
    }
    List extendedStatements = new ArrayList(statements.length);

    for (SqlStatement statement : statements) {
      if (statement instanceof AddPrimaryKeyStatement) {
        extendedStatements.add(new AddPrimaryKeyStatementMSSQL((AddPrimaryKeyStatement)statement, fillFactor));
      } else {
        extendedStatements.add(statement);
      }
    }

    return extendedStatements.toArray(new SqlStatement[0]);
  }
}