Add liqubase UpdateGenerator for CH

src/main/java/liquibase/ext/clickhouse/sqlgenerator/UpdateGeneratorClickHouse.java

@@ -0,0 +1,123 @@
+/*-
+ * #%L
+ * Liquibase extension for ClickHouse
+ * %%
+ * Copyright (C) 2020 - 2023 Mediarithmics
+ * %%
+ * Licensed 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.
+ * #L%
+ */
+package liquibase.ext.clickhouse.sqlgenerator;
+
+import static liquibase.util.SqlUtil.replacePredicatePlaceholders;
+
+import java.util.Date;
+
+import liquibase.ext.clickhouse.database.ClickHouseDatabase;
+import liquibase.ext.clickhouse.params.ClusterConfig;
+import liquibase.ext.clickhouse.params.ParamsLoader;
+
+import liquibase.database.Database;
+import liquibase.datatype.DataTypeFactory;
+import liquibase.sql.Sql;
+import liquibase.sqlgenerator.SqlGeneratorChain;
+import liquibase.sqlgenerator.core.UpdateGenerator;
+import liquibase.statement.DatabaseFunction;
+import liquibase.statement.core.UpdateStatement;
+
+public class UpdateGeneratorClickHouse extends UpdateGenerator {
+    @Override
+    public int getPriority() {
+        return PRIORITY_DATABASE;
+    }
+
+    @Override
+    public boolean supports(UpdateStatement statement, Database database) {
+        return database instanceof ClickHouseDatabase;
+    }
+
+    @Override
+    public Sql[] generateSql(
+            UpdateStatement statement, Database database, SqlGeneratorChain sqlGeneratorChain) {
+        ClusterConfig properties = ParamsLoader.getLiquibaseClickhouseProperties();
+
+        StringBuilder sb =
+                new StringBuilder(
+                        String.format(
+                                "ALTER TABLE %s.%s " + SqlGeneratorUtil.generateSqlOnClusterClause(properties),
+                                statement.getCatalogName(),
+                                statement.getTableName()
+                        )
+                );
+
+        sb.append(" UPDATE ");
+        for (String column : statement.getNewColumnValues().keySet()) {
+            sb.append(" ")
+                    .append(column)
+                    .append(" = ")
+                    .append(convertToString(statement.getNewColumnValues().get(column), database))
+                    .append(",");
+        }
+        int lastComma = sb.lastIndexOf(",");
+        if (lastComma >= 0) {
+            sb.deleteCharAt(lastComma);
+        }
+        if (statement.getWhereClause() != null) {
+            sb.append(" WHERE ")
+                    .append(
+                            replacePredicatePlaceholders(
+                                    database,
+                                    statement.getWhereClause(),
+                                    statement.getWhereColumnNames(),
+                                    statement.getWhereParameters()
+                            )
+                    );
+        }
+
+        String updateQuery = sb.toString();
+
+        return SqlGeneratorUtil.generateSql(database, updateQuery);
+    }
+
+    private String convertToString(Object newValue, Database database) {
+        String sqlString;
+        if ((newValue == null) || "NULL".equalsIgnoreCase(newValue.toString())) {
+            sqlString = "NULL";
+        } else if ((newValue instanceof String)
+                && !looksLikeFunctionCall(((String) newValue), database)) {
+            sqlString =
+                    DataTypeFactory.getInstance()
+                            .fromObject(newValue, database)
+                            .objectToSql(newValue, database);
+        } else if (newValue instanceof Date) {
+            // converting java.util.Date to java.sql.Date
+            Date date = (Date) newValue;
+            if (date.getClass().equals(java.util.Date.class)) {
+                date = new java.sql.Date(date.getTime());
+            }
+
+            sqlString = database.getDateLiteral(date);
+        } else if (newValue instanceof Boolean) {
+            if (((Boolean) newValue)) {
+                sqlString = DataTypeFactory.getInstance().getTrueBooleanValue(database);
+            } else {
+                sqlString = DataTypeFactory.getInstance().getFalseBooleanValue(database);
+            }
+        } else if (newValue instanceof DatabaseFunction) {
+            sqlString = database.generateDatabaseFunctionValue((DatabaseFunction) newValue);
+        } else {
+            sqlString = newValue.toString();
+        }
+        return sqlString;
+    }
+}

src/main/resources/META-INF/services/liquibase.sqlgenerator.SqlGenerator

@@ -6,4 +6,5 @@ liquibase.ext.clickhouse.sqlgenerator.UnlockDatabaseChangelogClickHouse
 liquibase.ext.clickhouse.sqlgenerator.ModifyDataTypeClickHouse
 liquibase.ext.clickhouse.sqlgenerator.RemoveChangeSetRanStatusClickHouse
 liquibase.ext.clickhouse.sqlgenerator.UpdateChangeSetChecksumClickHouse
-liquibase.ext.clickhouse.sqlgenerator.TagDatabaseGeneratorClickhouse
+liquibase.ext.clickhouse.sqlgenerator.TagDatabaseGeneratorClickhouse
+liquibase.ext.clickhouse.sqlgenerator.UpdateGeneratorClickHouse