From 7fb4dbefecb26870771a7a467d1d22cbac24f366 Mon Sep 17 00:00:00 2001
From: Isira Seneviratne <isirasen96@gmail.com>
Date: Sat, 20 Jul 2024 16:55:56 +0530
Subject: [PATCH] Reuse map if it is a SortedMap instead of creating a new one

---
 .../schabi/newpipe/extractor/downloader/Response.java | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/extractor/src/main/java/org/schabi/newpipe/extractor/downloader/Response.java b/extractor/src/main/java/org/schabi/newpipe/extractor/downloader/Response.java
index 7bc43bb2c..c95b86cad 100644
--- a/extractor/src/main/java/org/schabi/newpipe/extractor/downloader/Response.java
+++ b/extractor/src/main/java/org/schabi/newpipe/extractor/downloader/Response.java
@@ -5,6 +5,7 @@ import javax.annotation.Nullable;
 import java.util.Collections;
 import java.util.List;
 import java.util.Map;
+import java.util.SortedMap;
 import java.util.TreeMap;
 
 /**
@@ -13,7 +14,7 @@ import java.util.TreeMap;
 public class Response {
     private final int responseCode;
     private final String responseMessage;
-    private final Map<String, List<String>> responseHeaders;
+    private final SortedMap<String, List<String>> responseHeaders;
     private final String responseBody;
     private final String latestUrl;
 
@@ -25,8 +26,12 @@ public class Response {
         this.responseCode = responseCode;
         this.responseMessage = responseMessage;
 
-        this.responseHeaders = new TreeMap<>(String.CASE_INSENSITIVE_ORDER);
-        this.responseHeaders.putAll(responseHeaders);
+        if (responseHeaders instanceof SortedMap) {
+            this.responseHeaders = (SortedMap<String, List<String>>) responseHeaders;
+        } else {
+            this.responseHeaders = new TreeMap<>(String.CASE_INSENSITIVE_ORDER);
+            this.responseHeaders.putAll(responseHeaders);
+        }
 
         this.responseBody = responseBody == null ? "" : responseBody;
         this.latestUrl = latestUrl;