fix RE with empty table partition keys

hackolade · Jun 1, 2020 · 294efcb · 294efcb
1 parent abc7373
commit 294efcb
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/reverse_engineering/api.js b/reverse_engineering/api.js
@@ -110,6 +110,7 @@ module.exports = {
 
 const mapTableData = ({ Table }, dbDescription) => {
 	const classification = getClassification(Table.Parameters);
+	const partitionKeys = Table.PartitionKeys || [];
 	const tableData = {
 		dbName: Table.DatabaseName,
 		collectionName: Table.Name,
@@ -120,7 +121,7 @@ const mapTableData = ({ Table }, dbDescription) => {
 			description: Table.Description,
 			externalTable: Table.TableType === 'EXTERNAL_TABLE',
 			tableProperties: mapTableProperties(Table.Parameters),
-			compositePartitionKey: Table.PartitionKeys.map(item => item.Name),
+			compositePartitionKey: partitionKeys.map(item => item.Name),
 			compositeClusteringKey: Table.StorageDescriptor.BucketColumns,
 			sortedByKey: mapSortColumns(Table.StorageDescriptor.SortColumns),
 			compressed: Table.StorageDescriptor.Compressed,
@@ -137,7 +138,7 @@ const mapTableData = ({ Table }, dbDescription) => {
 		},
 		documents: [],
 		validation: {
-			jsonSchema:	getColumns([...Table.StorageDescriptor.Columns, ...Table.PartitionKeys])
+			jsonSchema:	getColumns([...Table.StorageDescriptor.Columns, ...partitionKeys])
 		}
 	};
 	return tableData;