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 package org.apache.hc.client5.http.async.methods;
28
29 import java.io.IOException;
30 import java.io.UnsupportedEncodingException;
31 import java.nio.charset.Charset;
32 import java.nio.charset.StandardCharsets;
33 import java.nio.charset.UnsupportedCharsetException;
34
35 import org.apache.hc.core5.http.ContentType;
36 import org.apache.hc.core5.http.EntityDetails;
37 import org.apache.hc.core5.http.HttpException;
38 import org.apache.hc.core5.http.HttpRequest;
39 import org.apache.hc.core5.http.HttpResponse;
40 import org.apache.hc.core5.http.config.CharCodingConfig;
41 import org.apache.hc.core5.http.nio.AsyncPushConsumer;
42 import org.apache.hc.core5.http.nio.entity.AbstractCharDataConsumer;
43 import org.apache.hc.core5.http.protocol.HttpContext;
44
45
46
47
48
49
50 public abstract class AbstractCharPushConsumer extends AbstractCharDataConsumer implements AsyncPushConsumer {
51
52 private final Charset defaultCharset;
53
54 public AbstractCharPushConsumer() {
55 this.defaultCharset = StandardCharsets.UTF_8;
56 }
57
58 protected AbstractCharPushConsumer(final int bufSize,
59 final CharCodingConfig charCodingConfig) {
60 super(bufSize, charCodingConfig);
61 this.defaultCharset = charCodingConfig != null && charCodingConfig.getCharset() != null
62 ? charCodingConfig.getCharset() : StandardCharsets.UTF_8;
63 }
64
65
66
67
68
69
70
71
72 protected abstract void start(HttpRequest promise, HttpResponse response, ContentType contentType) throws HttpException, IOException;
73
74 @Override
75 public final void consumePromise(
76 final HttpRequest promise,
77 final HttpResponse response,
78 final EntityDetails entityDetails,
79 final HttpContext context) throws HttpException, IOException {
80 if (entityDetails != null) {
81 final ContentType contentType;
82 try {
83 contentType = ContentType.parse(entityDetails.getContentType());
84 } catch (final UnsupportedCharsetException ex) {
85 throw new UnsupportedEncodingException(ex.getMessage());
86 }
87 Charset charset = contentType != null ? contentType.getCharset() : null;
88 if (charset == null) {
89 charset = defaultCharset;
90 }
91 setCharset(charset);
92 start(promise, response, contentType != null ? contentType : ContentType.DEFAULT_TEXT);
93 } else {
94 start(promise, response, null);
95 completed();
96 }
97 }
98
99 @Override
100 public void failed(final Exception cause) {
101 }
102
103 }