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.core5.http2.impl.nio;
29
30 import org.apache.hc.core5.annotation.Contract;
31 import org.apache.hc.core5.annotation.Internal;
32 import org.apache.hc.core5.annotation.ThreadingBehavior;
33 import org.apache.hc.core5.http.HttpHost;
34 import org.apache.hc.core5.http.URIScheme;
35 import org.apache.hc.core5.http.impl.nio.ClientHttp1IOEventHandler;
36 import org.apache.hc.core5.http.impl.nio.ClientHttp1StreamDuplexerFactory;
37 import org.apache.hc.core5.http.impl.nio.HttpConnectionEventHandler;
38 import org.apache.hc.core5.http.nio.ssl.TlsStrategy;
39 import org.apache.hc.core5.http2.HttpVersionPolicy;
40 import org.apache.hc.core5.reactor.EndpointParameters;
41 import org.apache.hc.core5.reactor.IOEventHandlerFactory;
42 import org.apache.hc.core5.reactor.ProtocolIOSession;
43 import org.apache.hc.core5.util.Args;
44 import org.apache.hc.core5.util.Asserts;
45 import org.apache.hc.core5.util.Timeout;
46
47
48
49
50
51
52
53
54 @Contract(threading = ThreadingBehavior.IMMUTABLE_CONDITIONAL)
55 @Internal
56 public class ClientHttpProtocolNegotiationStarter implements IOEventHandlerFactory {
57
58 private final ClientHttp1StreamDuplexerFactory http1StreamHandlerFactory;
59 private final ClientH2StreamMultiplexerFactory http2StreamHandlerFactory;
60 private final HttpVersionPolicy versionPolicy;
61 private final TlsStrategy tlsStrategy;
62 private final Timeout handshakeTimeout;
63
64 public ClientHttpProtocolNegotiationStarter(
65 final ClientHttp1StreamDuplexerFactory http1StreamHandlerFactory,
66 final ClientH2StreamMultiplexerFactory http2StreamHandlerFactory,
67 final HttpVersionPolicy versionPolicy,
68 final TlsStrategy tlsStrategy,
69 final Timeout handshakeTimeout) {
70 this.http1StreamHandlerFactory = Args.notNull(http1StreamHandlerFactory, "HTTP/1.1 stream handler factory");
71 this.http2StreamHandlerFactory = Args.notNull(http2StreamHandlerFactory, "HTTP/2 stream handler factory");
72 this.versionPolicy = versionPolicy != null ? versionPolicy : HttpVersionPolicy.NEGOTIATE;
73 this.tlsStrategy = tlsStrategy;
74 this.handshakeTimeout = handshakeTimeout;
75 }
76
77 @Override
78 public HttpConnectionEventHandler createHandler(final ProtocolIOSession ioSession, final Object attachment) {
79 HttpVersionPolicy endpointPolicy = versionPolicy;
80 if (attachment instanceof EndpointParameters) {
81 final EndpointParameters params = (EndpointParameters) attachment;
82 if (URIScheme.HTTPS.same(params.getScheme())) {
83 Asserts.notNull(tlsStrategy, "TLS strategy");
84 final HttpHost host = new HttpHost(params.getScheme(), params.getHostName(), params.getPort());
85 tlsStrategy.upgrade(
86 ioSession,
87 host,
88 ioSession.getLocalAddress(),
89 ioSession.getRemoteAddress(),
90 params.getAttachment(),
91 handshakeTimeout);
92 }
93 if (params.getAttachment() instanceof HttpVersionPolicy) {
94 endpointPolicy = (HttpVersionPolicy) params.getAttachment();
95 }
96 }
97 switch (endpointPolicy) {
98 case FORCE_HTTP_2:
99 return new H2OnlyClientProtocolNegotiator(ioSession, http2StreamHandlerFactory, false);
100 case FORCE_HTTP_1:
101 return new ClientHttp1IOEventHandler(http1StreamHandlerFactory.create(ioSession));
102 default:
103 return new ClientHttpProtocolNegotiator(ioSession, http1StreamHandlerFactory, http2StreamHandlerFactory, HttpVersionPolicy.NEGOTIATE);
104 }
105 }
106
107 }