Java Code Examples for org.springframework.http.MediaType#getCharset()

public void setContentType(@Nullable String contentType) {
	this.contentType = contentType;
	if (contentType != null) {
		try {
			MediaType mediaType = MediaType.parseMediaType(contentType);
			if (mediaType.getCharset() != null) {
				this.characterEncoding = mediaType.getCharset().name();
			}
		}
		catch (IllegalArgumentException ex) {
			// Try to get charset value anyway
			int charsetIndex = contentType.toLowerCase().indexOf(CHARSET_PREFIX);
			if (charsetIndex != -1) {
				this.characterEncoding = contentType.substring(charsetIndex + CHARSET_PREFIX.length());
			}
		}
		updateContentTypeHeader();
	}
}
 

@Override
protected void applyHeaders() {
	MediaType contentType = getHeaders().getContentType();
	HttpServletResponse response = getNativeResponse();
	if (response.getContentType() == null && contentType != null) {
		response.setContentType(contentType.toString());
	}
	Charset charset = (contentType != null ? contentType.getCharset() : null);
	if (response.getCharacterEncoding() == null && charset != null) {
		response.setCharacterEncoding(charset.name());
	}
	long contentLength = getHeaders().getContentLength();
	if (contentLength != -1) {
		response.setContentLengthLong(contentLength);
	}
}
 

@Override
protected Properties readInternal(Class<? extends Properties> clazz, HttpInputMessage inputMessage) throws IOException, HttpMessageNotReadableException {
    Properties properties = new Properties();
    // 获取请求头
    HttpHeaders headers = inputMessage.getHeaders();
    // 获取 content-type
    MediaType contentType = headers.getContentType();
    // 获取编码
    Charset charset = null;
    if (contentType != null) {
        charset = contentType.getCharset();
    }

    charset = charset == null ? Charset.forName("UTF-8") : charset;

    // 获取请求体
    InputStream body = inputMessage.getBody();
    InputStreamReader inputStreamReader = new InputStreamReader(body, charset);

    properties.load(inputStreamReader);
    return properties;
}
 

@Override
@SuppressWarnings("unchecked")
protected T readInternal(Class<? extends T> clazz, HttpInputMessage inputMessage)
		throws IOException, HttpMessageNotReadableException {

	WireFeedInput feedInput = new WireFeedInput();
	MediaType contentType = inputMessage.getHeaders().getContentType();
	Charset charset = (contentType != null && contentType.getCharset() != null ?
			contentType.getCharset() : DEFAULT_CHARSET);
	try {
		Reader reader = new InputStreamReader(inputMessage.getBody(), charset);
		return (T) feedInput.build(reader);
	}
	catch (FeedException ex) {
		throw new HttpMessageNotReadableException("Could not read WireFeed: " + ex.getMessage(), ex);
	}
}
 

@Override
protected void applyHeaders() {
	HttpServletResponse response = getNativeResponse();
	MediaType contentType = getHeaders().getContentType();
	if (response.getContentType() == null && contentType != null) {
		response.setContentType(contentType.toString());
	}
	Charset charset = (contentType != null ? contentType.getCharset() : null);
	if (response.getCharacterEncoding() == null && charset != null) {
		response.setCharacterEncoding(charset.name());
	}
	long contentLength = getHeaders().getContentLength();
	if (contentLength != -1) {
		response.setContentLengthLong(contentLength);
	}
}
 

@Override
public Mono<Void> write(Publisher<? extends MultiValueMap<String, String>> inputStream,
		ResolvableType elementType, @Nullable MediaType mediaType, ReactiveHttpOutputMessage message,
		Map<String, Object> hints) {

	mediaType = getMediaType(mediaType);
	message.getHeaders().setContentType(mediaType);

	Charset charset = mediaType.getCharset();
	Assert.notNull(charset, "No charset"); // should never occur

	return Mono.from(inputStream).flatMap(form -> {
		logFormData(form, hints);
		String value = serializeForm(form, charset);
		ByteBuffer byteBuffer = charset.encode(value);
		DataBuffer buffer = message.bufferFactory().wrap(byteBuffer);
		message.getHeaders().setContentLength(byteBuffer.remaining());
		return message.writeWith(Mono.just(buffer));
	});
}
 

@Override
public void setContentType(@Nullable String contentType) {
	this.contentType = contentType;
	if (contentType != null) {
		try {
			MediaType mediaType = MediaType.parseMediaType(contentType);
			if (mediaType.getCharset() != null) {
				this.characterEncoding = mediaType.getCharset().name();
				this.charset = true;
			}
		}
		catch (Exception ex) {
			// Try to get charset value anyway
			int charsetIndex = contentType.toLowerCase().indexOf(CHARSET_PREFIX);
			if (charsetIndex != -1) {
				this.characterEncoding = contentType.substring(charsetIndex + CHARSET_PREFIX.length());
				this.charset = true;
			}
		}
		updateContentTypeHeader();
	}
}
 

public void setContentType(@Nullable String contentType) {
	this.contentType = contentType;
	if (contentType != null) {
		try {
			MediaType mediaType = MediaType.parseMediaType(contentType);
			if (mediaType.getCharset() != null) {
				this.characterEncoding = mediaType.getCharset().name();
			}
		}
		catch (IllegalArgumentException ex) {
			// Try to get charset value anyway
			int charsetIndex = contentType.toLowerCase().indexOf(CHARSET_PREFIX);
			if (charsetIndex != -1) {
				this.characterEncoding = contentType.substring(charsetIndex + CHARSET_PREFIX.length());
			}
		}
		updateContentTypeHeader();
	}
}
 

