|
21 | 21 | import org.springframework.web.server.ServerWebExchange;
|
22 | 22 | import reactor.core.publisher.Mono;
|
23 | 23 |
|
| 24 | +import java.net.URI; |
| 25 | +import java.util.HashMap; |
| 26 | +import java.util.Map; |
| 27 | + |
24 | 28 | /**
|
25 | 29 | * 网关鉴权
|
26 | 30 | *
|
@@ -79,7 +83,29 @@ public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
|
79 | 83 | //先记录下访问接口的开始时间
|
80 | 84 | exchange.getAttributes().put(BEGIN_VISIT_TIME, System.currentTimeMillis());
|
81 | 85 |
|
82 |
| - return chain.filter(exchange.mutate().request(mutate.build()).build()); |
| 86 | +// return chain.filter(exchange.mutate().request(mutate.build()).build()); |
| 87 | + |
| 88 | + // Mono.fromRunnable 是非阻塞的,适合在 then 中处理后续的日志逻辑。 |
| 89 | + return chain.filter(exchange).then(Mono.fromRunnable(() -> { |
| 90 | + try { |
| 91 | + // 记录接口访问日志 |
| 92 | + Long beginVisitTime = exchange.getAttribute(BEGIN_VISIT_TIME); |
| 93 | + if (beginVisitTime != null) { |
| 94 | + URI uri = exchange.getRequest().getURI(); |
| 95 | + Map<String, Object> logData = new HashMap<>(); |
| 96 | + logData.put("host", uri.getHost()); |
| 97 | + logData.put("port", uri.getPort()); |
| 98 | + logData.put("path", uri.getPath()); |
| 99 | + logData.put("query", uri.getRawQuery()); |
| 100 | + logData.put("duration", (System.currentTimeMillis() - beginVisitTime) + "ms"); |
| 101 | + |
| 102 | + log.info("访问接口信息: {}", logData); |
| 103 | + log.info("我是美丽分割线: ###################################################"); |
| 104 | + } |
| 105 | + } catch (Exception e) { |
| 106 | + log.error("记录日志时发生异常: ", e); |
| 107 | + } |
| 108 | + })); |
83 | 109 | }
|
84 | 110 |
|
85 | 111 | private void addHeader(ServerHttpRequest.Builder mutate, String name, Object value) {
|
|
0 commit comments