RE: implement reverse-engineering of primary keys with single key as pk on field level

Author: Vitalii4as

reverse_engineering/helpers/postgresHelpers/reorganizePrimaryKeys.js

@@ -0,0 +1,37 @@
+const reorganizePrimaryKeys = (attributes, entityLevel) => {
+	if (entityLevel.primaryKey?.[0]?.compositePrimaryKey?.length !== 1) {
+		return { attributes, entityLevel };
+	}
+
+	const primaryKey = entityLevel.primaryKey[0];
+	const primaryKeyAttributeName = primaryKey.compositePrimaryKey[0];
+	const attributesWithPk = attributes.map(attribute => {
+		if (attribute.name !== primaryKeyAttributeName) {
+			return attribute;
+		}
+
+		return setPrimaryKeyData(attribute, primaryKey);
+	});
+
+	const filteredEntityLevel = { ...entityLevel, primaryKey: [] };
+
+	return { attributes: attributesWithPk, entityLevel: filteredEntityLevel };
+};
+
+const setPrimaryKeyData = (attribute, primaryKey) => {
+	return {
+		...attribute,
+		primaryKey: true,
+		primaryKeyOptions: [
+			{
+				constraintName: primaryKey.constraintName,
+				indexInclude: primaryKey.include,
+				indexStorageParameters: primaryKey.indexStorageParameters,
+				indexTablespace: primaryKey.indexTablespace,
+				indexComment: primaryKey.indexComment,
+			},
+		],
+	};
+};
+
+module.exports = { reorganizePrimaryKeys };

reverse_engineering/helpers/postgresService.js

@@ -45,6 +45,7 @@ const {
 } = require('./postgresHelpers/viewHelper');
 const { setDependencies: setDependenciesInTriggerHelper, getTriggers } = require('./postgresHelpers/triggerHelper');
 const queryConstants = require('./queryConstants');
+const { reorganizePrimaryKeys } = require('./postgresHelpers/reorganizePrimaryKeys');
 
 let currentSshTunnel = null;
 let _ = null;
@@ -304,10 +305,12 @@ module.exports = {
 			targetAttributes = setSubtypeFromSampledJsonValues(targetAttributes, documents);
 		}
 
+		const { attributes, entityLevel: updatedEntityLevel } = reorganizePrimaryKeys(targetAttributes, entityLevel);
+
 		return {
 			name: tableName,
-			entityLevel,
-			jsonSchema: getJsonSchema(targetAttributes),
+			entityLevel: updatedEntityLevel,
+			jsonSchema: getJsonSchema(attributes),
 			documents,
 			relationships,
 		};