private Charset determineCharset() {
	MediaType contentType = getHeaders().getContentType();
	if (contentType != null) {
		Charset charset = contentType.getCharset();
		if (charset != null) {
			return charset;
		}
	}
	String encoding = this.multipartRequest.getCharacterEncoding();
	return (encoding != null ? Charset.forName(encoding) : FORM_CHARSET);
}
 

private MediaType getMediaType(@Nullable MediaType mediaType) {
	if (mediaType == null) {
		return DEFAULT_FORM_DATA_MEDIA_TYPE;
	}
	else if (mediaType.getCharset() == null) {
		return new MediaType(mediaType, getDefaultCharset());
	}
	else {
		return mediaType;
	}
}
 

private String determineEncoding(String contentTypeHeader, String defaultEncoding) {
	if (!StringUtils.hasText(contentTypeHeader)) {
		return defaultEncoding;
	}
	MediaType contentType = MediaType.parseMediaType(contentTypeHeader);
	Charset charset = contentType.getCharset();
	return (charset != null ? charset.name() : defaultEncoding);
}
 

@Override
protected Message readInternal(Class<? extends Message> clazz, HttpInputMessage inputMessage)
		throws IOException, HttpMessageNotReadableException {

	MediaType contentType = inputMessage.getHeaders().getContentType();
	if (contentType == null) {
		contentType = PROTOBUF;
	}
	Charset charset = contentType.getCharset();
	if (charset == null) {
		charset = DEFAULT_CHARSET;
	}

	Message.Builder builder = getMessageBuilder(clazz);
	if (PROTOBUF.isCompatibleWith(contentType)) {
		builder.mergeFrom(inputMessage.getBody(), this.extensionRegistry);
	}
	else if (TEXT_PLAIN.isCompatibleWith(contentType)) {
		InputStreamReader reader = new InputStreamReader(inputMessage.getBody(), charset);
		TextFormat.merge(reader, this.extensionRegistry, builder);
	}
	else if (this.protobufFormatSupport != null) {
		this.protobufFormatSupport.merge(
				inputMessage.getBody(), charset, contentType, this.extensionRegistry, builder);
	}
	return builder.build();
}
 

private static HttpHeaders initHeaders(HttpHeaders headers, HttpServletRequest request) {
	MediaType contentType = headers.getContentType();
	if (contentType == null) {
		String requestContentType = request.getContentType();
		if (StringUtils.hasLength(requestContentType)) {
			contentType = MediaType.parseMediaType(requestContentType);
			headers.setContentType(contentType);
		}
	}
	if (contentType != null && contentType.getCharset() == null) {
		String encoding = request.getCharacterEncoding();
		if (StringUtils.hasLength(encoding)) {
			Charset charset = Charset.forName(encoding);
			Map<String, String> params = new LinkedCaseInsensitiveMap<>();
			params.putAll(contentType.getParameters());
			params.put("charset", charset.toString());
			headers.setContentType(
					new MediaType(contentType.getType(), contentType.getSubtype(),
							params));
		}
	}
	if (headers.getContentLength() == -1) {
		int contentLength = request.getContentLength();
		if (contentLength != -1) {
			headers.setContentLength(contentLength);
		}
	}
	return headers;
}
 

private Charset getCharset() {
	Charset charset = null;
	MediaType contentType = getHeaders().getContentType();
	if (contentType != null) {
		charset = contentType.getCharset();
	}
	return (charset != null ? charset : StandardCharsets.UTF_8);
}
 

private MediaType getMediaType(@Nullable MediaType mediaType) {
	if (mediaType == null) {
		return DEFAULT_FORM_DATA_MEDIA_TYPE;
	}
	else if (mediaType.getCharset() == null) {
		return new MediaType(mediaType, this.charset);
	}
	else {
		return mediaType;
	}
}
 

/**
 * Determine the JSON encoding to use for the given content type.
 *
 * @param contentType the media type as requested by the caller
 * @return the JSON encoding to use (never <code>null</code>)
 */
private JsonEncoding getJsonEncoding(MediaType contentType) {
	if (contentType != null && contentType.getCharset() != null) {
		Charset charset = contentType.getCharset();
		for (JsonEncoding encoding : JsonEncoding.values()) {
			if (charset.name().equals(encoding.getJavaName())) {
				return encoding;
			}
		}
	}
	return JsonEncoding.UTF8;
}
 

/**
 * Determine the charset of the response (for inclusion in a status exception).
 * @param response the response to inspect
 * @return the associated charset, or {@code null} if none
 * @since 4.3.8
 */
@Nullable
protected Charset getCharset(ClientHttpResponse response) {
	HttpHeaders headers = response.getHeaders();
	MediaType contentType = headers.getContentType();
	return (contentType != null ? contentType.getCharset() : null);
}
 

private String determineEncoding(String contentTypeHeader, String defaultEncoding) {
	if (!StringUtils.hasText(contentTypeHeader)) {
		return defaultEncoding;
	}
	MediaType contentType = MediaType.parseMediaType(contentTypeHeader);
	Charset charset = contentType.getCharset();
	return (charset != null ? charset.name() : defaultEncoding);
}
 

private void setCharset(MediaType contentType, Marshaller marshaller) throws PropertyException {
    if (contentType != null && contentType.getCharset() != null) {
        marshaller.setProperty(Marshaller.JAXB_ENCODING, contentType.getCharset().name());
    }
}
 

private void setCharset(@Nullable MediaType contentType, Marshaller marshaller) throws PropertyException {
	if (contentType != null && contentType.getCharset() != null) {
		marshaller.setProperty(Marshaller.JAXB_ENCODING, contentType.getCharset().name());
	}
}