package liquibase.sqlgenerator.core;

 import liquibase.database.Database;

 import liquibase.database.core.DB2Database;

 import liquibase.database.structure.Column;

 import liquibase.database.structure.Table;

 import liquibase.sql.Sql;

 import liquibase.sql.UnparsedSql;

 import liquibase.sqlgenerator.SqlGeneratorChain;

 import liquibase.statement.core.AddAutoIncrementStatement;

 public class AddAutoIncrementGeneratorDB2 extends AddAutoIncrementGenerator {

     @Override

     public int getPriority() {

         return PRIORITY_DATABASE;

     }

     @Override

     public boolean supports(AddAutoIncrementStatement statement, Database database) {

         return database instanceof DB2Database;

     }

     @Override

     public Sql[] generateSql(AddAutoIncrementStatement statement, Database database, SqlGeneratorChain sqlGeneratorChain) {

         return new Sql[] { new UnparsedSql("ALTER TABLE "

                 + database.escapeTableName(statement.getSchemaName(), statement.getTableName())

                 + " ALTER COLUMN "

                 + database.escapeColumnName(statement.getSchemaName(), statement.getTableName(),

                         statement.getColumnName()) + " SET GENERATED ALWAYS AS IDENTITY", new Column().setTable(

                 new Table(statement.getTableName()).setSchema(statement.getSchemaName())).setName(

                 statement.getColumnName())) };

     }

 }