mirrors/public-transport-enabler
1
0
Fork 0
mirror of https://gitlab.com/oeffi/public-transport-enabler.git synced 2025-07-18 08:19:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

HttpClient: Support custom request headers.

Browse source
This commit is contained in:
Andreas Schildbach 2015-07-19 14:34:16 +02:00
parent 00a395a9ad
commit 05391cab75
1 changed files with 9 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

9
enabler/src/de/schildbach/pte/util/HttpClient.java
View file

@ -28,6 +28,7 @@ import java.net.HttpURLConnection;
import java.net.MalformedURLException; import java.net.MalformedURLException;
import java.net.URL; import java.net.URL;
import java.nio.charset.Charset; import java.nio.charset.Charset;
import java.util.HashMap;
import java.util.List; import java.util.List;
import java.util.Map; import java.util.Map;
import java.util.regex.Matcher; import java.util.regex.Matcher;
@ -54,6 +55,7 @@ public final class HttpClient
{ {
@Nullable @Nullable
private String userAgent = null; private String userAgent = null;
private Map<String, String> headers = new HashMap<String, String>();
@Nullable @Nullable
private String sessionCookieName = null; private String sessionCookieName = null;
@Nullable @Nullable
@ -73,6 +75,11 @@ public final class HttpClient
this.userAgent = userAgent; this.userAgent = userAgent;
} }
public void setHeader(final String headerName, final String headerValue)
{
this.headers.put(headerName, headerValue);
}
public void setSessionCookieName(final String sessionCookieName) public void setSessionCookieName(final String sessionCookieName)
{ {
this.sessionCookieName = sessionCookieName; this.sessionCookieName = sessionCookieName;
@ -136,6 +143,8 @@ public final class HttpClient
connection.setDoOutput(postRequest != null); connection.setDoOutput(postRequest != null);
connection.setConnectTimeout(SCRAPE_CONNECT_TIMEOUT); connection.setConnectTimeout(SCRAPE_CONNECT_TIMEOUT);
connection.setReadTimeout(SCRAPE_READ_TIMEOUT); connection.setReadTimeout(SCRAPE_READ_TIMEOUT);
for (final Map.Entry<String, String> entry : headers.entrySet())
connection.addRequestProperty(entry.getKey(), entry.getValue());
if (userAgent != null) if (userAgent != null)
connection.addRequestProperty("User-Agent", userAgent); connection.addRequestProperty("User-Agent", userAgent);
connection.addRequestProperty("Accept", SCRAPE_ACCEPT); connection.addRequestProperty("Accept", SCRAPE_ACCEPT);