Polish #615, rename spring-cloud-alibaba-sentinel-zuul to spring-cloud-alibaba-sentinel-gateway

spring-cloud-alibaba-sentinel-gateway/src/main/java/org/springframework/cloud/alibaba/sentinel/gateway/SentinelSpringCloudGatewayAutoConfiguration.java

@@ -0,0 +1,91 @@
+/*
+ * Copyright (C) 2018 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.springframework.cloud.alibaba.sentinel.gateway;
+
+import java.util.Collections;
+import java.util.List;
+import java.util.Optional;
+
+import javax.annotation.PostConstruct;
+
+import com.alibaba.csp.sentinel.adapter.gateway.sc.SentinelGatewayFilter;
+import com.alibaba.csp.sentinel.adapter.gateway.sc.callback.BlockRequestHandler;
+import com.alibaba.csp.sentinel.adapter.gateway.sc.callback.GatewayCallbackManager;
+import com.alibaba.csp.sentinel.adapter.gateway.sc.exception.SentinelGatewayBlockExceptionHandler;
+
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.beans.factory.ObjectProvider;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
+import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
+import org.springframework.cloud.gateway.filter.GlobalFilter;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.core.Ordered;
+import org.springframework.core.annotation.Order;
+import org.springframework.http.codec.ServerCodecConfigurer;
+import org.springframework.web.reactive.result.view.ViewResolver;
+
+/**
+ * @author <a href="mailto:fangjian0423@gmail.com">Jim</a>
+ */
+@Configuration
+@ConditionalOnClass(GlobalFilter.class)
+@ConditionalOnProperty(prefix = "spring.cloud.sentinel.spring-cloud-gateway", name = "enabled", havingValue = "true",
+    matchIfMissing = true)
+public class SentinelSpringCloudGatewayAutoConfiguration {
+
+    private static final Logger logger = LoggerFactory
+        .getLogger(SentinelSpringCloudGatewayAutoConfiguration.class);
+
+    private final List<ViewResolver> viewResolvers;
+    private final ServerCodecConfigurer serverCodecConfigurer;
+
+    @Autowired
+    private Optional<BlockRequestHandler> blockRequestHandlerOptional;
+
+    @PostConstruct
+    private void init() {
+        blockRequestHandlerOptional
+            .ifPresent(GatewayCallbackManager::setBlockHandler);
+    }
+
+    public SentinelSpringCloudGatewayAutoConfiguration(
+        ObjectProvider<List<ViewResolver>> viewResolversProvider,
+        ServerCodecConfigurer serverCodecConfigurer) {
+        this.viewResolvers = viewResolversProvider.getIfAvailable(Collections::emptyList);
+        this.serverCodecConfigurer = serverCodecConfigurer;
+    }
+
+    @Bean
+    @Order(Ordered.HIGHEST_PRECEDENCE)
+    public SentinelGatewayBlockExceptionHandler sentinelGatewayBlockExceptionHandler() {
+        // Register the block exception handler for Spring Cloud Gateway.
+        logger.info("[Sentinel SpringCloudGateway] register SentinelGatewayBlockExceptionHandler");
+        return new SentinelGatewayBlockExceptionHandler(viewResolvers,
+            serverCodecConfigurer);
+    }
+
+    @Bean
+    @Order(-1)
+    public GlobalFilter sentinelGatewayFilter() {
+        logger.info("[Sentinel SpringCloudGateway] register SentinelGatewayFilter");
+        return new SentinelGatewayFilter();
+    }
+
+}

spring-cloud-alibaba-sentinel-gateway/src/main/java/org/springframework/cloud/alibaba/sentinel/gateway/SentinelZuulAutoConfiguration.java

