1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28 package org.apache.hc.client5.http.protocol;
29
30 import java.io.IOException;
31
32 import org.apache.hc.client5.http.RouteInfo;
33 import org.apache.hc.core5.annotation.Contract;
34 import org.apache.hc.core5.annotation.ThreadingBehavior;
35 import org.apache.hc.core5.http.EntityDetails;
36 import org.apache.hc.core5.http.HeaderElements;
37 import org.apache.hc.core5.http.HttpException;
38 import org.apache.hc.core5.http.HttpHeaders;
39 import org.apache.hc.core5.http.HttpRequest;
40 import org.apache.hc.core5.http.HttpRequestInterceptor;
41 import org.apache.hc.core5.http.protocol.HttpContext;
42 import org.apache.hc.core5.util.Args;
43 import org.slf4j.Logger;
44 import org.slf4j.LoggerFactory;
45
46
47
48
49
50
51
52
53 @Contract(threading = ThreadingBehavior.STATELESS)
54 public class RequestClientConnControl implements HttpRequestInterceptor {
55
56 private static final Logger LOG = LoggerFactory.getLogger(RequestClientConnControl.class);
57
58 public RequestClientConnControl() {
59 super();
60 }
61
62 @Override
63 public void process(final HttpRequest request, final EntityDetails entity, final HttpContext context)
64 throws HttpException, IOException {
65 Args.notNull(request, "HTTP request");
66
67 final String method = request.getMethod();
68 if (method.equalsIgnoreCase("CONNECT")) {
69 return;
70 }
71
72 final HttpClientContext clientContext = HttpClientContext.adapt(context);
73
74
75 final RouteInfo route = clientContext.getHttpRoute();
76 if (route == null) {
77 LOG.debug("Connection route not set in the context");
78 return;
79 }
80
81 if (route.getHopCount() == 1 || route.isTunnelled()) {
82 if (!request.containsHeader(HttpHeaders.CONNECTION)) {
83 request.addHeader(HttpHeaders.CONNECTION, HeaderElements.KEEP_ALIVE);
84 }
85 }
86 }
87
88 }