From 6447ae861477200137935615b7bd0c2516d4b75e Mon Sep 17 00:00:00 2001 From: fangjian0423 Date: Fri, 11 Jan 2019 15:00:57 +0800 Subject: [PATCH] Autowired RequestOriginParser --- .../alibaba/sentinel/custom/SentinelAutoConfiguration.java | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/spring-cloud-alibaba-sentinel/src/main/java/org/springframework/cloud/alibaba/sentinel/custom/SentinelAutoConfiguration.java b/spring-cloud-alibaba-sentinel/src/main/java/org/springframework/cloud/alibaba/sentinel/custom/SentinelAutoConfiguration.java index a4d4176b..8e6f8db8 100644 --- a/spring-cloud-alibaba-sentinel/src/main/java/org/springframework/cloud/alibaba/sentinel/custom/SentinelAutoConfiguration.java +++ b/spring-cloud-alibaba-sentinel/src/main/java/org/springframework/cloud/alibaba/sentinel/custom/SentinelAutoConfiguration.java @@ -33,6 +33,7 @@ import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Configuration; import org.springframework.util.StringUtils; +import com.alibaba.csp.sentinel.adapter.servlet.callback.RequestOriginParser; import com.alibaba.csp.sentinel.adapter.servlet.callback.UrlBlockHandler; import com.alibaba.csp.sentinel.adapter.servlet.callback.UrlCleaner; import com.alibaba.csp.sentinel.adapter.servlet.callback.WebCallbackManager; @@ -74,6 +75,9 @@ public class SentinelAutoConfiguration { @Autowired private Optional urlBlockHandlerOptional; + @Autowired + private Optional requestOriginParserOptional; + @PostConstruct private void init() { if (StringUtils.isEmpty(System.getProperty(AppNameUtil.APP_NAME)) @@ -138,6 +142,7 @@ public class SentinelAutoConfiguration { urlBlockHandlerOptional.ifPresent(WebCallbackManager::setUrlBlockHandler); urlCleanerOptional.ifPresent(WebCallbackManager::setUrlCleaner); + requestOriginParserOptional.ifPresent(WebCallbackManager::setRequestOriginParser); // earlier initialize if (properties.isEager()) {