mirror of
https://github.com/CodePhiliaX/Chat2DB.git
synced 2025-07-29 02:32:33 +08:00
Support for custom drivers
This commit is contained in:
jipengfei-jpf
23
chat2db-server/chat2db-plugins/chat2db-postgresql/pom.xml
Normal file
23
chat2db-server/chat2db-plugins/chat2db-postgresql/pom.xml
Normal file
@ -0,0 +1,23 @@
|
||||
<?xml version="1.0" encoding="UTF-8"?>
|
||||
<project xmlns="http://maven.apache.org/POM/4.0.0"
|
||||
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
|
||||
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
|
||||
<modelVersion>4.0.0</modelVersion>
|
||||
<parent>
|
||||
<groupId>ai.chat2db</groupId>
|
||||
<artifactId>chat2db-plugins</artifactId>
|
||||
<version>${revision}</version>
|
||||
<relativePath>../pom.xml</relativePath>
|
||||
</parent>
|
||||
|
||||
<artifactId>chat2db-postgresql</artifactId>
|
||||
|
||||
|
||||
<dependencies>
|
||||
<dependency>
|
||||
<groupId>ai.chat2db</groupId>
|
||||
<artifactId>chat2db-spi</artifactId>
|
||||
</dependency>
|
||||
</dependencies>
|
||||
|
||||
</project>
|
||||
@ -0,0 +1,54 @@
|
||||
package ai.chat2db.plugin.postgresql;
|
||||
|
||||
import ai.chat2db.spi.DBManage;
|
||||
import ai.chat2db.spi.sql.Chat2DBContext;
|
||||
import ai.chat2db.spi.sql.ConnectInfo;
|
||||
import ai.chat2db.spi.sql.SQLExecutor;
|
||||
|
||||
import java.sql.SQLException;
|
||||
|
||||
public class PostgreSQLDBManage implements DBManage {
|
||||
@Override
|
||||
public void connectDatabase(String database) {
|
||||
try {
|
||||
SQLExecutor.getInstance().execute("connect " + database + "");
|
||||
} catch (SQLException e) {
|
||||
throw new RuntimeException(e);
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public void modifyDatabase(String databaseName, String newDatabaseName) {
|
||||
|
||||
}
|
||||
|
||||
@Override
|
||||
public void createDatabase(String databaseName) {
|
||||
|
||||
}
|
||||
|
||||
@Override
|
||||
public void dropDatabase(String databaseName) {
|
||||
|
||||
}
|
||||
|
||||
@Override
|
||||
public void createSchema(String databaseName, String schemaName) {
|
||||
|
||||
}
|
||||
|
||||
@Override
|
||||
public void dropSchema(String databaseName, String schemaName) {
|
||||
|
||||
}
|
||||
|
||||
@Override
|
||||
public void modifySchema(String databaseName, String schemaName, String newSchemaName) {
|
||||
|
||||
}
|
||||
|
||||
@Override
|
||||
public void dropTable(String databaseName, String schemaName, String tableName) {
|
||||
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,181 @@
|
||||
package ai.chat2db.plugin.postgresql;
|
||||
|
||||
import ai.chat2db.spi.MetaData;
|
||||
import ai.chat2db.spi.jdbc.DefaultMetaService;
|
||||
import ai.chat2db.spi.sql.SQLExecutor;
|
||||
|
||||
import java.sql.SQLException;
|
||||
|
||||
public class PostgreSQLMetaData extends DefaultMetaService implements MetaData {
|
||||
private String functionSQL =
|
||||
" CREATE OR REPLACE FUNCTION showcreatetable(namespace character varying, tablename character "
|
||||
+ "varying)\n"
|
||||
+ " RETURNS character varying AS\n"
|
||||
+ "\n"
|
||||
+ " $BODY$\n"
|
||||
+ " declare\n"
|
||||
+ " tableScript character varying default '';\n"
|
||||
+ "\n"
|
||||
+ " begin\n"
|
||||
+ " -- columns\n"
|
||||
+ " tableScript:=tableScript || ' CREATE TABLE '|| tablename|| ' ( '|| chr(13)||chr(10) || "
|
||||
+ "array_to_string"
|
||||
+ "(\n"
|
||||
+ " array(\n"
|
||||
+ " select ' ' || concat_ws(' ',fieldName, fieldType, fieldLen, indexType, isNullStr, fieldComment"
|
||||
+ " ) as "
|
||||
+ "column_line\n"
|
||||
+ " from (\n"
|
||||
+ " select a.attname as fieldName,format_type(a.atttypid,a.atttypmod) as fieldType,(case when "
|
||||
+ "atttypmod-4>0 then\n"
|
||||
+ " atttypmod-4 else 0 end) as fieldLen,\n"
|
||||
+ " (case when (select count(*) from pg_constraint where conrelid = a.attrelid and conkey[1]=attnum "
|
||||
+ "and\n"
|
||||
+ " contype='p')>0 then 'PRI'\n"
|
||||
+ " when (select count(*) from pg_constraint where conrelid = a.attrelid and conkey[1]=attnum and "
|
||||
+ "contype='u')>0\n"
|
||||
+ " then 'UNI'\n"
|
||||
+ " when (select count(*) from pg_constraint where conrelid = a.attrelid and conkey[1]=attnum and "
|
||||
+ "contype='f')>0\n"
|
||||
+ " then 'FRI'\n"
|
||||
+ " else '' end) as indexType,\n"
|
||||
+ " (case when a.attnotnull=true then 'not null' else 'null' end) as isNullStr,\n"
|
||||
+ " ' comment ' || col_description(a.attrelid,a.attnum) as fieldComment\n"
|
||||
+ " from pg_attribute a where attstattarget=-1 and attrelid = (select c.oid from pg_class c,"
|
||||
+ "pg_namespace n"
|
||||
+ " where\n"
|
||||
+ " c.relnamespace=n.oid and n.nspname =namespace and relname =tablename)\n"
|
||||
+ "\n"
|
||||
+ " ) as string_columns\n"
|
||||
+ " ),','||chr(13)||chr(10)) || ',';\n"
|
||||
+ "\n"
|
||||
+ "\n"
|
||||
+ " -- 约束\n"
|
||||
+ " tableScript:= tableScript || chr(13)||chr(10) || array_to_string(\n"
|
||||
+ " array(\n"
|
||||
+ " select concat(' CONSTRAINT ',conname ,c ,u,p,f) from (\n"
|
||||
+ " select conname,\n"
|
||||
+ " case when contype='c' then ' CHECK('|| ( select findattname(namespace,tablename,'c') ) ||')' "
|
||||
+ "end "
|
||||
+ "as c "
|
||||
+ ",\n"
|
||||
+ " case when contype='u' then ' UNIQUE('|| ( select findattname(namespace,tablename,'u') ) ||')' "
|
||||
+ "end "
|
||||
+ "as u"
|
||||
+ " ,\n"
|
||||
+ " case when contype='p' then ' PRIMARY KEY ('|| ( select findattname(namespace,tablename,'p') ) "
|
||||
+ "||')' "
|
||||
+ "end as p ,\n"
|
||||
+ " case when contype='f' then ' FOREIGN KEY('|| ( select findattname(namespace,tablename,'u') ) "
|
||||
+ "||') "
|
||||
+ "REFERENCES '||\n"
|
||||
+ " (select p.relname from pg_class p where p.oid=c.confrelid ) || '('|| ( select\n"
|
||||
+ " findattname(namespace,tablename,'u') ) ||')' end as f\n"
|
||||
+ " from pg_constraint c\n"
|
||||
+ " where contype in('u','c','f','p') and conrelid=(\n"
|
||||
+ " select oid from pg_class where relname=tablename and relnamespace =(\n"
|
||||
+ " select oid from pg_namespace where nspname = namespace\n"
|
||||
+ " )\n"
|
||||
+ " )\n"
|
||||
+ " ) as t\n"
|
||||
+ " ) ,',' || chr(13)||chr(10) ) || chr(13)||chr(10) ||' ); ';\n"
|
||||
+ "\n"
|
||||
+ " -- indexs\n"
|
||||
+ " -- CREATE UNIQUE INDEX pg_language_oid_index ON pg_language USING btree (oid); -- table "
|
||||
+ "pg_language\n"
|
||||
+ "\n"
|
||||
+ "\n"
|
||||
+ " --\n"
|
||||
+ " /** **/\n"
|
||||
+ " --- 获取非约束索引 column\n"
|
||||
+ " -- CREATE UNIQUE INDEX pg_language_oid_index ON pg_language USING btree (oid); -- table "
|
||||
+ "pg_language\n"
|
||||
+ " tableScript:= tableScript || chr(13)||chr(10) || chr(13)||chr(10) || array_to_string(\n"
|
||||
+ " array(\n"
|
||||
+ " select 'CREATE INDEX ' || indexrelname || ' ON ' || tablename || ' USING btree '|| '(' || "
|
||||
+ "attname "
|
||||
+ "|| "
|
||||
+ "');' from (\n"
|
||||
+ " SELECT\n"
|
||||
+ " i.relname AS indexrelname , x.indkey,\n"
|
||||
+ "\n"
|
||||
+ " ( select array_to_string (\n"
|
||||
+ " array(\n"
|
||||
+ " select a.attname from pg_attribute a where attrelid=c.oid and a.attnum in ( select unnest(x"
|
||||
+ ".indkey) )\n"
|
||||
+ "\n"
|
||||
+ " )\n"
|
||||
+ " ,',' ) )as attname\n"
|
||||
+ "\n"
|
||||
+ " FROM pg_class c\n"
|
||||
+ " JOIN pg_index x ON c.oid = x.indrelid\n"
|
||||
+ " JOIN pg_class i ON i.oid = x.indexrelid\n"
|
||||
+ " LEFT JOIN pg_namespace n ON n.oid = c.relnamespace\n"
|
||||
+ " WHERE c.relname=tablename and i.relname not in\n"
|
||||
+ " ( select constraint_name from information_schema.key_column_usage where table_name=tablename )\n"
|
||||
+ " )as t\n"
|
||||
+ " ) ,','|| chr(13)||chr(10));\n"
|
||||
+ "\n"
|
||||
+ "\n"
|
||||
+ " -- COMMENT COMMENT ON COLUMN sys_activity.id IS '主键';\n"
|
||||
+ " tableScript:= tableScript || chr(13)||chr(10) || chr(13)||chr(10) || array_to_string(\n"
|
||||
+ " array(\n"
|
||||
+ " SELECT 'COMMENT ON COLUMN' || tablename || '.' || a.attname ||' IS '|| ''''|| d.description "
|
||||
+ "||''''\n"
|
||||
+ " FROM pg_class c\n"
|
||||
+ " JOIN pg_description d ON c.oid=d.objoid\n"
|
||||
+ " JOIN pg_attribute a ON c.oid = a.attrelid\n"
|
||||
+ " WHERE c.relname=tablename\n"
|
||||
+ " AND a.attnum = d.objsubid),','|| chr(13)||chr(10)) ;\n"
|
||||
+ "\n"
|
||||
+ " return tableScript;\n"
|
||||
+ "\n"
|
||||
+ " end\n"
|
||||
+ " $BODY$ LANGUAGE plpgsql;\n"
|
||||
+ "\n"
|
||||
+ " CREATE OR REPLACE FUNCTION findattname(namespace character varying, tablename character "
|
||||
+ "varying, "
|
||||
+ "ctype"
|
||||
+ " character\n"
|
||||
+ " varying)\n"
|
||||
+ " RETURNS character varying as $BODY$\n"
|
||||
+ "\n"
|
||||
+ " declare\n"
|
||||
+ " tt oid ;\n"
|
||||
+ " aname character varying default '';\n"
|
||||
+ "\n"
|
||||
+ " begin\n"
|
||||
+ " tt := oid from pg_class where relname= tablename and relnamespace =(select oid from "
|
||||
+ "pg_namespace "
|
||||
+ "where\n"
|
||||
+ " nspname=namespace) ;\n"
|
||||
+ " aname:= array_to_string(\n"
|
||||
+ " array(\n"
|
||||
+ " select a.attname from pg_attribute a\n"
|
||||
+ " where a.attrelid=tt and a.attnum in (\n"
|
||||
+ " select unnest(conkey) from pg_constraint c where contype=ctype\n"
|
||||
+ " and conrelid=tt and array_to_string(conkey,',') is not null\n"
|
||||
+ " )\n"
|
||||
+ " ),',');\n"
|
||||
+ "\n"
|
||||
+ " return aname;\n"
|
||||
+ " end\n"
|
||||
+ " $BODY$ LANGUAGE plpgsql";
|
||||
|
||||
@Override
|
||||
public String tableDDL(String databaseName, String schemaName, String tableName) {
|
||||
SQLExecutor.getInstance().executeSql(functionSQL.replaceFirst("tableSchema", schemaName), resultSet -> null);
|
||||
String ddlSql = "select showcreatetable('" + schemaName + "','" + tableName + "') as sql";
|
||||
return SQLExecutor.getInstance().executeSql(ddlSql, resultSet -> {
|
||||
try {
|
||||
if (resultSet.next()) {
|
||||
return resultSet.getString("sql");
|
||||
}
|
||||
} catch (SQLException e) {
|
||||
throw new RuntimeException(e);
|
||||
}
|
||||
return null;
|
||||
});
|
||||
}
|
||||
|
||||
}
|
||||
@ -0,0 +1,24 @@
|
||||
package ai.chat2db.plugin.postgresql;
|
||||
|
||||
import ai.chat2db.plugin.postgresql.builder.DBConfigBuilder;
|
||||
import ai.chat2db.spi.DBManage;
|
||||
import ai.chat2db.spi.MetaData;
|
||||
import ai.chat2db.spi.Plugin;
|
||||
import ai.chat2db.spi.config.DBConfig;
|
||||
|
||||
public class PostgreSQLPlugin implements Plugin {
|
||||
@Override
|
||||
public DBConfig getDBConfig() {
|
||||
return DBConfigBuilder.buildDBConfig();
|
||||
}
|
||||
|
||||
@Override
|
||||
public MetaData getMetaData() {
|
||||
return new PostgreSQLMetaData();
|
||||
}
|
||||
|
||||
@Override
|
||||
public DBManage getDBManage() {
|
||||
return new PostgreSQLDBManage();
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,22 @@
|
||||
package ai.chat2db.plugin.postgresql.builder;
|
||||
|
||||
import ai.chat2db.spi.config.DBConfig;
|
||||
import ai.chat2db.spi.config.DriverConfig;
|
||||
import com.google.common.collect.Lists;
|
||||
|
||||
public class DBConfigBuilder {
|
||||
|
||||
public static DBConfig buildDBConfig() {
|
||||
DBConfig dbConfig = new DBConfig();
|
||||
dbConfig.setName("PostgreSQL");
|
||||
dbConfig.setDbType("POSTGRESQL");
|
||||
DriverConfig driverConfig = new DriverConfig();
|
||||
driverConfig.setJdbcDriver("postgresql-42.5.1.jar");
|
||||
driverConfig.setJdbcDriverClass("org.postgresql.Driver");
|
||||
driverConfig.setDownloadJdbcDriverUrls(Lists.newArrayList("https://oss-chat2db.alibaba.com/lib/postgresql-42.5.1.jar"));
|
||||
driverConfig.setName(driverConfig.getJdbcDriver() + ":" + driverConfig.getJdbcDriverClass());
|
||||
dbConfig.setDefaultDriverConfig(driverConfig);
|
||||
dbConfig.setDriverConfigList(Lists.newArrayList(driverConfig));
|
||||
return dbConfig;
|
||||
}
|
||||
}
|
||||
@ -0,0 +1 @@
|
||||
ai.chat2db.plugin.postgresql.PostgreSQLPlugin
|
||||
Reference in New Issue
Block a user