总监:咳咳咳…小王啊,最近挺清闲啊!
我:没有没有,还有好多bug没改呢。
总监:有个需求需要你弄一下。
我:好的好的。(摸鱼被抓不敢反抗啊!!!)
总监:现在的数据响应我觉得还可以提升一下,你把所有的响应在响应前都压缩一遍。还有服务之间feign调用的形式。
我:好的好的,现在就干。
总监:晚上给我个demo!
一:GateWay网关实现数据压缩
思路:
- 要网关吐出压缩后的数据,则要实现一个网关拦截器,在最后一层的拦截器当中修改响应的数据。
- 查看浏览器请求头中的Accept-Encoding: gzip, deflate一项,这里使用gzip压缩。
好啦,既然没有那么复杂,就开始干吧。
import cn.hutool.core.util.StrUtil;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import lombok.extern.log4j.Log4j2;
import org.reactivestreams.Publisher;
import org.springframework.cloud.gateway.filter.GatewayFilterChain;
import org.springframework.cloud.gateway.filter.GlobalFilter;
import org.springframework.cloud.gateway.filter.NettyWriteResponseFilter;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.Ordered;
import org.springframework.core.io.buffer.DataBuffer;
import org.springframework.core.io.buffer.DataBufferFactory;
import org.springframework.core.io.buffer.DataBufferUtils;
import org.springframework.core.io.buffer.DefaultDataBufferFactory;
import org.springframework.http.HttpStatus;
import org.springframework.http.server.reactive.ServerHttpRequest;
import org.springframework.http.server.reactive.ServerHttpResponse;
import org.springframework.http.server.reactive.ServerHttpResponseDecorator;
import org.springframework.web.server.ServerWebExchange;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.nio.charset.Charset;
import java.util.zip.GZIPOutputStream;
@Configuration
@Log4j2
public class ResponseFilter implements GlobalFilter, Ordered {
/**
* 约定的压缩格式
*/
private final static String GZIP = "gzip";
@Override
public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
//获取响应体
ServerHttpResponse originalResponse = exchange.getResponse();
//获取请求体
ServerHttpRequest request = exchange.getRequest();
DataBufferFactory bufferFactory = originalResponse.bufferFactory();
ServerHttpResponseDecorator decoratedResponse = new ServerHttpResponseDecorator(originalResponse) {
@Override
public Mono<Void> writeWith(Publisher<? extends DataBuffer> body) {
if (body instanceof Flux) {
Flux<? extends DataBuffer> fluxBody = (Flux<? extends DataBuffer>) body;
return super.writeWith(fluxBody.buffer().map(dataBuffers -> {
DataBufferFactory dataBufferFactory = new DefaultDataBufferFactory();
DataBuffer join = dataBufferFactory.join(dataBuffers);
byte[] content = new byte[join.readableByteCount()];
join.read(content);
// 释放掉内存
DataBufferUtils.release(join);
String responseString = new String(content, Charset.forName("UTF-8"));
//判断该浏览器是否支持gzip解码,如果支持gzip解码,则进行压缩
String acceptEncoding = request.getHeaders().getFirst("Accept-Encoding");
if(!StrUtil.hasBlank(acceptEncoding)){
assert acceptEncoding != null;
//是否支持压缩
if(acceptEncoding.contains(GZIP)){
//支持压缩
ByteArrayOutputStream bout = new ByteArrayOutputStream();
//压缩输出流中的数据
GZIPOutputStream gout;
try {
gout = new GZIPOutputStream(bout);
gout.write(content);
gout.close();
content = bout.toByteArray();
originalResponse.getHeaders().setContentLength(content.length);
originalResponse.getHeaders().set("content-encoding", GZIP);
} catch (IOException e) {
e.printStackTrace();
}
}else{
originalResponse.getHeaders().setContentLength(responseString.getBytes().length);
}
}
return bufferFactory.wrap(content);
}));
}
return super.writeWith(body);
}
};
return chain.filter(exchange.mutate().response(decoratedResponse).build());
}
@Override
public int getOrder() {
return NettyWriteResponseFilter.WRITE_RESPONSE_FILTER_ORDER - 2;
}
}
好啦,代码撸完啦!试一下通过网关请求下游服务是否返回了压缩的数据。
ok!网关搞定啦,接下来feign就简单了!
二:Feign之间调用实现gzip压缩
只需要在yaml配置文件中增加如下配置:
#feign开启gzip压缩配置
feign:
compression:
request:
enabled: true
mime-types: ["text/xml,application/xml,application/json"]
min-request-size: 2048
response:
enabled: true
开启debug级别的日志,服务之间调用一下可以看到已经开启了gzip。
我:嘿嘿,全部搞定。晚上才提交,又可以开心的划水了。