27 lines
969 B
Java
27 lines
969 B
Java
package org.springblade.core.cloud.sentinel;
|
|
|
|
import com.alibaba.csp.sentinel.adapter.spring.webmvc.callback.BlockExceptionHandler;
|
|
import com.alibaba.csp.sentinel.slots.block.BlockException;
|
|
import org.springblade.core.tool.api.R;
|
|
import org.springblade.core.tool.jackson.JsonUtil;
|
|
import org.springframework.http.HttpStatus;
|
|
import org.springframework.http.MediaType;
|
|
|
|
import javax.servlet.http.HttpServletRequest;
|
|
import javax.servlet.http.HttpServletResponse;
|
|
|
|
/**
|
|
* Sentinel统一限流策略
|
|
*
|
|
* @author Chill
|
|
*/
|
|
public class BladeBlockExceptionHandler implements BlockExceptionHandler {
|
|
@Override
|
|
public void handle(HttpServletRequest request, HttpServletResponse response, BlockException e) throws Exception {
|
|
// Return 429 (Too Many Requests) by default.
|
|
response.setStatus(HttpStatus.TOO_MANY_REQUESTS.value());
|
|
response.setContentType(MediaType.APPLICATION_JSON_VALUE);
|
|
response.getWriter().print(JsonUtil.toJson(R.fail(e.getMessage())));
|
|
}
|
|
}
|