package liquibase.change.core;

 import liquibase.change.AbstractChange;

 import liquibase.change.ChangeMetaData;

 import liquibase.change.ChangeWithColumns;

 import liquibase.change.ColumnConfig;

 import liquibase.change.TextNode;

 import liquibase.database.Database;

 import liquibase.statement.SqlStatement;

 import liquibase.statement.core.UpdateStatement;

 import liquibase.util.StringUtils;

 import java.util.ArrayList;

 import java.util.List;

 public class UpdateDataChange extends AbstractChange implements ChangeWithColumns {

     private String schemaName;

     private String tableName;

     private List<ColumnConfig> columns;

     @TextNode(nodeName = "where")

     private String whereClause;

     public UpdateDataChange() {

         super("update", "Update Data", ChangeMetaData.PRIORITY_DEFAULT);

         columns = new ArrayList<ColumnConfig>();

     }

     public String getSchemaName() {

         return schemaName;

     }

     public void setSchemaName(String schemaName) {

         this.schemaName = StringUtils.trimToNull(schemaName);

     }

     public String getTableName() {

         return tableName;

     }

     public void setTableName(String tableName) {

         this.tableName = tableName;

     }

     public List<ColumnConfig> getColumns() {

         return columns;

     }

     public void setColumns(List<ColumnConfig> columns) {

         this.columns = columns;

     }

     public void addColumn(ColumnConfig column) {

         columns.add(column);

     }

     public void removeColumn(ColumnConfig column) {

         columns.remove(column);

     }

     public String getWhereClause() {

         return whereClause;

     }

     public void setWhereClause(String whereClause) {

         this.whereClause = whereClause;

     }

     public SqlStatement[] generateStatements(Database database) {

         UpdateStatement statement = new UpdateStatement(getSchemaName() == null ? database.getDefaultSchemaName()

                 : getSchemaName(), getTableName());

         for (ColumnConfig column : columns) {

             statement.addNewColumnValue(column.getName(), column.getValueObject());

         }

         statement.setWhereClause(whereClause);

         return new SqlStatement[] { statement };

     }

     public String getConfirmationMessage() {

         return "Data updated in " + getTableName();

     }

 }