4 changed files with 130 additions and 7 deletions
@ -0,0 +1,83 @@
@@ -0,0 +1,83 @@
|
||||
/* |
||||
* Copyright 2002-present the original author or authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
*/ |
||||
|
||||
package org.springframework.web.servlet.mvc.method.annotation; |
||||
|
||||
import org.jspecify.annotations.Nullable; |
||||
|
||||
import org.springframework.core.MethodParameter; |
||||
import org.springframework.http.HttpEntity; |
||||
import org.springframework.web.context.request.NativeWebRequest; |
||||
import org.springframework.web.method.support.HandlerMethodReturnValueHandler; |
||||
import org.springframework.web.method.support.ModelAndViewContainer; |
||||
|
||||
/** |
||||
* Handler for return values of type {@link org.springframework.http.ResponseEntity} |
||||
* that delegates to one of the following: |
||||
* |
||||
* <ul> |
||||
* <li>{@link HttpEntityMethodProcessor} for responses with a concrete body value |
||||
* <li>{@link ResponseBodyEmitterReturnValueHandler} for responses with a body |
||||
* that is a {@link ResponseBodyEmitter} or an async/reactive type. |
||||
* </ul> |
||||
* |
||||
* <p>Use of this wrapper allows for late check in {@link #handleReturnValue} of |
||||
* the type of the actual body value in case the method signature does not |
||||
* provide enough information to decide via {@link #supportsReturnType}. |
||||
* |
||||
* @author Rossen Stoyanchev |
||||
* @since 7.0 |
||||
*/ |
||||
public class ResponseEntityReturnValueHandler implements HandlerMethodReturnValueHandler { |
||||
|
||||
private final HttpEntityMethodProcessor httpEntityMethodProcessor; |
||||
|
||||
private final ResponseBodyEmitterReturnValueHandler responseBodyEmitterHandler; |
||||
|
||||
|
||||
public ResponseEntityReturnValueHandler( |
||||
HttpEntityMethodProcessor httpEntityMethodProcessor, |
||||
ResponseBodyEmitterReturnValueHandler responseBodyEmitterHandler) { |
||||
|
||||
this.httpEntityMethodProcessor = httpEntityMethodProcessor; |
||||
this.responseBodyEmitterHandler = responseBodyEmitterHandler; |
||||
} |
||||
|
||||
|
||||
@Override |
||||
public boolean supportsReturnType(MethodParameter returnType) { |
||||
return this.httpEntityMethodProcessor.supportsReturnType(returnType); |
||||
} |
||||
|
||||
@Override |
||||
public void handleReturnValue( |
||||
@Nullable Object returnValue, MethodParameter returnType, ModelAndViewContainer mavContainer, |
||||
NativeWebRequest request) throws Exception { |
||||
|
||||
if (returnValue instanceof HttpEntity<?> httpEntity) { |
||||
Object body = httpEntity.getBody(); |
||||
if (body != null) { |
||||
if (this.responseBodyEmitterHandler.supportsBodyType(body.getClass())) { |
||||
this.responseBodyEmitterHandler.handleReturnValue(returnValue, returnType, mavContainer, request); |
||||
return; |
||||
} |
||||
} |
||||
} |
||||
|
||||
this.httpEntityMethodProcessor.handleReturnValue(returnValue, returnType, mavContainer, request); |
||||
} |
||||
|
||||
} |
||||
Loading…
Reference in new issue