puzhibing
2023-07-17 f195cf48cf57635c8848b2bc32afd4541c4090ed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
//package cn.mb.cloud.gateway.filter;
//
//import com.alibaba.fastjson.JSON;
//import lombok.extern.slf4j.Slf4j;
//import org.reactivestreams.Publisher;
//import org.springframework.cloud.gateway.filter.GatewayFilterChain;
//import org.springframework.cloud.gateway.filter.GlobalFilter;
//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.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.stereotype.Component;
//import org.springframework.web.server.ServerWebExchange;
//import reactor.core.publisher.Flux;
//import reactor.core.publisher.Mono;
//
//import java.nio.charset.StandardCharsets;
//import java.util.ArrayList;
//import java.util.List;
//
///**
// * 日志拦截器
// */
//@Slf4j
//@Component
//public class ResponseParamFilter implements GlobalFilter, Ordered {
//
//    @Override
//    public int getOrder() {
//        return -1;
//    }
//
//    @Override
//    public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
//        try {
//            ServerHttpRequest request = exchange.getRequest();
//            String uri = request.getURI().getRawPath();
//            ServerHttpResponse originalResponse = exchange.getResponse();
//            DataBufferFactory bufferFactory = originalResponse.bufferFactory();
//            HttpStatus statusCode = originalResponse.getStatusCode();
//            if(statusCode == HttpStatus.OK){
//                ServerHttpResponseDecorator decoratedResponse = new ServerHttpResponseDecorator(originalResponse) {
//                    @Override
//                    public Mono<Void> writeWith(Publisher<? extends DataBuffer> body) {
//                        if (body instanceof Flux) {
//                            Flux<? extends DataBuffer> fluxBody = Flux.from(body);
//                            return super.writeWith(fluxBody.map(dataBuffer -> {
//                                byte[] content = new byte[dataBuffer.readableByteCount()];
//                                dataBuffer.read(content);
//                                DataBufferUtils.release(dataBuffer);//释放掉内存 Release memory
//                                // 构建日志 build Journal
//                                StringBuffer sb = new StringBuffer();
//                                sb.append(" 响应操作 -> " + uri + " || 响应参数-> {} \n");
//
//                                List<Object> rspArgs = new ArrayList<>();
//                                rspArgs.add(originalResponse.getStatusCode());
//                                String data = new String(content, StandardCharsets.UTF_8);//data
//                                sb.append(data);
//                                log.info(sb.toString(), rspArgs.toArray());
//                                return bufferFactory.wrap(content);
//                            }));
//                        } else {
//                            log.error("响应操作 -> " + uri + " || 响应参数-> {} \n", getStatusCode());
//                        }
//                        return super.writeWith(body);
//                    }
//                };
//                return chain.filter(exchange.mutate().response(decoratedResponse).build());
//            }
//            return chain.filter(exchange);//降级处理返回数据 Downgrade processing returns data
//        }catch (Exception e){
//            log.error("gateway log exception.\n" + e);
//            return chain.filter(exchange);
//        }
//    }
//}