Code optimization

WeDataSphere · Oct 25, 2023 · 47e0d02 · 47e0d02
1 parent 99a8128
commit 47e0d02
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/...vice/jdbc/src/main/java/org/apache/linkis/metadata/query/service/mysql/SqlConnection.java b/...vice/jdbc/src/main/java/org/apache/linkis/metadata/query/service/mysql/SqlConnection.java
@@ -132,6 +132,7 @@ public List<MetaColumnInfo> getColumns(String database, String table)
     ResultSet rs = null;
     ResultSetMetaData meta = null;
     try {
+      List<String> primaryKeys = getPrimaryKeys(getDBConnection(connectMessage, database), table);
       ps = conn.prepareStatement(columnSql);
       ps.setString(1, database);
       ps.setString(2, table);
@@ -143,6 +144,9 @@ public List<MetaColumnInfo> getColumns(String database, String table)
         info.setIndex(i);
         info.setName(meta.getColumnName(i));
         info.setType(meta.getColumnTypeName(i));
+        if (primaryKeys.contains(meta.getColumnName(i))) {
+          info.setPrimaryKey(true);
+        }
         columns.add(info);
       }
     } finally {