Cwikius-Spring/spring-security
1
0
Fork 0
mirror of https://github.com/spring-projects/spring-security.git synced 2025-06-27 14:22:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Optimize SecurityReactorContextConfiguration

Browse Source 
Issue gh-7422
This commit is contained in:
Joe Grandja 2019-09-27 14:46:39 -04:00
parent 3670888616
commit 0fea57d6a1
1 changed files with 17 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
config/src/main/java/org/springframework/security/config/annotation/web/configuration/SecurityReactorContextConfiguration.java
View File

@ -23,7 +23,6 @@ import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration; import org.springframework.context.annotation.Configuration;
import org.springframework.security.core.Authentication; import org.springframework.security.core.Authentication;
import org.springframework.security.core.context.SecurityContextHolder; import org.springframework.security.core.context.SecurityContextHolder;
import org.springframework.util.CollectionUtils;
import org.springframework.web.context.request.RequestContextHolder; import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes; import org.springframework.web.context.request.ServletRequestAttributes;
import reactor.core.CoreSubscriber; import reactor.core.CoreSubscriber;
@ -70,7 +69,7 @@ class SecurityReactorContextConfiguration {
Operators.liftPublisher((pub, sub) -> createSubscriberIfNecessary(sub)); Operators.liftPublisher((pub, sub) -> createSubscriberIfNecessary(sub));
Hooks.onLastOperator(SECURITY_REACTOR_CONTEXT_OPERATOR_KEY, pub -> { Hooks.onLastOperator(SECURITY_REACTOR_CONTEXT_OPERATOR_KEY, pub -> {
if (CollectionUtils.isEmpty(getContextAttributes())) { if (!contextAttributesAvailable()) {
// No need to decorate so return original Publisher // No need to decorate so return original Publisher
return pub; return pub;
} }
@ -91,6 +90,22 @@ class SecurityReactorContextConfiguration {
return new SecurityReactorContextSubscriber<>(delegate, getContextAttributes()); return new SecurityReactorContextSubscriber<>(delegate, getContextAttributes());
} }
private static boolean contextAttributesAvailable() {
HttpServletRequest servletRequest = null;
HttpServletResponse servletResponse = null;
ServletRequestAttributes requestAttributes =
(ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
if (requestAttributes != null) {
servletRequest = requestAttributes.getRequest();
servletResponse = requestAttributes.getResponse();
}
Authentication authentication = SecurityContextHolder.getContext().getAuthentication();
if (authentication != null || servletRequest != null || servletResponse != null) {
return true;
}
return false;
}
private static Map<Object, Object> getContextAttributes() { private static Map<Object, Object> getContextAttributes() {
HttpServletRequest servletRequest = null; HttpServletRequest servletRequest = null;
HttpServletResponse servletResponse = null; HttpServletResponse servletResponse = null;