package liquibase.snapshot.jvm;

 import liquibase.database.Database;

 import liquibase.database.core.DB2Database;

 import liquibase.exception.DatabaseException;

 import liquibase.executor.ExecutorService;

 import liquibase.statement.core.RawSqlStatement;

 import java.util.List;

 import java.util.Map;

 public class DB2DatabaseSnapshotGenerator extends JdbcDatabaseSnapshotGenerator {

     public boolean supports(Database database) {

         return database instanceof DB2Database;

     }

     public int getPriority(Database database) {

         return PRIORITY_DATABASE;

     }

     @Override

     protected String convertTableNameToDatabaseTableName(String tableName) {

         return tableName.toUpperCase();

     }

     @Override

     protected String convertColumnNameToDatabaseTableName(String columnName) {

         return columnName.toUpperCase();

     }

     @Override

     public boolean isColumnAutoIncrement(Database database, String schemaName, String tableName, String columnName)

             throws DatabaseException {

         boolean autoIncrement = false;

         List<Map> rs = ExecutorService

                 .getInstance()

                 .getExecutor(database)

                 .queryForList(

                         new RawSqlStatement("SELECT IDENTITY FROM SYSCAT.COLUMNS WHERE TABSCHEMA = '"

                                 + database.convertRequestedSchemaToSchema(schemaName) + "' AND TABNAME = '" + tableName

                                 + "' AND COLNAME = '" + columnName + "' AND HIDDEN != 'S'"));

         for (Map row : rs) {

             String identity = row.get("IDENTITY").toString();

             if (identity.equalsIgnoreCase("Y")) {

                 autoIncrement = true;

             }

         }

         return autoIncrement;

     }

 }