Spring/spring-framework
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Polishing contribution

Browse Source 
Closes gh-26502
This commit is contained in:
Rossen Stoyanchev
2021-02-08 17:35:59 +00:00
parent 0c2c66b38b
commit 26000ee29f
3 changed files with 54 additions and 89 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
spring-web/src/main/java/org/springframework/http/server/reactive/HttpHandlerDecoratorFactory.java
View File

@@ -1,5 +1,5 @@
/*
* Copyright 2002-2018 the original author or authors.
* Copyright 2002-2021 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.
@@ -16,25 +16,12 @@
package org.springframework.http.server.reactive;
import org.springframework.context.ApplicationContext;
import java.util.function.Function;
import java.util.function.UnaryOperator;
/**
* Allows registering a bean that will decorate the instance of {@link HttpHandler},
* used by {@link org.springframework.web.server.adapter.WebHttpHandlerBuilder#applicationContext(ApplicationContext)};
*
* Contract for applying a decorator to an {@code HttpHandler}.
* @author Christophe Maillard
* @since 5.3.4
*/
public interface HttpHandlerDecoratorFactory extends UnaryOperator<HttpHandler> {
static HttpHandlerDecoratorFactory identity() {
return x -> x;
}
default Function<HttpHandler, HttpHandler> toFunction() {
return this;
}
public interface HttpHandlerDecoratorFactory extends Function<HttpHandler, HttpHandler> {
}

9
spring-web/src/main/java/org/springframework/web/server/adapter/WebHttpHandlerBuilder.java
View File

@@ -1,5 +1,5 @@
/*
* Copyright 2002-2020 the original author or authors.
* Copyright 2002-2021 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.
@@ -176,12 +176,9 @@ public final class WebHttpHandlerBuilder {
.collect(Collectors.toList());
builder.exceptionHandlers(handlers -> handlers.addAll(exceptionHandlers));
Function<HttpHandler, HttpHandler> httpHandlerDecorator = context
.getBeanProvider(HttpHandlerDecoratorFactory.class)
context.getBeanProvider(HttpHandlerDecoratorFactory.class)
.orderedStream()
.map(HttpHandlerDecoratorFactory::toFunction)
.reduce(Function.identity(), Function::andThen);
builder.httpHandlerDecorator(httpHandlerDecorator);
.forEach(builder::httpHandlerDecorator);
try {
builder.sessionManager(