com.litongjava.tio.http.server.util.Resps Maven / Gradle / Ivy
package com.litongjava.tio.http.server.util;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.PrintWriter;
import java.io.StringWriter;
import java.io.UnsupportedEncodingException;
import java.net.URLEncoder;
import java.nio.file.Files;
import java.util.Date;
import java.util.Map;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.litongjava.tio.http.common.HeaderName;
import com.litongjava.tio.http.common.HeaderValue;
import com.litongjava.tio.http.common.HttpConfig;
import com.litongjava.tio.http.common.HttpRequest;
import com.litongjava.tio.http.common.HttpResource;
import com.litongjava.tio.http.common.HttpResponse;
import com.litongjava.tio.http.common.HttpResponseStatus;
import com.litongjava.tio.http.common.MimeType;
import com.litongjava.tio.http.common.RequestHeaderKey;
import com.litongjava.tio.http.common.RequestLine;
import com.litongjava.tio.utils.IoUtils;
import com.litongjava.tio.utils.environment.EnvUtils;
import com.litongjava.tio.utils.hutool.ClassUtil;
import com.litongjava.tio.utils.hutool.FileUtil;
import com.litongjava.tio.utils.hutool.StrUtil;
import com.litongjava.tio.utils.json.Json;
/**
* @author tanyaowu
* 2017年6月29日 下午4:17:24
*/
public class Resps {
private static Logger log = LoggerFactory.getLogger(Resps.class);
private Resps() {
}
/**
* 构建css响应
* Content-Type: text/css;charset=utf-8
*
* @param request
* @param bodyString
* @return
* @author tanyaowu
*/
public static HttpResponse css(HttpRequest request, String bodyString) {
return css(request, bodyString, request.httpConfig.getCharset());
}
/**
* 构建css响应
* Content-Type: text/css;charset=utf-8
*
* @param request
* @param bodyString
* @param charset
* @return
* @author tanyaowu
*/
public static HttpResponse css(HttpRequest request, String bodyString, String charset) {
HttpResponse ret = string(request, bodyString, charset, getMimeTypeStr(MimeType.TEXT_CSS_CSS, charset));
return ret;
}
/**
* 根据byte[]创建响应
*
* @param request
* @param bodyBytes
* @param extension 后缀，可以为空
* @return
* @author tanyaowu
*/
public static HttpResponse bytes(HttpRequest request, byte[] bodyBytes, String extension) {
String contentType = null;
// String extension = FilenameUtils.getExtension(filename);
if (StrUtil.isNotBlank(extension)) {
MimeType mimeType = MimeType.fromExtension(extension);
if (mimeType != null) {
contentType = mimeType.getType();
} else {
contentType = "application/octet-stream";
}
}
return bytesWithContentType(request, bodyBytes, contentType);
}
/**
* @param response
* @param byteOne
* @param extension
* @return
*/
public static HttpResponse bytes(HttpResponse response, Byte byteOne, String extension) {
String contentType = null;
if (StrUtil.isNotBlank(extension)) {
MimeType mimeType = MimeType.fromExtension(extension);
if (mimeType != null) {
contentType = mimeType.getType();
} else {
contentType = "application/octet-stream";
}
}
return bytesWithContentType(response, byteOne, contentType);
}
/**
* @param response
* @param bodyBytes
* @param extension
* @return
*/
public static HttpResponse bytes(HttpResponse response, byte[] bodyBytes, String extension) {
String contentType = null;
// String extension = FilenameUtils.getExtension(filename);
if (StrUtil.isNotBlank(extension)) {
MimeType mimeType = MimeType.fromExtension(extension);
if (mimeType != null) {
contentType = mimeType.getType();
} else {
contentType = "application/octet-stream";
}
}
return bytesWithContentType(response, bodyBytes, contentType);
}
/**
* 根据文件创建响应
*
* @param request
* @param fileOnServer
* @return
* @throws IOException
* @author tanyaowu
*/
public static HttpResponse file(HttpRequest request, File fileOnServer) throws Exception {
if (fileOnServer == null || !fileOnServer.exists()) {
return request.httpConfig.getHttpRequestHandler().resp404(request, request.getRequestLine());
}
Date lastModified = new Date(fileOnServer.lastModified());
HttpResponse ret = try304(request, lastModified.getTime());
if (ret != null) {
return ret;
}
byte[] bodyBytes = Files.readAllBytes(fileOnServer.toPath());
String filename = fileOnServer.getName();
String extension = FileUtil.extName(filename);
ret = bytes(request, bodyBytes, extension);
// ret.addHeader(HeaderName.Content_Disposition, HeaderValue.from("attachment;filename=" + fileOnServer.getName()));
ret.setLastModified(HeaderValue.from(lastModified.getTime() + ""));
return ret;
}
/**
* @param request
* @param path
* @return
* @throws Exception
* @author tanyaowu
*/
public static HttpResponse file(HttpRequest request, String path) throws Exception {
HttpResource httpResource = request.httpConfig.getResource(request, path);
if (httpResource == null) {
return null;
} else {
path = httpResource.getPath();
File file = httpResource.getFile();
if (file != null) {
return file(request, file);
}
InputStream inputStream = httpResource.getInputStream();
byte[] bs = IoUtils.toByteArray(inputStream);
return Resps.bytes(request, bs, FileUtil.extName(path));
}
}
/**
*
* @param request
* @param path 文件在服务器上的相对pageRoot的路径，形如："/user/index.html"
* @param httpConfig
* @return
* @throws IOException
* @author: tanyaowu
*/
// public static HttpResponse file(HttpRequest request, String path, HttpConfig httpConfig) throws Exception {
// File file = httpConfig.getFile(request, path);//(pageRoot + path);
// if (!file.exists()) {
// return resp404(request, request.getRequestLine(), httpConfig);
// }
// return file(request, file);
// }
/**
* @param request
* @param requestLine
* @param httpConfig
* @return
* @throws Exception
* @author: tanyaowu
*/
public static HttpResponse resp404(HttpRequest request, RequestLine requestLine, HttpConfig httpConfig)
throws Exception {
String file404 = httpConfig.getPage404();
HttpResource httpResource = request.httpConfig.getResource(request, file404);
if (httpResource != null) {
file404 = httpResource.getPath();
HttpResponse ret = Resps.forward(request,
file404 + "?tio_initpath=" + URLEncoder.encode(requestLine.getPathAndQuery(), httpConfig.getCharset()));
return ret;
}
// File file = httpConfig.getFile(request, file404);// new File(pageRoot + file404);
// if (file.exists()) {
// HttpResponse ret = Resps.redirect(request, file404 + "?tio_initpath=" + requestLine.getPathAndQuery());
// return ret;
// }
HttpResponse ret = Resps.html(request, "404");
ret.setStatus(HttpResponseStatus.C404);
return ret;
}
/**
* @param request
* @return
* @throws Exception
*/
public static HttpResponse resp404(HttpRequest request) throws Exception {
return resp404(request, request.requestLine, request.httpConfig);
}
// /**
// *
// * @param newPath
// * @param request
// * @return
// * @throws Exception
// * @author tanyaowu
// */
// public static HttpResponse forward(String newPath, HttpRequest request) throws Exception {
// RequestLine requestLine = request.getRequestLine();
// HttpConfig httpConfig = request.getHttpConfig();
//
// httpConfig.getContextPath()
//
// if (newPath != null) {
// requestLine.setPath(newPath);
// }
//
// HttpRequestHandler httpRequestHandler = request.getHttpConfig().getHttpRequestHandler();
// return httpRequestHandler.handler(request);
// }
/**
*
* @param request
* @param requestLine
* @param httpConfig
* @param throwable
* @return
* @throws Exception
*/
public static HttpResponse resp500(HttpRequest request, RequestLine requestLine, HttpConfig httpConfig,
Throwable throwable) throws Exception {
String file500 = httpConfig.getPage500();
HttpResource httpResource = request.httpConfig.getResource(request, file500);
if (httpResource != null) {
file500 = httpResource.getPath();
HttpResponse ret = Resps.forward(request, file500 + "?tio_initpath=" + requestLine.getPathAndQuery());
return ret;
}
HttpResponse ret = null;
if (EnvUtils.getBoolean("http.response.showExceptionDetails", false)) {
// 获取完整的堆栈跟踪
StringWriter sw = new StringWriter();
PrintWriter pw = new PrintWriter(sw);
throwable.printStackTrace(pw);
ret = Resps.txt(request, sw.toString());
} else {
ret = Resps.html(request, "500");
}
ret.setStatus(HttpResponseStatus.C500);
return ret;
}
/**
* @param request
* @param throwable
* @return
* @throws Exception
*/
public static HttpResponse resp500(HttpRequest request, Throwable throwable) throws Exception {
return resp500(request, request.requestLine, request.httpConfig, throwable);
}
/**
* @param request
* @param bodyBytes
* @param contentType 形如:application/octet-stream等
* @return
* @author tanyaowu
*/
public static HttpResponse bytesWithContentType(HttpRequest request, byte[] bodyBytes, String contentType) {
HttpResponse ret = new HttpResponse(request);
ret.setBody(bodyBytes);
if (StrUtil.isBlank(contentType)) {
ret.addHeader(HeaderName.Content_Type, HeaderValue.Content_Type.DEFAULT_TYPE);
} else {
ret.addHeader(HeaderName.Content_Type, HeaderValue.Content_Type.from(contentType));
}
return ret;
}
/**
* @param response
* @param byteOne
* @param contentType
* @return
*/
public static HttpResponse bytesWithContentType(HttpResponse response, byte byteOne, String contentType) {
response.setBody(byteOne);
if (StrUtil.isBlank(contentType)) {
response.addHeader(HeaderName.Content_Type, HeaderValue.Content_Type.DEFAULT_TYPE);
} else {
response.addHeader(HeaderName.Content_Type, HeaderValue.Content_Type.from(contentType));
}
return response;
}
public static HttpResponse bytesWithContentType(HttpResponse response, byte[] bodyBytes, String contentType) {
response.setBody(bodyBytes);
if (StrUtil.isBlank(contentType)) {
response.addHeader(HeaderName.Content_Type, HeaderValue.Content_Type.DEFAULT_TYPE);
} else {
response.addHeader(HeaderName.Content_Type, HeaderValue.Content_Type.from(contentType));
}
return response;
}
/**
* @param request
* @param bodyBytes
* @param headers
* @return
* @author tanyaowu
*/
public static HttpResponse bytesWithHeaders(HttpRequest request, byte[] bodyBytes,
Map headers) {
HttpResponse ret = new HttpResponse(request);
ret.setBody(bodyBytes);
ret.addHeaders(headers);
return ret;
}
/**
* @param request
* @param bodyString
* @return
* @author tanyaowu
*/
public static HttpResponse html(HttpRequest request, String bodyString) {
return html(request, bodyString, request.httpConfig.getCharset());
}
/**
* @param response
* @param bodyString
* @return
*/
public static HttpResponse html(HttpResponse response, String bodyString) {
return html(response, bodyString, response.getHttpRequest().httpConfig.getCharset());
}
/**
* @param request
* @param newPath
* @return
* @throws Exception
*/
public static HttpResponse forward(HttpRequest request, String newPath) throws Exception {
return request.forward(newPath);
}
/**
* Content-Type: text/html;charset=utf-8
*
* @param request
* @param bodyString
* @param charset
* @return
* @author tanyaowu
*/
public static HttpResponse html(HttpRequest request, String bodyString, String charset) {
HttpResponse ret = string(request, bodyString, charset, getMimeTypeStr(MimeType.TEXT_HTML_HTML, charset));
return ret;
}
/**
* @param response
* @param bodyString
* @param charset
* @return
*/
public static HttpResponse html(HttpResponse response, String bodyString, String charset) {
return string(response, bodyString, charset, getMimeTypeStr(MimeType.TEXT_HTML_HTML, charset));
}
/**
* Content-Type: application/javascript;charset=utf-8
*
* @param request
* @param bodyString
* @return
* @author tanyaowu
*/
public static HttpResponse js(HttpRequest request, String bodyString) {
return js(request, bodyString, request.httpConfig.getCharset());
}
/**
* Content-Type: application/javascript;charset=utf-8
*
* @param request
* @param bodyString
* @param charset
* @return
* @author tanyaowu
*/
public static HttpResponse js(HttpRequest request, String bodyString, String charset) {
HttpResponse ret = string(request, bodyString, charset,
getMimeTypeStr(MimeType.APPLICATION_JAVASCRIPT_JS, charset));
return ret;
}
/**
* Content-Type: application/json;charset=utf-8
*
* @param request
* @param body
* @return
* @author tanyaowu
*/
public static HttpResponse json(HttpRequest request, Object body) {
return json(request, body, request.httpConfig.getCharset());
}
public static HttpResponse json(HttpResponse response, Object body) {
String charset = response.getHttpRequest().getHttpConfig().getCharset();
return json(response, body, charset);
}
/**
* Content-Type: application/json;charset=utf-8
*
* @param request
* @param body
* @param charset
* @return
* @author tanyaowu
*/
public static HttpResponse json(HttpRequest request, Object body, String charset) {
HttpResponse ret = null;
if (body == null) {
ret = string(request, "", charset, getMimeTypeStr(MimeType.TEXT_PLAIN_JSON, charset));
} else {
if (body.getClass() == String.class || ClassUtil.isBasicType(body.getClass())) {
ret = string(request, body + "", charset, getMimeTypeStr(MimeType.TEXT_PLAIN_JSON, charset));
} else {
ret = string(request, Json.getJson().toJson(body), charset, getMimeTypeStr(MimeType.TEXT_PLAIN_JSON, charset));
}
}
return ret;
}
public static HttpResponse json(HttpResponse response, Object body, String charset) {
if (body == null) {
response = string(response, "", charset, getMimeTypeStr(MimeType.TEXT_PLAIN_JSON, charset));
} else {
if (body.getClass() == String.class || ClassUtil.isBasicType(body.getClass())) {
response = string(response, body + "", charset, getMimeTypeStr(MimeType.TEXT_PLAIN_JSON, charset));
} else {
response = string(response, Json.getJson().toJson(body), charset,
getMimeTypeStr(MimeType.TEXT_PLAIN_JSON, charset));
}
}
return response;
}
private static String getMimeTypeStr(MimeType mimeType, String charset) {
if (charset == null) {
return mimeType.getType();
} else {
return mimeType.getType() + ";charset=" + charset;
}
}
/**
* 重定向
*
* @param request
* @param path
* @return
* @author tanyaowu
*/
public static HttpResponse redirect(HttpRequest request, String path) {
return redirect(request, path, HttpResponseStatus.C302);
}
/**
* 永久重定向
*
* @param request
* @param path
* @return
*/
public static HttpResponse redirectForever(HttpRequest request, String path) {
return redirect(request, path, HttpResponseStatus.C301);
}
/**
* @param request
* @param path
* @param status
* @return
*/
public static HttpResponse redirect(HttpRequest request, String path, HttpResponseStatus status) {
HttpResponse ret = new HttpResponse(request);
ret.setStatus(status);
ret.addHeader(HeaderName.Location, HeaderValue.from(path));
return ret;
}
/**
* 用页面重定向
*
* @param request
* @param path
* @return
* @author tanyaowu
*/
public static HttpResponse redirectWithPage(HttpRequest request, String path) {
StringBuilder sb = new StringBuilder(256);
sb.append("");
return Resps.html(request, sb.toString());
}
/**
* 创建字符串输出
*
* @param request
* @param bodyString
* @param Content_Type
* @return
* @author tanyaowu
*/
public static HttpResponse string(HttpRequest request, String bodyString, String Content_Type) {
return string(request, bodyString, request.httpConfig.getCharset(), Content_Type);
}
/**
* 创建字符串输出
*
* @param request
* @param bodyString
* @param charset
* @param mimeTypeStr
* @return
* @author tanyaowu
*/
public static HttpResponse string(HttpRequest request, String bodyString, String charset, String mimeTypeStr) {
HttpResponse ret = new HttpResponse(request);
//
// //处理jsonp
// String jsonp = request.getParam(request.httpConfig.getJsonpParamName());
// if (StrUtil.isNotBlank(jsonp)) {
// bodyString = jsonp + "(" + bodyString + ")";
// }
if (bodyString != null) {
if (charset == null) {
ret.setBody(bodyString.getBytes());
} else {
try {
ret.setBody(bodyString.getBytes(charset));
} catch (UnsupportedEncodingException e) {
log.error(e.toString(), e);
}
}
}
ret.addHeader(HeaderName.Content_Type, HeaderValue.Content_Type.from(mimeTypeStr));
return ret;
}
public static HttpResponse string(HttpResponse response, String bodyString, String charset, String mimeTypeStr) {
if (bodyString != null) {
if (charset == null) {
response.setBody(bodyString.getBytes());
} else {
try {
response.setBody(bodyString.getBytes(charset));
} catch (UnsupportedEncodingException e) {
log.error(e.toString(), e);
}
}
}
response.addHeader(HeaderName.Content_Type, HeaderValue.Content_Type.from(mimeTypeStr));
return response;
}
/**
* 尝试返回304，这个会new一个HttpResponse返回
*
* @param request
* @param lastModifiedOnServer 服务器中资源的lastModified
* @return
* @author tanyaowu
*/
public static HttpResponse try304(HttpRequest request, long lastModifiedOnServer) {
String If_Modified_Since = request.getHeader(RequestHeaderKey.If_Modified_Since);// If-Modified-Since
if (StrUtil.isNotBlank(If_Modified_Since)) {
Long If_Modified_Since_Date = null;
try {
If_Modified_Since_Date = Long.parseLong(If_Modified_Since);
if (lastModifiedOnServer <= If_Modified_Since_Date) {
HttpResponse ret = new HttpResponse(request);
ret.setStatus(HttpResponseStatus.C304);
return ret;
}
} catch (NumberFormatException e) {
log.warn("{}, {} is not an int，client:{}", request.getClientIp(), If_Modified_Since, request.getUserAgent());
return null;
}
}
return null;
}
/**
* Content-Type: text/plain;charset=utf-8
*
* @param request
* @param bodyString
* @return
* @author tanyaowu
*/
public static HttpResponse txt(HttpRequest request, String bodyString) {
return txt(request, bodyString, request.httpConfig.getCharset());
}
/**
* @param reqponse
* @param bodyString
* @return
*/
public static HttpResponse txt(HttpResponse reqponse, String bodyString) {
return txt(reqponse, bodyString, reqponse.getHttpRequest().getHttpConfig().getCharset());
}
/**
* Content-Type: text/plain;charset=utf-8
*
* @param request
* @param bodyString
* @param charset
* @return
* @author tanyaowu
*/
public static HttpResponse txt(HttpRequest request, String bodyString, String charset) {
HttpResponse ret = string(request, bodyString, charset, getMimeTypeStr(MimeType.TEXT_PLAIN_TXT, charset));
return ret;
}
public static HttpResponse txt(HttpResponse response, String bodyString, String charset) {
return string(response, bodyString, charset, getMimeTypeStr(MimeType.TEXT_PLAIN_TXT, charset));
}
} © 2015 - 2025 Weber Informatics LLC | Privacy Policy