add getAllRawTypesInvolvedInSignature() for JavaCodeUnit

This method returns the union of all raw types involved in the method's return type, all involved raw types of its parameters and its type parameters. Issue: #723 Signed-off-by: Leonard Husmann <[email protected]>
TNG · May 29, 2023 · cc1d801 · cc1d801
1 parent fcba4e0
commit cc1d801
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/archunit/src/main/java/com/tngtech/archunit/core/domain/JavaCodeUnit.java b/archunit/src/main/java/com/tngtech/archunit/core/domain/JavaCodeUnit.java
@@ -20,6 +20,7 @@
 import java.util.List;
 import java.util.Optional;
 import java.util.Set;
+import java.util.stream.Collectors;
 
 import com.google.common.collect.ImmutableList;
 import com.google.common.collect.ImmutableSet;
@@ -169,6 +170,14 @@ public JavaClass getRawReturnType() {
         return returnType.getRaw();
     }
 
+    public Set<JavaClass> getAllRawTypesInvolvedInSignature() {
+        return ImmutableSet.<JavaClass>builder().addAll(this.returnType.get().getAllInvolvedRawTypes())
+                .addAll(this.parameters.getParameterTypes().stream().map(JavaType::getAllInvolvedRawTypes).flatMap(Set::stream)
+                        .collect(Collectors.toSet()))
+                .addAll(this.typeParameters.stream().map(JavaTypeVariable::getAllInvolvedRawTypes).flatMap(Set::stream).collect(Collectors.toSet()))
+                .build();
+    }
+
     @PublicAPI(usage = ACCESS)
     public Set<JavaFieldAccess> getFieldAccesses() {
         return fieldAccesses;