@@ -38,10 +38,13 @@ import org.springframework.boot.SpringApplication;
|
||||
import org.springframework.boot.WebApplicationType;
|
||||
import org.springframework.boot.autoconfigure.SpringBootApplication;
|
||||
import org.springframework.cloud.function.context.FunctionCatalog;
|
||||
import org.springframework.cloud.function.context.FunctionRegistration;
|
||||
import org.springframework.cloud.function.context.FunctionType;
|
||||
import org.springframework.cloud.function.context.catalog.FunctionInspector;
|
||||
import org.springframework.context.ApplicationContext;
|
||||
import org.springframework.context.ApplicationContextInitializer;
|
||||
import org.springframework.context.ConfigurableApplicationContext;
|
||||
import org.springframework.context.support.GenericApplicationContext;
|
||||
import org.springframework.util.ClassUtils;
|
||||
|
||||
/**
|
||||
@@ -135,22 +138,39 @@ public class AzureSpringFunctionInitializer implements Closeable {
|
||||
}
|
||||
|
||||
private SpringApplication springApplication() {
|
||||
ApplicationContextInitializer<?> initializer = null;
|
||||
ApplicationContextInitializer<GenericApplicationContext> initializer = null;
|
||||
Class<?> sourceClass = configurationClass;
|
||||
if (ApplicationContextInitializer.class.isAssignableFrom(sourceClass)) {
|
||||
initializer = BeanUtils.instantiateClass(configurationClass, ApplicationContextInitializer.class);
|
||||
@SuppressWarnings("unchecked")
|
||||
ApplicationContextInitializer<GenericApplicationContext> instance = BeanUtils
|
||||
.instantiateClass(configurationClass,
|
||||
ApplicationContextInitializer.class);
|
||||
initializer = instance;
|
||||
sourceClass = Object.class;
|
||||
}
|
||||
else if (Function.class.isAssignableFrom(sourceClass)) {
|
||||
@SuppressWarnings("unchecked")
|
||||
final Class<Function<?, ?>> type = (Class<Function<?, ?>>) sourceClass;
|
||||
initializer = context -> {
|
||||
context.registerBean(FunctionRegistration.class,
|
||||
() -> new FunctionRegistration<>(
|
||||
context.getAutowireCapableBeanFactory().createBean(type))
|
||||
.type(FunctionType.of(type)));
|
||||
};
|
||||
sourceClass = Object.class;
|
||||
}
|
||||
SpringApplication application;
|
||||
if (initializer!=null) {
|
||||
if (initializer != null) {
|
||||
application = new SpringApplication(sourceClass) {
|
||||
@Override
|
||||
protected void load(ApplicationContext context, Object[] sources) {
|
||||
}
|
||||
};
|
||||
application.addInitializers(initializer);
|
||||
application.setDefaultProperties(Collections.singletonMap("spring.functional.enabled", "true"));
|
||||
} else {
|
||||
application.setDefaultProperties(
|
||||
Collections.singletonMap("spring.functional.enabled", "true"));
|
||||
}
|
||||
else {
|
||||
application = new SpringApplication(sourceClass);
|
||||
}
|
||||
application.setWebApplicationType(WebApplicationType.NONE);
|
||||
@@ -240,4 +260,16 @@ public class AzureSpringFunctionInitializer implements Closeable {
|
||||
}
|
||||
return null;
|
||||
}
|
||||
|
||||
public Function<Publisher<?>, Publisher<?>> getFunction() {
|
||||
return function;
|
||||
}
|
||||
|
||||
public FunctionCatalog getCatalog() {
|
||||
return catalog;
|
||||
}
|
||||
|
||||
public FunctionInspector getInspector() {
|
||||
return inspector;
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user