!55 修改网关返回异常信息时的response 请求头

Merge pull request !55 from caiqyxyx/N/A
This commit is contained in:
pig4cloud.com 2020-05-13 14:10:06 +08:00 committed by Gitee
commit 7dc9dc01bc
1 changed files with 2 additions and 0 deletions

View File

@ -34,6 +34,7 @@ import org.springframework.cloud.gateway.filter.GatewayFilter;
import org.springframework.cloud.gateway.filter.factory.AbstractGatewayFilterFactory; import org.springframework.cloud.gateway.filter.factory.AbstractGatewayFilterFactory;
import org.springframework.data.redis.core.RedisTemplate; import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.http.HttpStatus; import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.http.server.reactive.ServerHttpRequest; import org.springframework.http.server.reactive.ServerHttpRequest;
import org.springframework.http.server.reactive.ServerHttpResponse; import org.springframework.http.server.reactive.ServerHttpResponse;
import org.springframework.stereotype.Component; import org.springframework.stereotype.Component;
@ -81,6 +82,7 @@ public class ValidateCodeGatewayFilter extends AbstractGatewayFilterFactory {
} catch (Exception e) { } catch (Exception e) {
ServerHttpResponse response = exchange.getResponse(); ServerHttpResponse response = exchange.getResponse();
response.setStatusCode(HttpStatus.PRECONDITION_REQUIRED); response.setStatusCode(HttpStatus.PRECONDITION_REQUIRED);
response.getHeaders().setContentType(MediaType.APPLICATION_JSON);
try { try {
return response.writeWith(Mono.just(response.bufferFactory() return response.writeWith(Mono.just(response.bufferFactory()
.wrap(objectMapper.writeValueAsBytes( .wrap(objectMapper.writeValueAsBytes(