yudao
/
spring-cloud
mirror of https://gitee.com/zhijiantianya/yudao-cloud.git
3
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

规范优化

pull/137/head
koltZhang 2024-08-31 15:15:50 +08:00
parent 8a43cd4bef
commit 0724f10196
2 changed files with 13 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
yudao-framework/yudao-spring-boot-starter-sentinel/src/main/java/cn/iocoder/yudao/framework/sentinel/config/SentinelExceptionHandlerAutoConfiguration.java
View File

@ -6,12 +6,15 @@ import org.springframework.boot.autoconfigure.AutoConfiguration;
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean; import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Bean;
/**
* 当Spring容器中不存在MyService类型的bean时，才创建这个bean
*/
@AutoConfiguration @AutoConfiguration
@Slf4j @Slf4j
public class SentinelExceptionHandlerAutoConfiguration { public class SentinelExceptionHandlerAutoConfiguration {
@Bean @Bean
@ConditionalOnMissingBean // 当Spring容器中不存在MyService类型的bean时，才创建这个bean @ConditionalOnMissingBean
public SentinelExceptionHandler sentinelExceptionHandler() { public SentinelExceptionHandler sentinelExceptionHandler() {
return new SentinelExceptionHandler(); return new SentinelExceptionHandler();
} }

14
yudao-framework/yudao-spring-boot-starter-sentinel/src/main/java/cn/iocoder/yudao/framework/sentinel/core/handler/SentinelExceptionHandler.java
View File

@ -1,6 +1,7 @@
package cn.iocoder.yudao.framework.sentinel.core.handler; package cn.iocoder.yudao.framework.sentinel.core.handler;
import cn.iocoder.yudao.framework.common.pojo.CommonResult; import cn.iocoder.yudao.framework.common.pojo.CommonResult;
import cn.iocoder.yudao.framework.common.util.servlet.ServletUtils;
import com.alibaba.csp.sentinel.adapter.spring.webmvc.callback.BlockExceptionHandler; import com.alibaba.csp.sentinel.adapter.spring.webmvc.callback.BlockExceptionHandler;
import com.alibaba.csp.sentinel.slots.block.BlockException; import com.alibaba.csp.sentinel.slots.block.BlockException;
import com.alibaba.csp.sentinel.slots.block.authority.AuthorityException; import com.alibaba.csp.sentinel.slots.block.authority.AuthorityException;
@ -8,16 +9,19 @@ import com.alibaba.csp.sentinel.slots.block.degrade.DegradeException;
import com.alibaba.csp.sentinel.slots.block.flow.FlowException; import com.alibaba.csp.sentinel.slots.block.flow.FlowException;
import com.alibaba.csp.sentinel.slots.block.flow.param.ParamFlowException; import com.alibaba.csp.sentinel.slots.block.flow.param.ParamFlowException;
import com.alibaba.csp.sentinel.slots.system.SystemBlockException; import com.alibaba.csp.sentinel.slots.system.SystemBlockException;
import com.alibaba.fastjson.JSON;
import org.springframework.http.HttpStatus; import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse; import javax.servlet.http.HttpServletResponse;
/**
* Sentinel自定义阻断异常处理
*/
public class SentinelExceptionHandler implements BlockExceptionHandler { public class SentinelExceptionHandler implements BlockExceptionHandler {
@Override @Override
public void handle(HttpServletRequest request, HttpServletResponse response, BlockException e) throws Exception { public void handle(HttpServletRequest request, HttpServletResponse response, BlockException e) {
String msg = "未知异常"; String msg = "未知异常";
int status = HttpStatus.TOO_MANY_REQUESTS.value(); int status = HttpStatus.TOO_MANY_REQUESTS.value();
if (e instanceof FlowException) { if (e instanceof FlowException) {
@ -33,8 +37,6 @@ public class SentinelExceptionHandler implements BlockExceptionHandler {
msg = "系统规则限流或降级"; msg = "系统规则限流或降级";
} }
response.setContentType(MediaType.APPLICATION_JSON_UTF8_VALUE); ServletUtils.writeJSON(response, CommonResult.error(status, msg));
response.setStatus(status);
response.getOutputStream().println(JSON.toJSONString(CommonResult.error(status,msg)));
} }
} }