From 6447ae861477200137935615b7bd0c2516d4b75e Mon Sep 17 00:00:00 2001
From: fangjian0423 <fangjian0423@gmail.com>
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<UrlBlockHandler> urlBlockHandlerOptional;
 
+	@Autowired
+	private Optional<RequestOriginParser> 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()) {