fix(openapi): fix lookupAspectSpec (datahub-project#10478)

sleeperdeep · Jun 25, 2024 · 9e24f77 · 9e24f77
1 parent 813c3cf
commit 9e24f77
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/...enapi-servlet/src/main/java/io/datahubproject/openapi/v2/controller/EntityController.java b/...enapi-servlet/src/main/java/io/datahubproject/openapi/v2/controller/EntityController.java
@@ -768,11 +768,11 @@ public List<GenericEntity> toEntityListResponse(
    * @return
    */
   private static AspectSpec lookupAspectSpec(EntitySpec entitySpec, String aspectName) {
-    return Optional.ofNullable(entitySpec.getAspectSpec(aspectName))
-        .orElse(
-            entitySpec.getAspectSpecs().stream()
-                .filter(aspec -> aspec.getName().toLowerCase().equals(aspectName))
-                .findFirst()
-                .get());
+    return entitySpec.getAspectSpec(aspectName) != null
+        ? entitySpec.getAspectSpec(aspectName)
+        : entitySpec.getAspectSpecs().stream()
+            .filter(aspec -> aspec.getName().toLowerCase().equals(aspectName))
+            .findFirst()
+            .get();
   }
 }