opensource/Chat2DB
1
0
Fork 0
mirror of https://github.com/CodePhiliaX/Chat2DB.git synced 2025-07-30 11:12:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix issues 792 NullPointerException

Browse Source
This commit is contained in:
SwallowGG
2023-11-11 19:09:46 +08:00
parent 097900a04a
commit 06e779da8c
2 changed files with 39 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
chat2db-server/chat2db-plugins/chat2db-postgresql/src/main/java/ai/chat2db/plugin/postgresql/PostgreSQLMetaData.java
View File

@ -230,6 +230,10 @@ public class PostgreSQLMetaData extends DefaultMetaService implements MetaData {
TableIndex tableIndex = map.get(keyName); TableIndex tableIndex = map.get(keyName);
if (tableIndex != null) { if (tableIndex != null) {
List<TableIndexColumn> columnList = tableIndex.getColumnList(); List<TableIndexColumn> columnList = tableIndex.getColumnList();
if(columnList == null){
columnList = new ArrayList<>();
tableIndex.setColumnList(columnList);
}
columnList.add(getTableIndexColumn(resultSet)); columnList.add(getTableIndexColumn(resultSet));
columnList = columnList.stream().sorted(Comparator.comparing(TableIndexColumn::getOrdinalPosition)) columnList = columnList.stream().sorted(Comparator.comparing(TableIndexColumn::getOrdinalPosition))
.collect(Collectors.toList()); .collect(Collectors.toList());

6
chat2db-server/chat2db-server-domain/chat2db-server-domain-core/src/main/java/ai/chat2db/server/domain/core/impl/DlTemplateServiceImpl.java
View File

@ -392,6 +392,7 @@ public class DlTemplateServiceImpl implements DlTemplateService {
} }
private List<Header> setColumnInfo(List<Header> headers, String tableName, String schemaName, String databaseName) { private List<Header> setColumnInfo(List<Header> headers, String tableName, String schemaName, String databaseName) {
try {
TableQueryParam tableQueryParam = new TableQueryParam(); TableQueryParam tableQueryParam = new TableQueryParam();
tableQueryParam.setTableName(MetaNameUtils.getMetaName(tableName)); tableQueryParam.setTableName(MetaNameUtils.getMetaName(tableName));
tableQueryParam.setSchemaName(schemaName); tableQueryParam.setSchemaName(schemaName);
@ -402,7 +403,6 @@ public class DlTemplateServiceImpl implements DlTemplateService {
return headers; return headers;
} }
Map<String, TableColumn> columnMap = columns.stream().collect(Collectors.toMap(TableColumn::getName, tableColumn -> tableColumn)); Map<String, TableColumn> columnMap = columns.stream().collect(Collectors.toMap(TableColumn::getName, tableColumn -> tableColumn));
List<TableIndex> tableIndices = tableService.queryIndexes(tableQueryParam); List<TableIndex> tableIndices = tableService.queryIndexes(tableQueryParam);
if (!CollectionUtils.isEmpty(tableIndices)) { if (!CollectionUtils.isEmpty(tableIndices)) {
for (TableIndex tableIndex : tableIndices) { for (TableIndex tableIndex : tableIndices) {
@ -430,6 +430,10 @@ public class DlTemplateServiceImpl implements DlTemplateService {
header.setDecimalDigits(tableColumn.getDecimalDigits()); header.setDecimalDigits(tableColumn.getDecimalDigits());
} }
} }
}catch (Exception e){
log.error("setColumnInfo error:",e);
}
return headers; return headers;
} }