Examples with DefaultBlockExceptionHandler com.alibaba.csp.sentinel.adapter.spring.webmvc.callback.DefaultBlockExceptionHandler used on opensource projects

Example 1 with DefaultBlockExceptionHandler

use of com.alibaba.csp.sentinel.adapter.spring.webmvc.callback.DefaultBlockExceptionHandler in project Sentinel by alibaba.

the class InterceptorConfig method addSpringMvcInterceptor.

private void addSpringMvcInterceptor(InterceptorRegistry registry) {
    SentinelWebMvcConfig config = new SentinelWebMvcConfig();
    // Depending on your situation, you can choose to process the BlockException via
    // the BlockExceptionHandler or throw it directly, then handle it
    // in Spring web global exception handler.
    // config.setBlockExceptionHandler((request, response, e) -> { throw e; });
    // Use the default handler.
    config.setBlockExceptionHandler(new DefaultBlockExceptionHandler());
    // Custom configuration if necessary
    config.setHttpMethodSpecify(true);
    // By default web context is true, means that unify web context(i.e. use the default context name),
    // in most scenarios that's enough, and it could reduce the memory footprint.
    // If set it to false, entrance contexts will be separated by different URLs,
    // which is useful to support "chain" relation flow strategy.
    // We can change it and view different result in `Resource Chain` menu of dashboard.
    config.setWebContextUnify(true);
    config.setOriginParser(request -> request.getHeader("S-user"));
    // Add sentinel interceptor
    registry.addInterceptor(new SentinelWebInterceptor(config)).addPathPatterns("/**");
}

Example 2 with DefaultBlockExceptionHandler

use of com.alibaba.csp.sentinel.adapter.spring.webmvc.callback.DefaultBlockExceptionHandler in project spring-cloud-alibaba by alibaba.

the class SentinelWebAutoConfiguration method sentinelWebMvcConfig.

@Bean
@ConditionalOnProperty(name = "spring.cloud.sentinel.filter.enabled", matchIfMissing = true)
public SentinelWebMvcConfig sentinelWebMvcConfig() {
    SentinelWebMvcConfig sentinelWebMvcConfig = new SentinelWebMvcConfig();
    sentinelWebMvcConfig.setHttpMethodSpecify(properties.getHttpMethodSpecify());
    sentinelWebMvcConfig.setWebContextUnify(properties.getWebContextUnify());
    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;
}

Example 3 with DefaultBlockExceptionHandler

use of com.alibaba.csp.sentinel.adapter.spring.webmvc.callback.DefaultBlockExceptionHandler in project blade-tool by chillzhuang.

the class BladeSentinelFilterConfiguration method sentinelWebMvcConfig.

@Bean
public SentinelWebMvcConfig sentinelWebMvcConfig(SentinelProperties properties, Optional<UrlCleaner> urlCleanerOptional, Optional<BlockExceptionHandler> blockExceptionHandlerOptional, Optional<RequestOriginParser> requestOriginParserOptional) {
    SentinelWebMvcConfig sentinelWebMvcConfig = new SentinelWebMvcConfig();
    sentinelWebMvcConfig.setHttpMethodSpecify(properties.getHttpMethodSpecify());
    sentinelWebMvcConfig.setWebContextUnify(properties.getWebContextUnify());
    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;
}