Merge pull request #228 from shiyindaxiaojie/feature

Feature
shiyindaxiaojie · Feb 9, 2025 · 09fa916 · 09fa916
2 parents 8ffb299 + 8cc4a4e
commit 09fa916
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/...framework/src/main/java/org/ylzl/eden/spring/framework/web/extension/ResponseBuilder.java b/...framework/src/main/java/org/ylzl/eden/spring/framework/web/extension/ResponseBuilder.java
@@ -19,6 +19,7 @@
 import org.jetbrains.annotations.PropertyKey;
 import org.ylzl.eden.extension.ExtensionLoader;
 import org.ylzl.eden.extension.SPI;
+import org.ylzl.eden.spring.framework.beans.ApplicationContextHelper;
 import org.ylzl.eden.spring.framework.error.ErrorCodeLoader;
 
 /**
@@ -31,6 +32,10 @@
 public interface ResponseBuilder<T> {
 
 	static ResponseBuilder<?> builder() {
+		ResponseBuilder<?> builder = ApplicationContextHelper.getBean(ResponseBuilder.class);
+		if (builder != null) {
+			return builder;
+		}
 		return ExtensionLoader.getExtensionLoader(ResponseBuilder.class).getDefaultExtension();
 	}
 

diff --git a/...ava/org/ylzl/eden/spring/framework/web/rest/annotation/RestExceptionHandlerRegistrar.java b/...ava/org/ylzl/eden/spring/framework/web/rest/annotation/RestExceptionHandlerRegistrar.java
@@ -36,6 +36,7 @@ public class RestExceptionHandlerRegistrar implements ImportBeanDefinitionRegist
 	@Override
 	public void registerBeanDefinitions(@NotNull AnnotationMetadata metadata,
 										@NotNull BeanDefinitionRegistry registry) {
+
 		ApplicationContextHelper.registerBean(RestExceptionHandler.class, registry);
 	}
 }