Improve support of Webflux with Functional Endpoints

Author: bnasslahsen

springdoc-openapi-common/src/main/java/org/springdoc/core/visitor/AbstractRouterFunctionVisitor.java

@@ -0,0 +1,84 @@
+package org.springdoc.core.visitor;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Set;
+
+import org.springdoc.core.models.RouterFunctionData;
+
+import org.springframework.http.HttpHeaders;
+import org.springframework.http.HttpMethod;
+
+public class AbstractRouterFunctionVisitor {
+
+	protected List<RouterFunctionData> routerFunctionDatas = new ArrayList<>();
+
+	protected RouterFunctionData routerFunctionData;
+
+	public void method(Set<HttpMethod> methods) {
+		routerFunctionData.setMethods(methods);
+	}
+
+	public void path(String pattern) {
+		routerFunctionData.setPath(pattern);
+	}
+
+	public void header(String name, String value) {
+		if (HttpHeaders.ACCEPT.equals(name))
+			routerFunctionData.setConsumes(value);
+		else
+			routerFunctionData.setHeaders(name + "=" + value);
+	}
+
+	public List<RouterFunctionData> getRouterFunctionDatas() {
+		return routerFunctionDatas;
+	}
+
+	public void queryParam(String name, String value) {
+		// Not yet needed
+	}
+
+	public void pathExtension(String extension) {
+		// Not yet needed
+	}
+
+	public void param(String name, String value) {
+		// Not yet needed
+	}
+
+	public void startAnd() {
+		// Not yet needed
+	}
+
+	public void and() {
+		// Not yet needed
+	}
+
+	public void endAnd() {
+		// Not yet needed
+	}
+
+	public void startOr() {
+		// Not yet needed
+	}
+
+	public void or() {
+		// Not yet needed
+	}
+
+	public void endOr() {
+		// Not yet needed
+	}
+
+	public void startNegate() {
+		// Not yet needed
+	}
+
+	public void endNegate() {
+		// Not yet needed
+	}
+
+
+
+
+}

springdoc-openapi-webflux-core/src/main/java/org/springdoc/webflux/api/OpenApiResource.java

@@ -43,6 +43,7 @@
 import org.springdoc.core.annotations.RouterOperations;
 import org.springdoc.core.customizers.OpenApiCustomiser;
 import org.springdoc.core.customizers.OperationCustomizer;
+import org.springdoc.webflux.core.visitor.RouterFunctionVisitor;
 import reactor.core.publisher.Mono;
 
 import org.springframework.beans.factory.annotation.Autowired;
@@ -133,7 +134,7 @@ protected void getPaths(Map<String, Object> restControllers) {
 		getWebFluxRouterFunctionPaths();
 	}
 
-	private void getWebFluxRouterFunctionPaths() {
+	protected void getWebFluxRouterFunctionPaths() {
 		ApplicationContext applicationContext = requestMappingHandlerMapping.getApplicationContext();
 		Map<String, RouterFunction> routerBeans = applicationContext.getBeansOfType(RouterFunction.class);
 		for (Map.Entry<String, RouterFunction> entry : routerBeans.entrySet()) {

springdoc-openapi-webflux-core/src/main/java/org/springdoc/webflux/api/RouterFunctionVisitor.java

@@ -0,0 +1,52 @@
+package org.springdoc.webflux.core.visitor;
+
+import java.util.function.Function;
+
+import org.springdoc.core.models.RouterFunctionData;
+import org.springdoc.core.visitor.AbstractRouterFunctionVisitor;
+import reactor.core.publisher.Mono;
+
+import org.springframework.core.io.Resource;
+import org.springframework.web.reactive.function.server.HandlerFunction;
+import org.springframework.web.reactive.function.server.RequestPredicate;
+import org.springframework.web.reactive.function.server.RequestPredicates;
+import org.springframework.web.reactive.function.server.RouterFunction;
+import org.springframework.web.reactive.function.server.RouterFunctions;
+import org.springframework.web.reactive.function.server.ServerRequest;
+
+public class RouterFunctionVisitor extends AbstractRouterFunctionVisitor implements RouterFunctions.Visitor, RequestPredicates.Visitor {
+
+	@Override
+	public void route(RequestPredicate predicate, HandlerFunction<?> handlerFunction) {
+		this.routerFunctionData = new RouterFunctionData();
+		routerFunctionDatas.add(this.routerFunctionData);
+		predicate.accept(this);
+	}
+
+	@Override
+	public void startNested(RequestPredicate predicate) {
+		// Not yet needed
+	}
+
+	@Override
+	public void endNested(RequestPredicate predicate) {
+		// Not yet needed
+	}
+
+	@Override
+	public void resources(Function<ServerRequest, Mono<Resource>> lookupFunction) {
+		// Not yet needed
+	}
+
+	@Override
+	public void unknown(RouterFunction<?> routerFunction) {
+		// Not yet needed
+	}
+
+	@Override
+	public void unknown(RequestPredicate predicate) {
+		// Not yet needed
+	}
+}
+
+

springdoc-openapi-webflux-core/src/main/java/org/springdoc/webflux/core/visitor/RouterFunctionVisitor.java

@@ -47,6 +47,7 @@
 import org.springdoc.core.annotations.RouterOperations;
 import org.springdoc.core.customizers.OpenApiCustomiser;
 import org.springdoc.core.customizers.OperationCustomizer;
+import org.springdoc.webmvc.core.visitor.RouterFunctionVisitor;
 
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Value;
@@ -170,7 +171,7 @@ && isPathToMatch(operationPath)) {
 		getWebMvcRouterFunctionPaths();
 	}
 
-	private void getWebMvcRouterFunctionPaths() {
+	protected void getWebMvcRouterFunctionPaths() {
 		ApplicationContext applicationContext = requestMappingHandlerMapping.getApplicationContext();
 		Map<String, RouterFunction> routerBeans = applicationContext.getBeansOfType(RouterFunction.class);
 		for (Map.Entry<String, RouterFunction> entry : routerBeans.entrySet()) {