Performance improvement in Resource.Builder#mergeResources

Signed-off-by: Jan Supol <[email protected]>
eclipse-ee4j · Oct 21, 2019 · f1b5935 · f1b5935
1 parent 7b3a86b
commit f1b5935
Showing 1 changed file with 25 additions and 17 deletions.
diff --git a/core-server/src/main/java/org/glassfish/jersey/server/model/Resource.java b/core-server/src/main/java/org/glassfish/jersey/server/model/Resource.java
@@ -25,14 +25,19 @@
 import java.util.LinkedHashSet;
 import java.util.LinkedList;
 import java.util.List;
+import java.util.ListIterator;
+import java.util.Map;
 import java.util.Set;
 import java.util.stream.Collectors;
 
 import javax.ws.rs.Path;
+import javax.ws.rs.core.MultivaluedHashMap;
+import javax.ws.rs.core.MultivaluedMap;
 
 import org.glassfish.jersey.Severity;
 import org.glassfish.jersey.internal.Errors;
 import org.glassfish.jersey.internal.guava.Preconditions;
+import org.glassfish.jersey.internal.util.collection.MultivaluedStringMap;
 import org.glassfish.jersey.internal.util.collection.Value;
 import org.glassfish.jersey.internal.util.collection.Values;
 import org.glassfish.jersey.server.internal.LocalizationMessages;
@@ -543,30 +548,33 @@ private void onBuildChildResource(Builder childResourceBuilder, Resource.Data ch
         }
 
         private List<Resource.Data> mergeResources(List<Resource.Data> resources) {
-            List<Resource.Data> mergedResources = new ArrayList<>();
-            for (int i = 0; i < resources.size(); i++) {
-                Resource.Data outer = resources.get(i);
-                Resource.Builder builder = null;
+            final MultivaluedMap<String, Resource.Data> resourceData = new MultivaluedHashMap<>();
+            final List<Resource.Data> mergedResources = new ArrayList<>();
 
-                for (int j = i + 1; j < resources.size(); j++) {
-                    Resource.Data inner = resources.get(j);
+            for (Iterator<Resource.Data> resourcesIterator = resources.iterator(); resourcesIterator.hasNext();) {
+                Resource.Data data = resourcesIterator.next();
+                if (resourceData.containsKey(data.path)) {
+                    resourcesIterator.remove();
+                }
+                resourceData.add(data.path, data);
+            }
 
-                    if (outer.path.equals(inner.path)) {
-                        if (builder == null) {
-                            builder = Resource.builder(outer);
+            for (Map.Entry<String, List<Resource.Data>> entry : resourceData.entrySet()) {
+                if (entry.getValue().size() == 1) {
+                    mergedResources.add(entry.getValue().get(0));
+                } else {
+                    Resource.Builder builder = null;
+                    for (Resource.Data data : entry.getValue()) {
+                        if (builder != null) {
+                            builder.mergeWith(data);
+                        } else {
+                            builder = Resource.builder(data);
                         }
-                        builder.mergeWith(inner);
-                        resources.remove(j);
-                        //noinspection AssignmentToForLoopParameter
-                        j--;
                     }
-                }
-                if (builder == null) {
-                    mergedResources.add(outer);
-                } else {
                     mergedResources.add(builder.buildResourceData());
                 }
             }
+
             return mergedResources;
         }