opensource/Chat2DB
1
0
Fork 0
mirror of https://github.com/CodePhiliaX/Chat2DB.git synced 2025-09-27 09:03:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

Support for creating database and schema

Browse Source
This commit is contained in:
SwallowGG
2023-10-23 15:33:26 +08:00
parent 0df93d771c
commit 61b2ea14a1
23 changed files with 340 additions and 178 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
chat2db-server/chat2db-plugins/chat2db-db2/src/main/java/ai/chat2db/plugin/db2/builder/DB2SqlBuilder.java
View File

@ -2,15 +2,16 @@ package ai.chat2db.plugin.db2.builder;
import ai.chat2db.spi.SqlBuilder;
import ai.chat2db.spi.jdbc.DefaultSqlBuilder;
import ai.chat2db.spi.model.Schema;
import org.apache.commons.lang3.StringUtils;
public class DB2SqlBuilder extends DefaultSqlBuilder {
@Override
public String pageLimit(String sql, int offset, int pageNo, int pageSize) {
int startRow = offset + 1;
int endRow = offset+ pageSize;
int endRow = offset + pageSize;
StringBuilder sqlBuilder = new StringBuilder(sql.length() + 120);
sqlBuilder.append("SELECT * FROM (SELECT TMP_PAGE.*,ROWNUMBER() OVER() AS PAGEHELPER_ROW_ID FROM ( \n");
sqlBuilder.append(sql);
@ -20,4 +21,16 @@ public class DB2SqlBuilder extends DefaultSqlBuilder {
sqlBuilder.append(endRow);
return sqlBuilder.toString();
}
@Override
public String buildCreateSchemaSql(Schema schema) {
StringBuilder sqlBuilder = new StringBuilder();
sqlBuilder.append("CREATE SCHEMA \"" + schema.getName() + "\";");
if (StringUtils.isNotBlank(schema.getComment())) {
sqlBuilder.append("\nCOMMENT ON SCHEMA \"").append(schema.getName()).append("\" IS '").append(schema.getComment()).append("';");
}
return sqlBuilder.toString();
}
}