yeyinlaoying/neutrino-proxy
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

针对静态资源增加缓存、gzip压缩处理

Browse Source
This commit is contained in:
aoshiguchen
2023-02-15 11:08:07 +08:00
parent 90487d5540
commit fce65d1dfe
1 changed files with 49 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
neutrino-core/src/main/java/fun/asgc/neutrino/core/web/HttpRequestHandler.java
+49 -5
View File
@@ -44,16 +44,17 @@ import io.netty.channel.ChannelHandlerContext;
import io.netty.handler.codec.http.FullHttpRequest;
import io.netty.handler.codec.http.HttpHeaderNames;
import io.netty.handler.codec.http.HttpResponseStatus;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.extern.slf4j.Slf4j;
import java.io.ByteArrayOutputStream;
import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.lang.reflect.Parameter;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.Set;
import java.util.*;
import java.util.zip.GZIPOutputStream;
/**
*
@@ -69,6 +70,10 @@ public class HttpRequestHandler {
@Autowired
private DefaultHttpRouter defaultHttpRouter;
private PathMatcher pathMatcher = new AntPathMatcher();
/**
* 静态自愿缓存
*/
private Map<String, DataWrapper> staticResourceCache = new ConcurrentReferenceHashMap<>(16, 0.75F, 1, ConcurrentReferenceHashMap.ReferenceType.WEAK);
public void handle() {
ChannelHandlerContext context = HttpContextHolder.getChannelHandlerContext();
@@ -132,7 +137,7 @@ public class HttpRequestHandler {
postHandle(httpRouteResult.getPageRoute(), null, null);
HttpResponseWrapper httpResponseWrapper = HttpContextHolder.getHttpResponseWrapper();
httpResponseWrapper.setContent(Unpooled.wrappedBuffer(FileUtil.readBytes(httpRouteResult.getPageLocation())));
httpResponseWrapper.setContent(Unpooled.wrappedBuffer(getStaticResourceData(httpRouteResult.getPageLocation(), mimeType)));
httpResponseWrapper.headers().add(HttpHeaderNames.CONTENT_TYPE, mimeType);
httpResponseWrapper.headers().add(HttpHeaderNames.CONTENT_LANGUAGE, "zh-CN");
httpResponseWrapper.headers().add(HttpHeaderNames.SERVER, MetaDataConstant.SERVER_VS);
@@ -159,6 +164,38 @@ public class HttpRequestHandler {
}
}
private byte[] getStaticResourceData(String pageLocation, String mimeType) {
DataWrapper dataWrapper = staticResourceCache.get(pageLocation);
if (null != dataWrapper) {
if (!StringUtil.isEmpty(dataWrapper.getContentEncoding())) {
HttpContextHolder.getHttpResponseWrapper().headers().add("Content-Encoding", dataWrapper.getContentEncoding());
}
return dataWrapper.getData();
}
byte[] res = FileUtil.readBytes(pageLocation);
if (null == res) {
res = new byte[0];
staticResourceCache.put(pageLocation, new DataWrapper(res, null));
return res;
}
staticResourceCache.put(pageLocation, new DataWrapper(res, null));
if ((mimeType.startsWith("text/") || mimeType.endsWith("javascript")) && res.length > 2048) {
// 文本，且大于2kb才做gzip压缩
ByteArrayOutputStream baos = new ByteArrayOutputStream();
try (GZIPOutputStream gZipOs = new GZIPOutputStream(baos)){
gZipOs.write(res);
gZipOs.flush();
gZipOs.finish();
res = baos.toByteArray();
staticResourceCache.put(pageLocation, new DataWrapper(res, "gzip"));
HttpContextHolder.getHttpResponseWrapper().headers().add("Content-Encoding", "gzip");
} catch (Exception e) {
return null;
}
}
return res;
}
private boolean preHandle(String route, Method targetMethod) throws Exception {
if (!CollectionUtil.isEmpty(HttpContextHolder.getInterceptorList())) {
for (HandlerInterceptor handlerInterceptor : HttpContextHolder.getInterceptorList()) {
@@ -289,4 +326,11 @@ public class HttpRequestHandler {
private void release() {
HttpContextHolder.remove();
}
@AllArgsConstructor
@Data
public static class DataWrapper {
private byte[] data;
private String contentEncoding;
}
}