yujunjun/OpenMetadata
1
0
Fork 0
mirror of https://github.com/open-metadata/OpenMetadata.git synced 2025-08-29 11:26:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix Table Column Imports (#18924)

Browse Source 
* Fix Table Column Imports

* Fix import status and header name
This commit is contained in:
Mohit Yadav 2024-12-04 22:44:05 +05:30 committed by GitHub
parent 0629750e0a
commit fb9bd00a44
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 22 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
openmetadata-service/src/main/java/org/openmetadata/service/jdbi3/TableRepository.java
View File

@ -1285,34 +1285,38 @@ public class TableRepository extends EntityRepository<Table> {
// column.dataTypeDisplay,column.dataType, column.arrayDataType, column.dataLength,
// column.tags, column.glossaryTerms
Table originalEntity = JsonUtils.deepCopy(table, Table.class);
CSVRecord csvRecord = getNextRecord(printer, csvRecords);
if (csvRecord != null) {
updateColumnsFromCsv(printer, csvRecord);
while (recordIndex < csvRecords.size()) {
CSVRecord csvRecord = getNextRecord(printer, csvRecords);
if (csvRecord != null) {
updateColumnsFromCsv(printer, csvRecord);
String violations = ValidatorUtil.validate(table);
if (violations != null) {
// JSON schema based validation failed for the entity
importFailure(printer, violations, csvRecord);
return;
}
}
}
if (processRecord) {
patchColumns(originalEntity, printer, csvRecord, table);
patchColumns(originalEntity, printer, csvRecords, table);
}
}
private void patchColumns(
Table originalTable, CSVPrinter resultsPrinter, CSVRecord csvRecord, Table entity)
Table originalTable, CSVPrinter resultsPrinter, List<CSVRecord> records, Table entity)
throws IOException {
EntityRepository<Table> repository =
(EntityRepository<Table>) Entity.getEntityRepository(TABLE);
// Validate
String violations = ValidatorUtil.validate(entity);
if (violations != null) {
// JSON schema based validation failed for the entity
importFailure(resultsPrinter, violations, csvRecord);
return;
}
if (Boolean.FALSE.equals(importResult.getDryRun())) { // If not dry run, create the entity
try {
JsonPatch jsonPatch = JsonUtils.getJsonPatch(originalTable, table);
repository.patch(null, table.getId(), importedBy, jsonPatch);
} catch (Exception ex) {
importFailure(resultsPrinter, ex.getMessage(), csvRecord);
importResult.setStatus(ApiStatus.FAILURE);
for (int i = 1; i < records.size(); i++) {
importFailure(resultsPrinter, ex.getMessage(), records.get(i));
importResult.setStatus(ApiStatus.FAILURE);
}
return;
}
} else { // Dry run don't create the entity
@ -1324,7 +1328,9 @@ public class TableRepository extends EntityRepository<Table> {
dryRunCreatedEntities.put(entity.getFullyQualifiedName(), entity);
}
importSuccess(resultsPrinter, csvRecord, ENTITY_UPDATED);
for (int i = 1; i < records.size(); i++) {
importSuccess(resultsPrinter, records.get(i), ENTITY_UPDATED);
}
}
public void updateColumnsFromCsv(CSVPrinter printer, CSVRecord csvRecord) throws IOException {

2
openmetadata-service/src/main/resources/json/data/table/tableCsvDocumentation.json
View File

@ -2,7 +2,7 @@
"summary": "Table CSV file is used for importing and exporting table metadata from and to an **existing** table.",
"headers": [
{
"name": "column.fullyQualifiedName",
"name": "column.name",
"required": true,
"description": "Fully qualified name of the column.",
"examples": [