opensource/Chat2DB
1
0
Fork 0
mirror of https://github.com/CodePhiliaX/Chat2DB.git synced 2025-07-29 10:43:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat(PostgreSQL): Get the DDL of the PostgreSQL sequence

Browse Source
This commit is contained in:
Sylphy
2025-04-11 09:52:11 +08:00
parent 92fdeb368d
commit d9779a4b2e
14 changed files with 289 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

71
chat2db-server/chat2db-plugins/chat2db-postgresql/src/main/java/ai/chat2db/plugin/postgresql/PostgreSQLMetaData.java
View File

@ -11,16 +11,17 @@ import ai.chat2db.spi.sql.SQLExecutor;
import ai.chat2db.spi.util.SortUtils;
import com.google.common.collect.Lists;
import jakarta.validation.constraints.NotEmpty;
import lombok.SneakyThrows;
import org.apache.commons.lang3.StringUtils;
import java.sql.Connection;
import java.sql.DatabaseMetaData;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.*;
import java.util.stream.Collectors;
import static ai.chat2db.plugin.postgresql.consts.SQLConst.DROP_TYPE_SQL;
import static ai.chat2db.plugin.postgresql.consts.SQLConst.TABLE_DEF_FUNCTION_SQL;
import static ai.chat2db.plugin.postgresql.consts.SQLConst.*;
import static ai.chat2db.spi.util.SortUtils.sortDatabase;
public class PostgreSQLMetaData extends DefaultMetaService implements MetaData {
@ -104,7 +105,7 @@ public class PostgreSQLMetaData extends DefaultMetaService implements MetaData {
@Override
public String tableDDL(Connection connection, String databaseName, String schemaName, String tableName) {
SQLExecutor.getInstance().execute(connection, String.format(DROP_TYPE_SQL,schemaName,"tabledefs"), resultSet -> null);
SQLExecutor.getInstance().execute(connection, String.format(DROP_TYPE_SQL, schemaName, "tabledefs"), resultSet -> null);
SQLExecutor.getInstance().execute(connection, TABLE_DEF_FUNCTION_SQL, resultSet -> null);
String ddlSql = String.format("select * from pg_get_tabledef('%s','%s',false,'COMMENTS') as ddl;", schemaName, tableName);
return SQLExecutor.getInstance().execute(connection, ddlSql, resultSet -> {
@ -314,4 +315,68 @@ public class PostgreSQLMetaData extends DefaultMetaService implements MetaData {
public List<String> getSystemSchemas() {
return systemSchemas;
}
@Override
@SneakyThrows
public String sequenceDDL(Connection connection, @NotEmpty String databaseName, String schemaName,
@NotEmpty String sequenceName) {
DatabaseMetaData metaData = connection.getMetaData();
Double databaseProductVersion = Double.valueOf(metaData.getDatabaseProductVersion());
return SQLExecutor.getInstance().preExecute(connection, EXPORT_SEQUENCE_DDL_SQL, resultSet -> {
StringBuilder stringBuilder = new StringBuilder();
if (resultSet.next()) {
String nspname = resultSet.getString("nspname");
String relname = resultSet.getString("relname");
String typname = resultSet.getString("typname");
String seqcache = resultSet.getString("seqcache");
String rolname = resultSet.getString("rolname");
String comment = resultSet.getString("comment");
String seqstart = resultSet.getString("seqstart");
String seqincrement = resultSet.getString("seqincrement");
String seqmax = resultSet.getString("seqmax");
String seqmin = resultSet.getString("seqmin");
Boolean seqcycle = resultSet.getBoolean("seqcycle");
switch (typname) {
case "int2" -> typname = "SMALLINT";
case "int8" -> typname = "BIGINT";
default -> typname = "INTEGER";
}
stringBuilder.append("CREATE SEQUENCE ").append(nspname).append(".").append(relname).append("\n ");
if (databaseProductVersion >= 10.0){
stringBuilder.append(" AS ").append(typname).append("\n ");
}
Optional.ofNullable(seqstart).ifPresent(v -> stringBuilder.append(" START WITH ").append(v).append("\n "));
Optional.ofNullable(seqincrement).ifPresent(v -> stringBuilder.append(" INCREMENT BY ").append(v).append("\n "));
Optional.ofNullable(seqmin).ifPresent(v -> stringBuilder.append(" MINVALUE ").append(v).append("\n "));
Optional.ofNullable(seqmax).ifPresent(v -> stringBuilder.append(" MAXVALUE ").append(v).append("\n "));
Optional.ofNullable(seqcache).ifPresent(v -> stringBuilder.append(" CACHE ").append(v).append("\n "));
Optional.ofNullable(seqcycle).ifPresent(v -> {
if (seqcycle) {
stringBuilder.append(" CYCLE ").append("\n ");
}
});
stringBuilder.append(";").append("\n ").append("\n ");
Optional.ofNullable(comment).ifPresent(v -> stringBuilder.append("COMMENT ON SEQUENCE ")
.append(nspname).append(".").append(relname)
.append(" IS '").append(comment).append("';").append("\n ").append("\n "));
Optional.ofNullable(rolname).ifPresent(v -> stringBuilder.append("ALTER SEQUENCE ")
.append(nspname).append(".").append(relname)
.append(" OWNED BY ").append(v).append(";"));
}
return stringBuilder.toString();
},
sequenceName, schemaName);
}
}

25
chat2db-server/chat2db-plugins/chat2db-postgresql/src/main/java/ai/chat2db/plugin/postgresql/consts/SQLConst.java
View File

@ -640,4 +640,29 @@ public class SQLConst {
JOIN pg_catalog.pg_namespace n ON n.oid = t.typnamespace
WHERE t.typtype = 'e'
GROUP BY n.nspname, t.typname;""";
public static final String EXPORT_SEQUENCE_DDL_SQL = """
SELECT n.nspname,
c.relname,
t.typname,
a.rolname,
obj_description(c.oid, 'pg_class') AS comment,
s.seqstart,
s.seqincrement,
s.seqmax,
s.seqmin,
s.seqcycle,
s.seqcache
FROM pg_sequence s
JOIN
pg_class c ON c.oid = s.seqrelid
JOIN
pg_namespace n ON n.oid = c.relnamespace
JOIN
pg_authid a ON a.oid = c.relowner
JOIN
pg_type t ON s.seqtypid = t.oid
WHERE c.relname = ?
and n.nspname = ?;
""";
}