2424import org .springframework .context .annotation .Configuration ;
2525import org .springframework .core .annotation .Order ;
2626import org .springframework .core .io .buffer .DataBufferFactory ;
27- import org .springframework .http .HttpStatus ;
2827import org .springframework .http .MediaType ;
2928import org .springframework .http .server .reactive .ServerHttpRequest ;
3029import org .springframework .http .server .reactive .ServerHttpResponse ;
31- import org .springframework .lang .NonNull ;
3230import org .springframework .web .server .ResponseStatusException ;
3331import org .springframework .web .server .ServerWebExchange ;
3432import reactor .core .publisher .Mono ;
3533
34+ import java .util .Map ;
35+
3636/**
3737 * 异常处理
3838 *
@@ -45,43 +45,42 @@ public class ErrorExceptionHandler implements ErrorWebExceptionHandler {
4545
4646 private final ObjectMapper objectMapper ;
4747
48- @ NonNull
4948 @ Override
50- public Mono <Void > handle (ServerWebExchange exchange , @ NonNull Throwable ex ) {
49+ public Mono <Void > handle (ServerWebExchange exchange , Throwable ex ) {
5150 ServerHttpRequest request = exchange .getRequest ();
5251 ServerHttpResponse response = exchange .getResponse ();
52+
5353 if (response .isCommitted ()) {
5454 return Mono .error (ex );
5555 }
56+
5657 response .getHeaders ().setContentType (MediaType .APPLICATION_JSON );
5758 if (ex instanceof ResponseStatusException ) {
58- response .setStatusCode (((ResponseStatusException ) ex ).getStatus ());
59+ response .setStatusCode (((ResponseStatusException ) ex ).getStatusCode ());
5960 }
61+
6062 return response .writeWith (Mono .fromSupplier (() -> {
6163 DataBufferFactory bufferFactory = response .bufferFactory ();
6264 try {
63- HttpStatus status = HttpStatus . BAD_GATEWAY ;
64- if (ex instanceof ResponseStatusException ) {
65- status = (( ResponseStatusException ) ex ). getStatus ();
65+ int status = 500 ;
66+ if (response . getStatusCode () != null ) {
67+ status = response . getStatusCode (). value ();
6668 }
67- return bufferFactory .wrap (objectMapper .writeValueAsBytes (ResponseProvider .response (status .value (), buildMessage (request , ex ))));
69+ Map <String , Object > result = ResponseProvider .response (status , this .buildMessage (request , ex ));
70+ return bufferFactory .wrap (objectMapper .writeValueAsBytes (result ));
6871 } catch (JsonProcessingException e ) {
69- e .printStackTrace ();
7072 return bufferFactory .wrap (new byte [0 ]);
7173 }
7274 }));
7375 }
7476
77+
7578 /**
7679 * 构建异常信息
7780 */
7881 private String buildMessage (ServerHttpRequest request , Throwable ex ) {
79- String uri = request .getURI ().toString ();
80- if (uri .endsWith ("doc.html" )) {
81- return "[Swagger聚合网关] 已迁移至 [blade-swagger] 服务，请开启 [blade-swagger] 服务并访问 [http://127.0.0.1:18000/doc.html]" ;
82- }
8382 StringBuilder message = new StringBuilder ("Failed to handle request [" );
84- message .append (request .getMethodValue ());
83+ message .append (request .getMethod (). name ());
8584 message .append (" " );
8685 message .append (request .getURI ());
8786 message .append ("]" );
@@ -92,4 +91,5 @@ private String buildMessage(ServerHttpRequest request, Throwable ex) {
9291 return message .toString ();
9392 }
9493
94+
9595}
0 commit comments