be-gay-do-crimes
/
NewPipeExtractor
mirror of https://github.com/TeamNewPipe/NewPipeExtractor
1
0
Fork 0
Code Issues Releases Wiki Activity

Use TreeMap in Response

This commit is contained in:
Isira Seneviratne 2024-07-20 13:10:15 +05:30
parent 592f1596e6
commit ec5fb7dc9a
1 changed files with 8 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
extractor/src/main/java/org/schabi/newpipe/extractor/downloader/Response.java
View File

@ -5,6 +5,7 @@ import javax.annotation.Nullable;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.TreeMap;
/**
* A Data class used to hold the results from requests made by the Downloader implementation.
@ -14,17 +15,18 @@ public class Response {
private final String responseMessage;
private final Map<String, List<String>> responseHeaders;
private final String responseBody;
private final String latestUrl;
public Response(final int responseCode,
final String responseMessage,
@Nullable final Map<String, List<String>> responseHeaders,
@Nonnull final Map<String, List<String>> responseHeaders,
@Nullable final String responseBody,
@Nullable final String latestUrl) {
this.responseCode = responseCode;
this.responseMessage = responseMessage;
this.responseHeaders = responseHeaders == null ? Collections.emptyMap() : responseHeaders;
this.responseHeaders = new TreeMap<>(String.CASE_INSENSITIVE_ORDER);
this.responseHeaders.putAll(responseHeaders);
this.responseBody = responseBody == null ? "" : responseBody;
this.latestUrl = latestUrl;
@ -71,13 +73,8 @@ public class Response {
*/
@Nullable
public String getHeader(final String name) {
for (final Map.Entry<String, List<String>> headerEntry : responseHeaders.entrySet()) {
final String key = headerEntry.getKey();
if (key != null && key.equalsIgnoreCase(name) && !headerEntry.getValue().isEmpty()) {
return headerEntry.getValue().get(0);
}
}
return null;
// Header lookup is case-insensitive
final var values = responseHeaders.getOrDefault(name, Collections.emptyList());
return values.isEmpty() ? null : values.get(0);
}
}