graalvm, error model alone need reflect (#2811)

Azure · Jun 13, 2024 · d7b5ef3 · d7b5ef3
1 parent 3b62cd6
commit d7b5ef3
Showing 1 changed file with 11 additions and 8 deletions.
diff --git a/javagen/src/main/java/com/azure/autorest/mapper/GraalVmConfigMapper.java b/javagen/src/main/java/com/azure/autorest/mapper/GraalVmConfigMapper.java
@@ -48,18 +48,21 @@ public GraalVmConfig map(ServiceAndModel data) {
         List<String> proxies;
         List<String> reflects;
 
+        final boolean streamStyle = JavaSettings.getInstance().isStreamStyleSerialization();
+
         // Reflect
+        // Exception and error model is still created by reflection in azure-core
         reflects = data.exceptions.stream()
                 .map(e -> e.getPackage() + "." + e.getName())
                 .collect(Collectors.toList());
-        if (!JavaSettings.getInstance().isStreamStyleSerialization()) {
-            reflects.addAll(data.models.stream()
-                    .map(e -> e.getPackage() + "." + e.getName())
-                    .collect(Collectors.toList()));
-            reflects.addAll(data.enums.stream()
-                    .map(m -> m.getPackage() + "." + m.getName())
-                    .collect(Collectors.toList()));
-        }
+        reflects.addAll(data.models.stream()
+                .filter(m -> !streamStyle || (m.getImplementationDetails() != null && m.getImplementationDetails().isException()))
+                .map(m -> m.getPackage() + "." + m.getName())
+                .collect(Collectors.toList()));
+        reflects.addAll(data.enums.stream()
+                .filter(m -> !streamStyle || (m.getImplementationDetails() != null && m.getImplementationDetails().isException()))
+                .map(m -> m.getPackage() + "." + m.getName())
+                .collect(Collectors.toList()));
 
         // Proxy
         proxies = data.serviceClients.stream()