encoding 설정 지원

This commit is contained in:
현성필
2024-12-06 16:28:09 +09:00
parent 50d31ebe46
commit 4c6689db83
@@ -1,18 +1,39 @@
package com.eactive.testmaster.api.dto;
import io.netty.util.CharsetUtil;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.net.URLDecoder;
import java.nio.charset.Charset;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import java.util.stream.Collectors;
import javax.servlet.http.HttpServletRequest;
import org.apache.commons.lang3.StringUtils;
public class MockRequest {
private static final Charset DEFAULT_CHARSET = determineDefaultCharset();
private static Charset determineDefaultCharset() {
try {
String systemEncoding = System.getProperty("file.encoding");
if (StringUtils.isNotBlank(systemEncoding)) {
return Charset.forName(systemEncoding);
}
} catch (IllegalArgumentException | SecurityException e) {
// Log warning about fallback to UTF-8
}
return CharsetUtil.UTF_8;
}
public static Map<String, String> parseQueryString(HttpServletRequest request) {
Map<String, String> queryParams = new HashMap<>();
String queryString = request.getQueryString()!=null?URLDecoder.decode(request.getQueryString()) : null;
String queryString = request.getQueryString() != null ? URLDecoder.decode(request.getQueryString()) : null;
if (queryString != null) {
String[] queryParamsArr = queryString.split("&");
for (String queryParam : queryParamsArr) {
@@ -28,6 +49,26 @@ public class MockRequest {
return queryParams;
}
public static final Pattern CHARSET_PATTERN = Pattern.compile("charset=([^;]+)");
private Charset extractCharset(Map<String, String> headers) {
String contentType = headers.get("content-type");
if (contentType != null) {
Matcher matcher = CHARSET_PATTERN.matcher(contentType.toLowerCase());
if (matcher.find()) {
try {
return Charset.forName(matcher.group(1).trim());
} catch (IllegalArgumentException e) {
// Log warning about unsupported charset
return DEFAULT_CHARSET;
}
}
}
return DEFAULT_CHARSET;
}
public MockRequest(HttpServletRequest request) throws IOException {
String contextPath = request.getContextPath();
this.method = request.getMethod();
@@ -37,7 +78,21 @@ public class MockRequest {
this.params = parseQueryString(request);
this.body = URLDecoder.decode(request.getReader().lines().collect(Collectors.joining(System.lineSeparator())));
String charset = extractCharset(headers).toString();
try (BufferedReader reader = new BufferedReader(
new InputStreamReader(request.getInputStream(), charset))) {
String rawBody = reader.lines()
.collect(Collectors.joining(System.lineSeparator()));
// Only URL decode if the content type indicates form submission
String contentType = request.getContentType();
if (contentType != null) {
this.body = URLDecoder.decode(rawBody, charset);
} else {
this.body = rawBody;
}
}
}
private String uri;