From 01e1d29ac29a691563364a954e3d196ee0aa75f5 Mon Sep 17 00:00:00 2001
From: tsantalis <tsantalis@gmail.com>
Date: Sat, 21 Dec 2024 20:12:41 -0500
Subject: [PATCH] Improve type inference based matching

Restores method pair mapping in class
org.hibernate.search.mapper.orm.loading.impl.HibernateOrmEntityIdEntityLoadingStrategy
https://github.com/hibernate/hibernate-search/commit/6133c600d0fc15ce3482aaf9c3a29b4a222c98e8
---
 .../gr/uom/java/xmi/decomposition/OperationInvocation.java  | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/main/java/gr/uom/java/xmi/decomposition/OperationInvocation.java b/src/main/java/gr/uom/java/xmi/decomposition/OperationInvocation.java
index d30634010b..2dbf8b6aca 100644
--- a/src/main/java/gr/uom/java/xmi/decomposition/OperationInvocation.java
+++ b/src/main/java/gr/uom/java/xmi/decomposition/OperationInvocation.java
@@ -589,6 +589,12 @@ private static boolean collectionMatch(UMLType parameterType, UMLType type) {
 					if(typeArgument.toString().length() == 1 && Character.isUpperCase(typeArgument.toString().charAt(0))) {
 						return true;
 					}
+					if(type.getTypeArguments().size() == 1) {
+						UMLType typeArgument2 = type.getTypeArguments().get(0);
+						if(typeArgument2.getClassType().equals(typeArgument.getClassType())) {
+							return true;
+						}
+					}
 				}
 			}
 		}