Merge pull request OtterMind#1229 from openai0229/db2-view-ddl-query

Chat2DB-Pro · web-flow · commit 0852517da018 · 2024-03-15T13:43:27.000+08:00
db2-view-ddl-query
diff --git a/chat2db-server/chat2db-plugins/chat2db-db2/src/main/java/ai/chat2db/plugin/db2/DB2MetaData.java b/chat2db-server/chat2db-plugins/chat2db-db2/src/main/java/ai/chat2db/plugin/db2/DB2MetaData.java
@@ -13,10 +13,7 @@
 import ai.chat2db.spi.MetaData;
 import ai.chat2db.spi.SqlBuilder;
 import ai.chat2db.spi.jdbc.DefaultMetaService;
-import ai.chat2db.spi.model.Schema;
-import ai.chat2db.spi.model.TableIndex;
-import ai.chat2db.spi.model.TableIndexColumn;
-import ai.chat2db.spi.model.TableMeta;
+import ai.chat2db.spi.model.*;
 import ai.chat2db.spi.sql.SQLExecutor;
 import ai.chat2db.spi.util.SortUtils;
 import com.google.common.collect.Lists;
@@ -130,6 +127,22 @@ public List<TableIndex> indexes(Connection connection, String databaseName, Stri
 
     }
 
+    private static String VIEW_DDL_SQL="select TEXT from syscat.views where VIEWSCHEMA='%s' and VIEWNAME='%s';";
+    @Override
+    public Table view(Connection connection, String databaseName, String schemaName, String viewName) {
+        String sql = String.format(VIEW_DDL_SQL, schemaName, viewName);
+        Table table = new Table();
+        table.setDatabaseName(databaseName);
+        table.setSchemaName(schemaName);
+        table.setName(viewName);
+        SQLExecutor.getInstance().execute(connection, sql, resultSet -> {
+            if (resultSet.next()) {
+                table.setDdl(resultSet.getString("TEXT")+";");
+            }
+        });
+        return table;
+    }
+
     private TableIndexColumn getTableIndexColumn(ResultSet resultSet) throws SQLException {
         TableIndexColumn tableIndexColumn = new TableIndexColumn();
         tableIndexColumn.setColumnName(resultSet.getString("COLNAME"));
