|
|
@@ -0,0 +1,207 @@
|
|
|
+/*
|
|
|
+ * Copyright (c) 2019 Of Him Code Technology Studio
|
|
|
+ * Jpom is licensed under Mulan PSL v2.
|
|
|
+ * You can use this software according to the terms and conditions of the Mulan PSL v2.
|
|
|
+ * You may obtain a copy of Mulan PSL v2 at:
|
|
|
+ * http://license.coscl.org.cn/MulanPSL2
|
|
|
+ * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
|
|
|
+ * See the Mulan PSL v2 for more details.
|
|
|
+ */
|
|
|
+package org.dromara.jpom.storage;
|
|
|
+
|
|
|
+import cn.hutool.core.collection.CollUtil;
|
|
|
+import cn.hutool.core.util.ObjectUtil;
|
|
|
+import cn.hutool.core.util.StrUtil;
|
|
|
+import org.dromara.jpom.common.i18n.I18nMessageUtil;
|
|
|
+import org.dromara.jpom.db.*;
|
|
|
+import org.springframework.util.Assert;
|
|
|
+
|
|
|
+import java.util.List;
|
|
|
+import java.util.stream.Collectors;
|
|
|
+
|
|
|
+/**
|
|
|
+ * @author bwcx_jzy
|
|
|
+ * @since 2025/5/8
|
|
|
+ */
|
|
|
+public class DamengTableBuilderImpl implements IStorageSqlBuilderService {
|
|
|
+
|
|
|
+ @Override
|
|
|
+ public DbExtConfig.Mode mode() {
|
|
|
+ return DbExtConfig.Mode.DAMENG;
|
|
|
+ }
|
|
|
+
|
|
|
+ @Override
|
|
|
+ public String generateIndexSql(List<TableViewIndexData> row) {
|
|
|
+ StringBuilder stringBuilder = new StringBuilder();
|
|
|
+ for (TableViewIndexData viewIndexData : row) {
|
|
|
+ String indexType = viewIndexData.getIndexType();
|
|
|
+ switch (indexType) {
|
|
|
+ case "ADD-UNIQUE": {
|
|
|
+ // ALTER TABLE `jpom`.`PROJECT_INFO`
|
|
|
+ //DROP INDEX `workspaceId`,
|
|
|
+ //ADD UNIQUE INDEX `workspaceId`(`workspaceId` ASC, `strike` ASC, `modifyUser`) USING BTREE;
|
|
|
+ String field = viewIndexData.getField();
|
|
|
+ List<String> fields = StrUtil.splitTrim(field, "+");
|
|
|
+ Assert.notEmpty(fields, I18nMessageUtil.get("i18n.index_field_not_configured.96d9"));
|
|
|
+ stringBuilder.append("call drop_index_if_exists('").append(viewIndexData.getTableName()).append("','").append(viewIndexData.getName()).append("')").append(";").append(StrUtil.LF);
|
|
|
+ stringBuilder.append(this.delimiter()).append(StrUtil.LF);
|
|
|
+ stringBuilder.append("ALTER TABLE ").append(viewIndexData.getTableName()).append(" ADD UNIQUE INDEX ").append(viewIndexData.getName()).append(" (").append(CollUtil.join(fields, StrUtil.COMMA)).append(")");
|
|
|
+ break;
|
|
|
+ }
|
|
|
+ case "ADD": {
|
|
|
+ // ALTER TABLE `jpom`.`PROJECT_INFO`
|
|
|
+ //DROP INDEX `workspaceId`,
|
|
|
+ //ADD UNIQUE INDEX `workspaceId`(`workspaceId` ASC, `strike` ASC, `modifyUser`) USING BTREE;
|
|
|
+ String field = viewIndexData.getField();
|
|
|
+ List<String> fields = StrUtil.splitTrim(field, "+");
|
|
|
+ Assert.notEmpty(fields, I18nMessageUtil.get("i18n.index_field_not_configured.96d9"));
|
|
|
+ stringBuilder.append("call drop_index_if_exists('").append(viewIndexData.getTableName()).append("','").append(viewIndexData.getName()).append("')").append(";").append(StrUtil.LF);
|
|
|
+ stringBuilder.append(this.delimiter()).append(StrUtil.LF);
|
|
|
+ stringBuilder.append("ALTER TABLE ").append(viewIndexData.getTableName()).append(" ADD INDEX ").append(viewIndexData.getName()).append(" (").append(CollUtil.join(fields, StrUtil.COMMA)).append(")");
|
|
|
+ break;
|
|
|
+ }
|
|
|
+ default:
|
|
|
+ throw new IllegalArgumentException(I18nMessageUtil.get("i18n.unsupported_type_with_colon2.7de2") + indexType);
|
|
|
+ }
|
|
|
+ stringBuilder.append(";").append(StrUtil.LF);
|
|
|
+ stringBuilder.append(this.delimiter()).append(StrUtil.LF);
|
|
|
+
|
|
|
+ }
|
|
|
+ return stringBuilder.toString();
|
|
|
+ }
|
|
|
+
|
|
|
+ @Override
|
|
|
+ public String generateAlterTableSql(List<TableViewAlterData> row) {
|
|
|
+ StringBuilder stringBuilder = new StringBuilder();
|
|
|
+ for (TableViewAlterData viewAlterData : row) {
|
|
|
+ String alterType = viewAlterData.getAlterType();
|
|
|
+ switch (alterType) {
|
|
|
+ case "DROP":
|
|
|
+ // ALTER TABLE NODE_INFO DROP COLUMN IF EXISTS `cycle`;
|
|
|
+ stringBuilder.append("CALL drop_column_if_exists('").append(viewAlterData.getTableName()).append("', '").append(viewAlterData.getName()).append("')");
|
|
|
+ break;
|
|
|
+ case "ADD":
|
|
|
+ // ALTER TABLE PROJECT_INFO ADD IF NOT EXISTS triggerToken VARCHAR (100) comment '触发器token';
|
|
|
+ String columnSql = this.generateColumnSql(viewAlterData, true);
|
|
|
+ stringBuilder.append("CALL add_column_if_not_exists('").append(viewAlterData.getTableName()).append("','").append(viewAlterData.getName()).append("','").append(columnSql).append("')");
|
|
|
+ break;
|
|
|
+ case "ALTER":
|
|
|
+ // alter table table1 modify column column1 decimal(10,1) DEFAULT NULL COMMENT '注释';
|
|
|
+ stringBuilder.append("ALTER TABLE ").append(viewAlterData.getTableName()).append(" modify column ");
|
|
|
+ stringBuilder.append(this.generateColumnSql(viewAlterData));
|
|
|
+ break;
|
|
|
+ case "DROP-TABLE":
|
|
|
+ stringBuilder.append("drop table if exists ").append(viewAlterData.getTableName());
|
|
|
+ break;
|
|
|
+ default:
|
|
|
+ throw new IllegalArgumentException(I18nMessageUtil.get("i18n.unsupported_type_with_colon2.7de2") + alterType);
|
|
|
+ }
|
|
|
+ stringBuilder.append(";").append(StrUtil.LF);
|
|
|
+ stringBuilder.append(this.delimiter()).append(StrUtil.LF);
|
|
|
+
|
|
|
+ }
|
|
|
+ return stringBuilder.toString();
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * CREATE TABLE IF NOT EXISTS "SCRIPT_INFO"
|
|
|
+ * (
|
|
|
+ * "id" VARCHAR(50) NOT NULL,
|
|
|
+ * "createTimeMillis" BIGINT,
|
|
|
+ * "modifyTimeMillis" BIGINT,
|
|
|
+ * PRIMARY KEY ("id")
|
|
|
+ * );
|
|
|
+ * <p>
|
|
|
+ * COMMENT ON TABLE "SCRIPT_INFO" IS '节点脚本模版';
|
|
|
+ * <p>
|
|
|
+ * COMMENT ON COLUMN "SCRIPT_INFO"."id" IS 'id';
|
|
|
+ * COMMENT ON COLUMN "SCRIPT_INFO"."createTimeMillis" IS '数据创建时间';
|
|
|
+ * COMMENT ON COLUMN "SCRIPT_INFO"."modifyTimeMillis" IS '数据修改时间';
|
|
|
+ *
|
|
|
+ * @param name 表名
|
|
|
+ * @param desc 描述
|
|
|
+ * @param row 字段信息
|
|
|
+ * @return sql
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public String generateTableSql(String name, String desc, List<TableViewData> row) {
|
|
|
+ StringBuilder stringBuilder = new StringBuilder();
|
|
|
+ stringBuilder.append("CREATE TABLE IF NOT EXISTS ").append(name).append(StrUtil.LF);
|
|
|
+ stringBuilder.append("(").append(StrUtil.LF);
|
|
|
+ for (TableViewData tableViewData : row) {
|
|
|
+ stringBuilder.append(StrUtil.TAB).append(this.generateColumnSql(tableViewData)).append(StrUtil.COMMA).append(StrUtil.LF);
|
|
|
+ }
|
|
|
+ // 主键
|
|
|
+ List<String> primaryKeys = row.stream()
|
|
|
+ .filter(tableViewData -> tableViewData.getPrimaryKey() != null && tableViewData.getPrimaryKey())
|
|
|
+ .map(TableViewRowData::getName)
|
|
|
+ .collect(Collectors.toList());
|
|
|
+ Assert.notEmpty(primaryKeys, I18nMessageUtil.get("i18n.table_without_primary_key.7392"));
|
|
|
+ stringBuilder.append(StrUtil.TAB).append("PRIMARY KEY (").append(CollUtil.join(primaryKeys, StrUtil.COMMA)).append(")").append(StrUtil.LF);
|
|
|
+ stringBuilder.append(") ").append("COMMENT=").append("'").append(desc).append("';");
|
|
|
+ return stringBuilder.toString();
|
|
|
+ }
|
|
|
+
|
|
|
+ @Override
|
|
|
+ public String generateColumnSql(TableViewRowData tableViewRowData) {
|
|
|
+ return generateColumnSql(tableViewRowData, false);
|
|
|
+ }
|
|
|
+
|
|
|
+ private String generateColumnSql(TableViewRowData tableViewRowData, boolean encode) {
|
|
|
+ // id VARCHAR(50) not null default '' comment 'id'
|
|
|
+ StringBuilder stringBuilder = new StringBuilder();
|
|
|
+ stringBuilder.append("`").append(tableViewRowData.getName()).append("`").append(StrUtil.SPACE);
|
|
|
+ String type = tableViewRowData.getType();
|
|
|
+ Assert.hasText(type, I18nMessageUtil.get("i18n.data_type_not_configured_correctly.bf16"));
|
|
|
+ type = type.toUpperCase();
|
|
|
+ switch (type) {
|
|
|
+ case "LONG":
|
|
|
+ stringBuilder.append("BIGINT").append(StrUtil.SPACE);
|
|
|
+ break;
|
|
|
+ case "STRING":
|
|
|
+ stringBuilder.append("VARCHAR(").append(ObjectUtil.defaultIfNull(tableViewRowData.getLen(), 255)).append(")").append(StrUtil.SPACE);
|
|
|
+ break;
|
|
|
+ case "TEXT":
|
|
|
+ stringBuilder.append("TEXT").append(StrUtil.SPACE);
|
|
|
+ break;
|
|
|
+ case "INTEGER":
|
|
|
+ stringBuilder.append("int").append(StrUtil.SPACE);
|
|
|
+ break;
|
|
|
+ case "TINYINT":
|
|
|
+ stringBuilder.append("TINYINT").append(StrUtil.SPACE);
|
|
|
+ break;
|
|
|
+ case "FLOAT":
|
|
|
+ stringBuilder.append("float").append(StrUtil.SPACE);
|
|
|
+ break;
|
|
|
+ case "DOUBLE":
|
|
|
+ stringBuilder.append("double").append(StrUtil.SPACE);
|
|
|
+ break;
|
|
|
+ default:
|
|
|
+ throw new IllegalArgumentException(I18nMessageUtil.get("i18n.data_type_not_supported.fd03") + type);
|
|
|
+ }
|
|
|
+ //
|
|
|
+ Boolean notNull = tableViewRowData.getNotNull();
|
|
|
+ if (notNull != null && notNull) {
|
|
|
+ stringBuilder.append("not null").append(StrUtil.SPACE);
|
|
|
+ }
|
|
|
+ //
|
|
|
+ String defaultValue = tableViewRowData.getDefaultValue();
|
|
|
+ if (StrUtil.isNotEmpty(defaultValue)) {
|
|
|
+ stringBuilder.append("default '").append(defaultValue).append("'").append(StrUtil.SPACE);
|
|
|
+ }
|
|
|
+ stringBuilder.append("comment '").append(tableViewRowData.getComment()).append("'");
|
|
|
+ //
|
|
|
+ String columnSql = stringBuilder.toString();
|
|
|
+ if (encode) {
|
|
|
+ columnSql = StrUtil.replace(columnSql, "'", "\\'");
|
|
|
+ }
|
|
|
+ int length = StrUtil.length(columnSql);
|
|
|
+ Assert.state(length <= 180, I18nMessageUtil.get("i18n.sql_statement_too_long.38d6"));
|
|
|
+ return columnSql;
|
|
|
+ }
|
|
|
+
|
|
|
+ @Override
|
|
|
+ public String delimiter() {
|
|
|
+ return "-- dameng delimiter";
|
|
|
+ }
|
|
|
+}
|
bwcx_jzy