Fix reversing vertexes with multiple edges

Author: MaxSlobodianyk

reverse_engineering/api.js

@@ -139,12 +139,13 @@ module.exports = {
 					bucketInfo,
 					partitionKey,
 				});
+				let relationships =[];
 				packages.labels.push(nodesData);
 				const labelNames = nodesData.reduce((result, packageData) => result.concat([packageData.collectionName]), []);
-				let relationships = await gremlinHelper.getRelationshipSchema(labelNames);
-				relationships = relationships.filter(data => {
-					return (labelNames.includes(data.start) && labelNames.includes(data.end));
-				});
+				const labelsRelationships = await gremlinHelper.getRelationshipSchema(labelNames);
+				labelsRelationships.forEach(labelRelationships => labelRelationships
+					.filter(relationship => labelNames.includes(relationship.start) && labelNames.includes(relationship.end))
+					.forEach(relationship => relationships.push(relationship)));
 				const relationshipData = await getRelationshipData(relationships, collectionName, recordSamplingSettings, fieldInference);
 				packages.relationships.push(relationshipData);
 				gremlinHelper.close();

reverse_engineering/gremlinHelper.js

@@ -134,15 +134,15 @@ const getRelationshipSchema = (labels, limit = 100) => {
 		return client.submit(`${graphName}.V().hasLabel('${label}').outE().limit(${limit}).as('edge').inV().as('end')
 			.select('edge', 'end').by(label).dedup()`)
 				.then(relationshipData => {
-					const relationship = _.first(relationshipData.toArray());
-					if (!relationship) {
-						return {};
+					const relationships = relationshipData.toArray();
+					if (!relationships) {
+						return [];
 					}
-					return {
-						start: label,
+					return relationships.map(relationship =>({
+							start: label,
 						relationship: relationship.edge,
 						end: relationship.end
-					}
+					}))
 				})
 	}));
 };