opensource/Chat2DB
1
0
Fork 0
mirror of https://github.com/CodePhiliaX/Chat2DB.git synced 2025-07-30 03:03:13 +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

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

@ -392,43 +392,47 @@ public class DlTemplateServiceImpl implements DlTemplateService {
}
private List<Header> setColumnInfo(List<Header> headers, String tableName, String schemaName, String databaseName) {
TableQueryParam tableQueryParam = new TableQueryParam();
tableQueryParam.setTableName(MetaNameUtils.getMetaName(tableName));
tableQueryParam.setSchemaName(schemaName);
tableQueryParam.setDatabaseName(databaseName);
tableQueryParam.setRefresh(true);
List<TableColumn> columns = tableService.queryColumns(tableQueryParam);
if (CollectionUtils.isEmpty(columns)) {
return headers;
}
Map<String, TableColumn> columnMap = columns.stream().collect(Collectors.toMap(TableColumn::getName, tableColumn -> tableColumn));
List<TableIndex> tableIndices = tableService.queryIndexes(tableQueryParam);
if (!CollectionUtils.isEmpty(tableIndices)) {
for (TableIndex tableIndex : tableIndices) {
if ("PRIMARY".equalsIgnoreCase(tableIndex.getType())) {
List<TableIndexColumn> columnList = tableIndex.getColumnList();
if (!CollectionUtils.isEmpty(columnList)) {
for (TableIndexColumn tableIndexColumn : columnList) {
TableColumn tableColumn = columnMap.get(tableIndexColumn.getColumnName());
if (tableColumn != null) {
tableColumn.setPrimaryKey(true);
try {
TableQueryParam tableQueryParam = new TableQueryParam();
tableQueryParam.setTableName(MetaNameUtils.getMetaName(tableName));
tableQueryParam.setSchemaName(schemaName);
tableQueryParam.setDatabaseName(databaseName);
tableQueryParam.setRefresh(true);
List<TableColumn> columns = tableService.queryColumns(tableQueryParam);
if (CollectionUtils.isEmpty(columns)) {
return headers;
}
Map<String, TableColumn> columnMap = columns.stream().collect(Collectors.toMap(TableColumn::getName, tableColumn -> tableColumn));
List<TableIndex> tableIndices = tableService.queryIndexes(tableQueryParam);
if (!CollectionUtils.isEmpty(tableIndices)) {
for (TableIndex tableIndex : tableIndices) {
if ("PRIMARY".equalsIgnoreCase(tableIndex.getType())) {
List<TableIndexColumn> columnList = tableIndex.getColumnList();
if (!CollectionUtils.isEmpty(columnList)) {
for (TableIndexColumn tableIndexColumn : columnList) {
TableColumn tableColumn = columnMap.get(tableIndexColumn.getColumnName());
if (tableColumn != null) {
tableColumn.setPrimaryKey(true);
}
}
}
}
}
}
}
for (Header header : headers) {
TableColumn tableColumn = columnMap.get(header.getName());
if (tableColumn != null) {
header.setPrimaryKey(tableColumn.getPrimaryKey());
header.setComment(tableColumn.getComment());
header.setDefaultValue(tableColumn.getDefaultValue());
header.setNullable(tableColumn.getNullable());
header.setColumnSize(tableColumn.getColumnSize());
header.setDecimalDigits(tableColumn.getDecimalDigits());
for (Header header : headers) {
TableColumn tableColumn = columnMap.get(header.getName());
if (tableColumn != null) {
header.setPrimaryKey(tableColumn.getPrimaryKey());
header.setComment(tableColumn.getComment());
header.setDefaultValue(tableColumn.getDefaultValue());
header.setNullable(tableColumn.getNullable());
header.setColumnSize(tableColumn.getColumnSize());
header.setDecimalDigits(tableColumn.getDecimalDigits());
}
}
}catch (Exception e){
log.error("setColumnInfo error:",e);
}
return headers;
}