mirror of
https://gitee.com/mirrors/Spring-Cloud-Alibaba.git
synced 2021-06-26 13:25:11 +08:00
parent
24425830e3
commit
cc4379a271
@ -16,9 +16,16 @@
|
|||||||
|
|
||||||
package org.springframework.cloud.alibaba.sentinel.custom;
|
package org.springframework.cloud.alibaba.sentinel.custom;
|
||||||
|
|
||||||
|
import java.util.Optional;
|
||||||
|
|
||||||
|
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;
|
||||||
|
import com.alibaba.csp.sentinel.init.InitExecutor;
|
||||||
import com.alibaba.csp.sentinel.transport.config.TransportConfig;
|
import com.alibaba.csp.sentinel.transport.config.TransportConfig;
|
||||||
import com.alibaba.csp.sentinel.util.AppNameUtil;
|
import com.alibaba.csp.sentinel.util.AppNameUtil;
|
||||||
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.beans.factory.annotation.Autowired;
|
||||||
|
import org.springframework.beans.factory.annotation.Qualifier;
|
||||||
import org.springframework.beans.factory.annotation.Value;
|
import org.springframework.beans.factory.annotation.Value;
|
||||||
import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
|
import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
|
||||||
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
|
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
|
||||||
@ -43,24 +50,35 @@ import javax.annotation.PostConstruct;
|
|||||||
@EnableConfigurationProperties(SentinelProperties.class)
|
@EnableConfigurationProperties(SentinelProperties.class)
|
||||||
public class SentinelAutoConfiguration {
|
public class SentinelAutoConfiguration {
|
||||||
|
|
||||||
@Value("${project.name:${spring.application.name:}}")
|
@Value("${project.name:${spring.application.name:}}")
|
||||||
private String projectName;
|
private String projectName;
|
||||||
|
|
||||||
@Autowired
|
@Autowired
|
||||||
private SentinelProperties properties;
|
private SentinelProperties properties;
|
||||||
|
|
||||||
@PostConstruct
|
@Autowired
|
||||||
private void init() {
|
private Optional<UrlBlockHandler> urlBlockHandlerOptional;
|
||||||
if (StringUtils.isEmpty(System.getProperty(AppNameUtil.APP_NAME))) {
|
|
||||||
System.setProperty(AppNameUtil.APP_NAME, projectName);
|
@Autowired
|
||||||
}
|
private Optional<UrlCleaner> urlCleanerOptional;
|
||||||
if (StringUtils.isEmpty(System.getProperty(TransportConfig.SERVER_PORT))) {
|
|
||||||
System.setProperty(TransportConfig.SERVER_PORT, properties.getPort());
|
@PostConstruct
|
||||||
}
|
private void init() {
|
||||||
if (StringUtils.isEmpty(System.getProperty(TransportConfig.CONSOLE_SERVER))) {
|
if (StringUtils.isEmpty(System.getProperty(AppNameUtil.APP_NAME))) {
|
||||||
System.setProperty(TransportConfig.CONSOLE_SERVER, properties.getDashboard());
|
System.setProperty(AppNameUtil.APP_NAME, projectName);
|
||||||
}
|
}
|
||||||
}
|
if (StringUtils.isEmpty(System.getProperty(TransportConfig.SERVER_PORT))) {
|
||||||
|
System.setProperty(TransportConfig.SERVER_PORT, properties.getPort());
|
||||||
|
}
|
||||||
|
if (StringUtils.isEmpty(System.getProperty(TransportConfig.CONSOLE_SERVER))) {
|
||||||
|
System.setProperty(TransportConfig.CONSOLE_SERVER, properties.getDashboard());
|
||||||
|
}
|
||||||
|
|
||||||
|
urlBlockHandlerOptional.ifPresent(WebCallbackManager::setUrlBlockHandler);
|
||||||
|
urlCleanerOptional.ifPresent(WebCallbackManager::setUrlCleaner);
|
||||||
|
|
||||||
|
InitExecutor.doInit();
|
||||||
|
}
|
||||||
|
|
||||||
@Bean
|
@Bean
|
||||||
@ConditionalOnMissingBean
|
@ConditionalOnMissingBean
|
||||||
|
Loading…
x
Reference in New Issue
Block a user