@@ -0,0 +1,115 @@
+/*
+ * Copyright 1999-2018 Alibaba Group Holding Ltd.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.springframework.cloud.alibaba.sentinel.gateway;
+
+import java.util.Optional;
+
+import javax.annotation.PostConstruct;
+
+import com.alibaba.csp.sentinel.adapter.gateway.zuul.callback.RequestOriginParser;
+import com.alibaba.csp.sentinel.adapter.gateway.zuul.callback.ZuulGatewayCallbackManager;
+import com.alibaba.csp.sentinel.adapter.gateway.zuul.filters.SentinelZuulErrorFilter;
+import com.alibaba.csp.sentinel.adapter.gateway.zuul.filters.SentinelZuulPostFilter;
+import com.alibaba.csp.sentinel.adapter.gateway.zuul.filters.SentinelZuulPreFilter;
+
+import com.netflix.zuul.ZuulFilter;
+import com.netflix.zuul.http.ZuulServlet;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.support.DefaultListableBeanFactory;
+import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
+import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
+import org.springframework.cloud.alibaba.sentinel.gateway.handler.FallBackProviderHandler;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.core.env.Environment;
+
+/**
+ * Sentinel Spring Cloud Zuul AutoConfiguration
+ *
+ * @author tiger
+ */
+@Configuration
+@ConditionalOnClass(ZuulServlet.class)
+@ConditionalOnProperty(prefix = SentinelZuulAutoConfiguration.PREFIX, name = "enabled", havingValue = "true",
+    matchIfMissing = true)
+public class SentinelZuulAutoConfiguration {
+
+    private static final Logger logger = LoggerFactory
+        .getLogger(SentinelZuulAutoConfiguration.class);
+
+    public static final String PREFIX = "spring.cloud.sentinel.zuul";
+
+    @Autowired
+    private Environment environment;
+
+    @Autowired
+    private Optional<RequestOriginParser> requestOriginParserOptional;
+
+    @PostConstruct
+    private void init() {
+        requestOriginParserOptional
+            .ifPresent(ZuulGatewayCallbackManager::setOriginParser);
+    }
+
+    @Bean
+    public ZuulFilter sentinelZuulPreFilter() {
+        String preOrderStr = environment.getProperty(PREFIX + "." + "order.pre");
+        int order = 10000;
+        try {
+            order = Integer.parseInt(preOrderStr);
+        } catch (NumberFormatException e) {
+            // ignore
+        }
+        logger.info("[Sentinel Zuul] register SentinelZuulPreFilter {}", order);
+        return new SentinelZuulPreFilter(order);
+    }
+
+    @Bean
+    public ZuulFilter sentinelZuulPostFilter() {
+        String postOrderStr = environment.getProperty(PREFIX + "." + "order.post");
+        int order = 1000;
+        try {
+            order = Integer.parseInt(postOrderStr);
+        } catch (NumberFormatException e) {
+            // ignore
+        }
+        logger.info("[Sentinel Zuul] register SentinelZuulPostFilter {}", order);
+        return new SentinelZuulPostFilter(order);
+    }
+
+    @Bean
+    public ZuulFilter sentinelZuulErrorFilter() {
+        String errorOrderStr = environment.getProperty(PREFIX + "." + "order.error");
+        int order = -1;
+        try {
+            order = Integer.parseInt(errorOrderStr);
+        } catch (NumberFormatException e) {
+            // ignore
+        }
+        logger.info("[Sentinel Zuul] register SentinelZuulErrorFilter {}", order);
+        return new SentinelZuulErrorFilter(order);
+    }
+
+    @Bean
+    public FallBackProviderHandler fallBackProviderHandler(
+        DefaultListableBeanFactory beanFactory) {
+        return new FallBackProviderHandler(beanFactory);
+    }
+
+}

spring-cloud-alibaba-sentinel-gateway/src/main/java/org/springframework/cloud/alibaba/sentinel/gateway/handler/FallBackProviderHandler.java

@@ -0,0 +1,44 @@
+package org.springframework.cloud.alibaba.sentinel.gateway.handler;
+
+import java.util.Map;
+
+import com.alibaba.csp.sentinel.adapter.gateway.zuul.fallback.DefaultBlockFallbackProvider;
+import com.alibaba.csp.sentinel.adapter.gateway.zuul.fallback.ZuulBlockFallbackManager;
+import com.alibaba.csp.sentinel.adapter.gateway.zuul.fallback.ZuulBlockFallbackProvider;
+
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.beans.factory.SmartInitializingSingleton;
+import org.springframework.beans.factory.support.DefaultListableBeanFactory;
+import org.springframework.util.CollectionUtils;
+
+/**
+ * @author tiger
+ */
+public class FallBackProviderHandler implements SmartInitializingSingleton {
+
+    private static final Logger logger = LoggerFactory
+        .getLogger(FallBackProviderHandler.class);
+
+    private final DefaultListableBeanFactory beanFactory;
+
+    public FallBackProviderHandler(DefaultListableBeanFactory beanFactory) {
+        this.beanFactory = beanFactory;
+    }
+
+    @Override
+    public void afterSingletonsInstantiated() {
+        Map<String, ZuulBlockFallbackProvider> providerMap = beanFactory
+            .getBeansOfType(ZuulBlockFallbackProvider.class);
+        if (!CollectionUtils.isEmpty(providerMap)) {
+            providerMap.forEach((k, v) -> {
+                logger.info("[Sentinel Zuul] Register provider name:{}, instance: {}", k,
+                    v);
+                ZuulBlockFallbackManager.registerProvider(v);
+            });
+        } else {
+            logger.info("[Sentinel Zuul] Register default fallback provider. ");
+            ZuulBlockFallbackManager.registerProvider(new DefaultBlockFallbackProvider());
+        }
+    }
+}

spring-cloud-alibaba-sentinel-gateway/src/main/resources/META-INF/spring.factories

@@ -0,0 +1,3 @@
+org.springframework.boot.autoconfigure.EnableAutoConfiguration=\
+org.springframework.cloud.alibaba.sentinel.gateway.SentinelZuulAutoConfiguration,\
+org.springframework.cloud.alibaba.sentinel.gateway.SentinelSpringCloudGatewayAutoConfiguration