HCK-10222: fix data type property for complex types (#68)

hackolade · Mar 7, 2025 · f1a5f33 · f1a5f33
1 parent 53036d4
commit f1a5f33
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/forward_engineering/helpers/utils.js b/forward_engineering/helpers/utils.js
@@ -314,14 +314,16 @@ const getColumnSchema =
  */
 const decorateType = ({ type, columnDefinition }) => {
 	const deps = { assignTemplates: (__, { name = '', type }) => name + type, tab: value => value, templates: {} };
-	const dataType = getColumnSchema(deps)({
+	const fullDataType = getColumnSchema(deps)({
 		name: '',
 		type,
 		dataTypeMode: columnDefinition.dataTypeMode,
 		jsonSchema: columnDefinition,
 	});
+	const dataType = fullDataType.trim();
+	const isComplexType = /^(array|struct)/i.test(dataType);
 
-	return dataType.trim();
+	return isComplexType ? dataType.replace(/<[\s\S]+>$/, '<>') : dataType;
 };
 
 const generateViewSelectStatement =