Support for custom drivers

This commit is contained in:
jipengfei-jpf
2023-06-23 18:25:41 +08:00
parent e4466fb34d
commit 5047569358
96 changed files with 2604 additions and 2 deletions

View File

@ -0,0 +1,22 @@
<?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-oracle</artifactId>
<dependencies>
<dependency>
<groupId>ai.chat2db</groupId>
<artifactId>chat2db-spi</artifactId>
</dependency>
</dependencies>
</project>

View File

@ -0,0 +1,45 @@
package ai.chat2db.plugin.oracle;
import ai.chat2db.spi.DBManage;
public class OracleDBManage implements DBManage {
@Override
public void connectDatabase(String database) {
}
@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) {
}
}

View File

@ -0,0 +1,26 @@
package ai.chat2db.plugin.oracle;
import ai.chat2db.spi.MetaData;
import ai.chat2db.spi.jdbc.DefaultMetaService;
import ai.chat2db.spi.sql.SQLExecutor;
import java.sql.SQLException;
public class OracleMetaData extends DefaultMetaService implements MetaData {
@Override
public String tableDDL(String databaseName, String schemaName, String tableName) {
String sql = "select dbms_metadata.get_ddl('TABLE','"+tableName+"') as sql from dual,"
+ "user_tables where table_name = '" + tableName + "'";
return SQLExecutor.getInstance().executeSql(sql, resultSet -> {
try {
if (resultSet.next()) {
return resultSet.getString("sql");
}
} catch (SQLException e) {
throw new RuntimeException(e);
}
return null;
});
}
}

View File

@ -0,0 +1,25 @@
package ai.chat2db.plugin.oracle;
import ai.chat2db.plugin.oracle.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 OraclePlugin implements Plugin {
@Override
public DBConfig getDBConfig() {
return DBConfigBuilder.buildDBConfig();
}
@Override
public MetaData getMetaData() {
return new OracleMetaData();
}
@Override
public DBManage getDBManage() {
return new OracleDBManage();
}
}

View File

@ -0,0 +1,24 @@
package ai.chat2db.plugin.oracle.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("Oracle");
dbConfig.setDbType("ORACLE");
DriverConfig driverConfig = new DriverConfig();
driverConfig.setJdbcDriver("ojdbc8-19.3.0.0.jar,orai18n-19.3.0.0.jar");
driverConfig.setJdbcDriverClass("oracle.jdbc.driver.OracleDriver");
driverConfig.setDownloadJdbcDriverUrls(Lists.newArrayList("https://oss-chat2db.alibaba.com/lib/ojdbc8-19.3.0.0.jar", "https://oss-chat2db.alibaba.com/lib/orai18n-19.3.0.0.jar"));
driverConfig.setName(driverConfig.getJdbcDriver() + ":" + driverConfig.getJdbcDriverClass());
dbConfig.setDefaultDriverConfig(driverConfig);
dbConfig.setDriverConfigList(Lists.newArrayList(driverConfig));
return dbConfig;
}
}

View File

@ -0,0 +1 @@
ai.chat2db.plugin.oracle.OraclePlugin