zhouxy108/hutool
1
0
Fork 0
mirror of https://gitee.com/chinabugotech/hutool.git synced 2025-07-21 15:09:48 +08:00
Code Issues Packages Projects Releases Wiki Activity

add ValidateObjectInputStream

Browse Source
This commit is contained in:
Looly
2020-04-03 12:13:56 +08:00
parent f2b44605d0
commit 3921a568dd
8 changed files with 299 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
hutool-http/src/main/java/cn/hutool/http/server/HttpServerRequest.java
View File

@@ -20,6 +20,7 @@ import java.net.HttpCookie;
import java.net.URI;
import java.nio.charset.Charset;
import java.util.Collection;
import java.util.Collections;
import java.util.Map;
/**
@@ -28,7 +29,7 @@ import java.util.Map;
* @author looly
* @since 5.2.6
*/
public class HttpServerRequest extends HttpServerBase{
public class HttpServerRequest extends HttpServerBase {
private Map<String, HttpCookie> cookieCache;
@@ -148,6 +149,21 @@ public class HttpServerRequest extends HttpServerBase{
return getHeader(Header.USER_AGENT);
}
/**
* 获取编码，获取失败默认使用UTF-8，获取规则如下：
*
* <pre>
* 1、从Content-Type头中获取编码，类似于：text/html;charset=utf-8
* </pre>
*
* @return 编码，默认UTF-8
*/
public Charset getCharset() {
final String contentType = getContentType();
final String charsetStr = HttpUtil.getCharset(contentType);
return CharsetUtil.parse(charsetStr, CharsetUtil.CHARSET_UTF_8);
}
/**
* 获得User-Agent
*
@@ -191,10 +207,10 @@ public class HttpServerRequest extends HttpServerBase{
*/
public Map<String, HttpCookie> getCookieMap() {
if (null == this.cookieCache) {
cookieCache = CollUtil.toMap(
cookieCache = Collections.unmodifiableMap(CollUtil.toMap(
NetUtil.parseCookies(getCookiesStr()),
new CaseInsensitiveMap<>(),
HttpCookie::getName);
HttpCookie::getName));
}
return cookieCache;
}
@@ -220,16 +236,12 @@ public class HttpServerRequest extends HttpServerBase{
/**
* 获取请求体文本，可以是form表单、json、xml等任意内容<br>
* 根据请求的Content-Type判断编码，判断失败使用UTF-8编码
* 使用{@link #getCharset()}判断编码，判断失败使用UTF-8编码
*
* @return 请求
*/
public String getBody() {
final String contentType = getContentType();
final String charsetStr = HttpUtil.getCharset(contentType);
final Charset charset = CharsetUtil.parse(charsetStr, CharsetUtil.CHARSET_UTF_8);
return getBody(charset);
return getBody(getCharset());
}
/**

165
hutool-http/src/main/java/cn/hutool/http/server/HttpServerResponse.java
View File

@@ -1,18 +1,33 @@
package cn.hutool.http.server;
import cn.hutool.core.io.FileUtil;
import cn.hutool.core.io.IORuntimeException;
import cn.hutool.core.io.IoUtil;
import cn.hutool.core.util.CharsetUtil;
import cn.hutool.core.util.ObjectUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.core.util.URLUtil;
import cn.hutool.http.Header;
import cn.hutool.http.HttpUtil;
import com.sun.net.httpserver.Headers;
import com.sun.net.httpserver.HttpExchange;
import java.io.BufferedInputStream;
import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.nio.charset.Charset;
import java.util.List;
import java.util.Map;
/**
* Http响应对象，用于写出数据到客户端
*/
public class HttpServerResponse extends HttpServerBase{
public class HttpServerResponse extends HttpServerBase {
private Charset charset;
/**
* 构造
@@ -49,6 +64,111 @@ public class HttpServerResponse extends HttpServerBase{
return this;
}
/**
* 获得所有响应头，获取后可以添加新的响应头
*
* @return 响应头
*/
public Headers getHeaders() {
return this.httpExchange.getResponseHeaders();
}
/**
* 添加响应头，如果已经存在，则追加
*
* @param header 头key
* @param value 值
* @return this
*/
public HttpServerResponse addHeader(String header, String value) {
getHeaders().add(header, value);
return this;
}
/**
* 设置响应头，如果已经存在，则覆盖
*
* @param header 头key
* @param value 值
* @return this
*/
public HttpServerResponse setHeader(Header header, String value) {
return setHeader(header.getValue(), value);
}
/**
* 设置响应头，如果已经存在，则覆盖
*
* @param header 头key
* @param value 值
* @return this
*/
public HttpServerResponse setHeader(String header, String value) {
getHeaders().set(header, value);
return this;
}
/**
* 设置响应头，如果已经存在，则覆盖
*
* @param header 头key
* @param value 值列表
* @return this
*/
public HttpServerResponse setHeader(String header, List<String> value) {
getHeaders().put(header, value);
return this;
}
/**
* 设置所有响应头，如果已经存在，则覆盖
*
* @param headers 响应头map
* @return this
*/
public HttpServerResponse setHeaders(Map<String, List<String>> headers) {
getHeaders().putAll(headers);
return this;
}
/**
* 设置Content-Type头，类似于:text/html;charset=utf-8<br>
* 如果用户传入的信息无charset信息，自动根据charset补充，charset设置见{@link #setCharset(Charset)}
*
* @param contentType Content-Type头内容
* @return this
*/
public HttpServerResponse setContentType(String contentType) {
if (null != contentType && null != this.charset) {
if (false == contentType.contains(";charset=")) {
contentType += ";charset=" + this.charset;
}
}
return setHeader(Header.CONTENT_TYPE, contentType);
}
/**
* 设置Content-Length头
*
* @param contentLength Content-Length头内容
* @return this
*/
public HttpServerResponse setContentLength(long contentLength) {
return setHeader(Header.CONTENT_LENGTH, String.valueOf(contentLength));
}
/**
* 设置响应的编码
*
* @param charset 编码
* @return this
*/
public HttpServerResponse setCharset(Charset charset) {
this.charset = charset;
return this;
}
/**
* 获取响应数据流
*
@@ -58,6 +178,15 @@ public class HttpServerResponse extends HttpServerBase{
return this.httpExchange.getResponseBody();
}
/**
* 获取响应数据流
*
* @return 响应数据流
*/
public OutputStream getWriter() {
return this.httpExchange.getResponseBody();
}
/**
* 写出数据到客户端
*
@@ -86,4 +215,38 @@ public class HttpServerResponse extends HttpServerBase{
}
return this;
}
/**
* 返回文件给客户端（文件下载）
*
* @param file 写出的文件对象
* @since 5.2.6
*/
public HttpServerResponse write(File file) {
final String fileName = file.getName();
final String contentType = ObjectUtil.defaultIfNull(HttpUtil.getMimeType(fileName), "application/octet-stream");
BufferedInputStream in = null;
try {
in = FileUtil.getInputStream(file);
write(in, contentType, fileName);
} finally {
IoUtil.close(in);
}
return this;
}
/**
* 返回数据给客户端
*
* @param in 需要返回客户端的内容
* @param contentType 返回的类型
* @param fileName 文件名
* @since 5.2.6
*/
public void write(InputStream in, String contentType, String fileName) {
final Charset charset = ObjectUtil.defaultIfNull(this.charset, CharsetUtil.CHARSET_UTF_8);
setHeader("Content-Disposition", StrUtil.format("attachment;filename={}", URLUtil.encode(fileName, charset)));
setContentType(contentType);
write(in);
}
}