From 1b21708c3068907ac6f27f5b955d3722fae530c0 Mon Sep 17 00:00:00 2001 From: StringKe Date: Sun, 26 May 2024 12:46:02 +0800 Subject: [PATCH] =?UTF-8?q?feat(dialect):=20=E6=B7=BB=E5=8A=A0=20PostgreSQ?= =?UTF-8?q?L=20=E6=96=B9=E8=A8=80=E5=AE=9E=E7=8E=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 添加了 PostgreSQLJdbcDialect 类的实现,以支持 Postgres 数据库的代码生成功能。 主要避免关键词导致的生成失败比如 user --- .../mybatisflex/codegen/dialect/IDialect.java | 6 ++++ .../dialect/impl/PostgreSQLJdbcDialect.java | 29 +++++++++++++++++++ 2 files changed, 35 insertions(+) create mode 100644 mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/dialect/impl/PostgreSQLJdbcDialect.java diff --git a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/dialect/IDialect.java b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/dialect/IDialect.java index 67ab6b5b..964159b6 100644 --- a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/dialect/IDialect.java +++ b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/dialect/IDialect.java @@ -20,6 +20,7 @@ import com.mybatisflex.codegen.dialect.impl.DefaultJdbcDialect; import com.mybatisflex.codegen.dialect.impl.MySqlJdbcDialect; import com.mybatisflex.codegen.dialect.impl.OracleJdbcDialect; import com.mybatisflex.codegen.dialect.impl.SqliteDialect; +import com.mybatisflex.codegen.dialect.impl.PostgreSQLJdbcDialect; import com.mybatisflex.codegen.entity.Table; import java.sql.Connection; @@ -54,6 +55,11 @@ public interface IDialect { */ IDialect SQLITE = new SqliteDialect(); + /** + * PostgreSQL 方言。 + */ + IDialect POSTGRESQL = new PostgreSQLJdbcDialect(); + /** * 构建表和列的信息。 * diff --git a/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/dialect/impl/PostgreSQLJdbcDialect.java b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/dialect/impl/PostgreSQLJdbcDialect.java new file mode 100644 index 00000000..2d9962db --- /dev/null +++ b/mybatis-flex-codegen/src/main/java/com/mybatisflex/codegen/dialect/impl/PostgreSQLJdbcDialect.java @@ -0,0 +1,29 @@ +/* + * Copyright (c) 2022-2023, Mybatis-Flex (fuhai999@gmail.com). + *

+ * 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. + */ +package com.mybatisflex.codegen.dialect.impl; + +import com.mybatisflex.codegen.dialect.AbstractJdbcDialect; +import com.mybatisflex.core.util.StringUtil; + +/** + * @author strignke + */ +public class PostgreSQLJdbcDialect extends AbstractJdbcDialect { + @Override + protected String forBuildColumnsSql(String schema, String tableName) { + return "SELECT * FROM " + (StringUtil.isNotBlank(schema) ? schema + "." : "") + "\"" + tableName + "\"" + " WHERE 1 = 2"; + } +}