[修改]sentinel 自定义默认异常

spring-cloud-timeloit-sentinel/src/main/java/com/timeloit/cloud/sentinel/SentinelWebAutoConfiguration.java

@@ -16,17 +16,15 @@
 
 package com.timeloit.cloud.sentinel;
 
-import java.util.Optional;
-
 import com.alibaba.csp.sentinel.adapter.spring.webmvc.SentinelWebInterceptor;
 import com.alibaba.csp.sentinel.adapter.spring.webmvc.callback.BlockExceptionHandler;
 import com.alibaba.csp.sentinel.adapter.spring.webmvc.callback.DefaultBlockExceptionHandler;
 import com.alibaba.csp.sentinel.adapter.spring.webmvc.callback.RequestOriginParser;
 import com.alibaba.csp.sentinel.adapter.spring.webmvc.callback.UrlCleaner;
 import com.alibaba.csp.sentinel.adapter.spring.webmvc.config.SentinelWebMvcConfig;
+import com.timeloit.cloud.sentinel.loit.CustomBlockExceptionHandler;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
-
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
 import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
@@ -39,6 +37,8 @@ import org.springframework.util.StringUtils;
 import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
 import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
 
+import java.util.Optional;
+
 /**
  * @author xiaojing
  * @author yuhuangbin
@@ -50,71 +50,74 @@ import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
 @EnableConfigurationProperties(SentinelProperties.class)
 public class SentinelWebAutoConfiguration implements WebMvcConfigurer {
 
-	private static final Logger log = LoggerFactory
-			.getLogger(SentinelWebAutoConfiguration.class);
-
-	@Autowired
-	private SentinelProperties properties;
-
-	@Autowired
-	private Optional<UrlCleaner> urlCleanerOptional;
-
-	@Autowired
-	private Optional<BlockExceptionHandler> blockExceptionHandlerOptional;
-
-	@Autowired
-	private Optional<RequestOriginParser> requestOriginParserOptional;
-
-	@Autowired
-	private Optional<SentinelWebInterceptor> sentinelWebInterceptorOptional;
-
-	@Override
-	public void addInterceptors(InterceptorRegistry registry) {
-		if (!sentinelWebInterceptorOptional.isPresent()) {
-			return;
-		}
-		SentinelProperties.Filter filterConfig = properties.getFilter();
-		registry.addInterceptor(sentinelWebInterceptorOptional.get())
-				.order(filterConfig.getOrder())
-				.addPathPatterns(filterConfig.getUrlPatterns());
-		log.info(
-				"[Sentinel Starter] register SentinelWebInterceptor with urlPatterns: {}.",
-				filterConfig.getUrlPatterns());
-	}
-
-	@Bean
-	@ConditionalOnProperty(name = "spring.cloud.sentinel.filter.enabled",
-			matchIfMissing = true)
-	public SentinelWebInterceptor sentinelWebInterceptor(
-			SentinelWebMvcConfig sentinelWebMvcConfig) {
-		return new SentinelWebInterceptor(sentinelWebMvcConfig);
-	}
-
-	@Bean
-	@ConditionalOnProperty(name = "spring.cloud.sentinel.filter.enabled",
-			matchIfMissing = true)
-	public SentinelWebMvcConfig sentinelWebMvcConfig() {
-		SentinelWebMvcConfig sentinelWebMvcConfig = new SentinelWebMvcConfig();
-		sentinelWebMvcConfig.setHttpMethodSpecify(properties.getHttpMethodSpecify());
-
-		if (blockExceptionHandlerOptional.isPresent()) {
-			blockExceptionHandlerOptional
-					.ifPresent(sentinelWebMvcConfig::setBlockExceptionHandler);
-		}
-		else {
-			if (StringUtils.hasText(properties.getBlockPage())) {
-				sentinelWebMvcConfig.setBlockExceptionHandler(((request, response,
-						e) -> response.sendRedirect(properties.getBlockPage())));
-			}
-			else {
-				sentinelWebMvcConfig
-						.setBlockExceptionHandler(new DefaultBlockExceptionHandler());
-			}
-		}
-
-		urlCleanerOptional.ifPresent(sentinelWebMvcConfig::setUrlCleaner);
-		requestOriginParserOptional.ifPresent(sentinelWebMvcConfig::setOriginParser);
-		return sentinelWebMvcConfig;
-	}
+    private static final Logger log = LoggerFactory
+            .getLogger(SentinelWebAutoConfiguration.class);
+
+    @Autowired
+    private SentinelProperties properties;
+
+    @Autowired
+    private Optional<UrlCleaner> urlCleanerOptional;
+
+    @Autowired
+    private Optional<BlockExceptionHandler> blockExceptionHandlerOptional;
+
+    @Autowired
+    private Optional<RequestOriginParser> requestOriginParserOptional;
+
+    @Autowired
+    private Optional<SentinelWebInterceptor> sentinelWebInterceptorOptional;
+
+
+    @Override
+    public void addInterceptors(InterceptorRegistry registry) {
+        if (!sentinelWebInterceptorOptional.isPresent()) {
+            return;
+        }
+        SentinelProperties.Filter filterConfig = properties.getFilter();
+        registry.addInterceptor(sentinelWebInterceptorOptional.get())
+                .order(filterConfig.getOrder())
+                .addPathPatterns(filterConfig.getUrlPatterns());
+        log.info(
+                "[Sentinel Starter] register SentinelWebInterceptor with urlPatterns: {}.",
+                filterConfig.getUrlPatterns());
+    }
+
+    @Bean
+    @ConditionalOnProperty(name = "spring.cloud.sentinel.filter.enabled",
+            matchIfMissing = true)
+    public SentinelWebInterceptor sentinelWebInterceptor(
+            SentinelWebMvcConfig sentinelWebMvcConfig) {
+        return new SentinelWebInterceptor(sentinelWebMvcConfig);
+    }
+
+    @Bean
+    @ConditionalOnProperty(name = "spring.cloud.sentinel.filter.enabled",
+            matchIfMissing = true)
+    public SentinelWebMvcConfig sentinelWebMvcConfig() {
+        SentinelWebMvcConfig sentinelWebMvcConfig = new SentinelWebMvcConfig();
+        sentinelWebMvcConfig.setHttpMethodSpecify(properties.getHttpMethodSpecify());
+
+        if (blockExceptionHandlerOptional.isPresent()) {
+            blockExceptionHandlerOptional
+                    .ifPresent(sentinelWebMvcConfig::setBlockExceptionHandler);
+        } else {
+            if (StringUtils.hasText(properties.getBlockPage())) {
+                sentinelWebMvcConfig.setBlockExceptionHandler(((request, response,
+                                                                e) -> response.sendRedirect(properties.getBlockPage())));
+            } else {
+//                sentinelWebMvcConfig
+//                        .setBlockExceptionHandler(new DefaultBlockExceptionHandler());
+
+                // for loitTime custom sentinel exception by chenshiying
+                sentinelWebMvcConfig
+                        .setBlockExceptionHandler(new CustomBlockExceptionHandler());
+            }
+        }
+
+        urlCleanerOptional.ifPresent(sentinelWebMvcConfig::setUrlCleaner);
+        requestOriginParserOptional.ifPresent(sentinelWebMvcConfig::setOriginParser);
+        return sentinelWebMvcConfig;
+    }
 
 }

spring-cloud-timeloit-sentinel/src/main/java/com/timeloit/cloud/sentinel/loit/CustomBlockExceptionHandler.java

+package com.timeloit.cloud.sentinel.loit;
+
+import com.alibaba.csp.sentinel.adapter.spring.webmvc.callback.BlockExceptionHandler;
+import com.alibaba.csp.sentinel.slots.block.BlockException;
+import org.springframework.http.HttpStatus;
+import org.springframework.http.MediaType;
+
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+import java.io.PrintWriter;
+
+
+public class CustomBlockExceptionHandler implements BlockExceptionHandler {
+
+    @Override
+    public void handle(HttpServletRequest request, HttpServletResponse response, BlockException e) throws Exception {
+        response.setStatus(HttpStatus.OK.value());
+
+        System.out.println(MediaType.APPLICATION_JSON.toString());
+        response.setContentType(MediaType.APPLICATION_JSON_UTF8_VALUE);
+
+        PrintWriter out = response.getWriter();
+        out.print("{\"success\":false,\"code\":\"444\",\"msg\":\"系统繁忙请稍后再试\"}");
+        out.flush();
+        out.close();
+    }
+
+}
+

spring-cloud-timeloit-sentinel/src/main/java/com/timeloit/cloud/sentinel/loit/README.md

+# 时代凌宇sentinel 自定义异常类
+
+SentinelWebAutoConfiguration.sentinelWebMvcConfig
+```$xslt
+
+                // for loitTime custom sentinel exception by chenshiying
+                sentinelWebMvcConfig
+                        .setBlockExceptionHandler(new CustomBlockExceptionHandler());
+```
\ No newline at end of file