io.github.icodegarden.nursery.springboot.web.reactive.handler.ReactiveSentinelAdaptiveNativeRestApiExceptionHandler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of nursery-spring-boot-starter-web-reactive Show documentation
The newest version!
package io.github.icodegarden.nursery.springboot.web.reactive.handler;

import java.lang.reflect.UndeclaredThrowableException;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.http.ResponseEntity;
import org.springframework.lang.Nullable;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.server.ServerWebExchange;

import com.alibaba.csp.sentinel.slots.block.BlockException;
import com.alibaba.csp.sentinel.slots.block.authority.AuthorityException;
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.param.ParamFlowException;
import com.alibaba.csp.sentinel.slots.system.SystemBlockException;

import io.github.icodegarden.nursery.springcloud.sentinel.SentinelEventObserverRegistry;

/**
 * 使用 @Bean 

 * 支持对Sentinel异常的适配
 * 
 * @author Fangfang.Xu
 *
 */
public class ReactiveSentinelAdaptiveNativeRestApiExceptionHandler extends ReactiveNativeRestApiExceptionHandler {

	private static final Logger log = LoggerFactory
			.getLogger(ReactiveSentinelAdaptiveNativeRestApiExceptionHandler.class);

	private static final String CLIENT_LIMITED_LOG_MODULE = "Client-Limited Sentinel";

	/**
	 * 为BlockException单独设立一个
	 */
	@ExceptionHandler(BlockException.class)
	public ResponseEntity onBlockException(ServerWebExchange exchange, BlockException e) throws Exception {
		SentinelEventObserverRegistry.getInstance().notifyBlockException(e);

		String message = null;
		/**
		 * 以下一律是触发了但没有降级
		 */
		if (e instanceof SystemBlockException) {
			message = "sentinel-system-limited:" + e.getRule().getResource();
		} else if (e instanceof AuthorityException) {
			message = "client.sentinel-authority-limited:" + e.getRule().getResource();
		} else if (e instanceof FlowException) {
			message = "client.sentinel-flow-limited:" + e.getRule().getResource();
		} else if (e instanceof ParamFlowException) {
			message = "client.sentinel-paramflow-limited:" + e.getRule().getResource();
		} else if (e instanceof DegradeException) {
			message = "sentinel-degrade-limited:" + e.getRule().getResource();
		} else {
			message = "client.sentinel-" + e.getClass().getSimpleName() + "-limited:" + e.getRule().getResource();
		}

		if (log.isWarnEnabled()) {
			log.warn("{} {}", CLIENT_LIMITED_LOG_MODULE, message, e);
		}
		return ResponseEntity.status(403).body(message);
	}

	/**
	 * 为UndeclaredThrowableException单独设立一个
	 */
	@ExceptionHandler(UndeclaredThrowableException.class)
	public ResponseEntity onUndeclaredThrowableException(ServerWebExchange exchange,
			UndeclaredThrowableException ex) throws Exception {
		BlockException blockException = causeBlockException(ex);
		if (blockException != null) {
			return onBlockException(exchange, blockException);
		}

		return onException(exchange, ex);
	}

	private @Nullable BlockException causeBlockException(Throwable t) {
		int counter = 0;
		Throwable cause = t;
		while (cause != null && counter++ < 10) {
			if (cause instanceof BlockException) {
				return (BlockException) cause;
			}
			cause = cause.getCause();
		}
		return null;
	}
}