SpringCloud/spring-cloud-function
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

GH-947 Ensure FunctionAroundWrapper is not registered

Browse Source 
In 4.x it will not since it is not a function, but here we needed a filtering logic during the default lookup

Resolves #947
This commit is contained in:
Oleg Zhurakousky
2022-10-31 15:54:57 +01:00
parent 1b3810ce60
commit 1facb670fd
1 changed files with 15 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
spring-cloud-function-context/src/main/java/org/springframework/cloud/function/context/catalog/BeanFactoryAwareFunctionRegistry.java
View File

@@ -19,7 +19,9 @@ package org.springframework.cloud.function.context.catalog;
import java.lang.reflect.Method;
import java.lang.reflect.ParameterizedType;
import java.lang.reflect.Type;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Set;
import java.util.function.BiConsumer;
import java.util.function.BiFunction;
@@ -90,6 +92,18 @@ public class BeanFactoryAwareFunctionRegistry extends SimpleFunctionRegistry imp
@Override
public Set<String> getNames(Class<?> type) {
Set<String> registeredNames = super.getNames(type);
//--- see https://github.com/spring-cloud/spring-cloud-function/issues/947
Set<String> arroundWrapperNeamnames = this.applicationContext.getBeansOfType(FunctionAroundWrapper.class).keySet();
String[] names = this.applicationContext.getBeanNamesForType(BiFunction.class);
List<String> biFunctions = new ArrayList<>();
for (int i = 0; i < names.length; i++) {
if (!arroundWrapperNeamnames.contains(names[i])) {
biFunctions.add(names[i]);
}
}
///
if (type == null) {
registeredNames
.addAll(Arrays.asList(this.applicationContext.getBeanNamesForType(Function.class)));
@@ -98,7 +112,7 @@ public class BeanFactoryAwareFunctionRegistry extends SimpleFunctionRegistry imp
registeredNames
.addAll(Arrays.asList(this.applicationContext.getBeanNamesForType(Consumer.class)));
registeredNames
.addAll(Arrays.asList(this.applicationContext.getBeanNamesForType(BiFunction.class)));
.addAll(biFunctions);
registeredNames
.addAll(Arrays.asList(this.applicationContext.getBeanNamesForType(BiConsumer.class)));
registeredNames