#159 review suggestions

mapstruct · Mar 17, 2024 · 811de7a · 811de7a
1 parent d19770f
commit 811de7a
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/src/main/java/org/mapstruct/intellij/util/MapstructUtil.java b/src/main/java/org/mapstruct/intellij/util/MapstructUtil.java
@@ -211,18 +211,18 @@ public static boolean isFluentSetter(@NotNull PsiMethod method, PsiType psiType)
             isAssignableFromReturnTypeOrSuperTypes( psiType, method.getReturnType() );
     }
 
-    private static boolean isAssignableFromReturnTypeOrSuperTypes(PsiType psiType, @Nullable PsiType returnType) {
-
-        if ( returnType == null ) {
-            return false;
-        }
+    private static boolean isAssignableFromReturnTypeOrSuperTypes(PsiType psiType, PsiType returnType) {
 
         if ( isAssignableFrom( psiType, returnType ) ) {
             return true;
         }
 
-        return Arrays.stream( returnType.getSuperTypes() )
-            .anyMatch( superType -> isAssignableFrom( psiType, superType ) );
+        for ( PsiType superType : returnType.getSuperTypes() ) {
+            if ( isAssignableFrom( psiType, superType ) ) {
+                return true;
+            }
+        }
+        return false;
     }
 
     private static boolean isAssignableFrom(PsiType psiType, @Nullable PsiType returnType) {